[AMD][CI/Build] Disambiguation of the function call for ROCm 6.2 headers compatibility (vllm-project#7477)

gshtras · charlifu · web-flow · commit 9984605412de · 2024-08-21T16:47:36.000-07:00
Co-authored-by: Charlie Fu &lt;Charlie.Fu@amd.com&gt;
diff --git a/csrc/attention/attention_utils.cuh b/csrc/attention/attention_utils.cuh
@@ -34,7 +34,7 @@ inline __device__ float qk_dot_(const Vec (&q)[N], const Vec (&k)[N]) {
   A_vec qk_vec = mul<A_vec, Vec, Vec>(q[0], k[0]);
 #pragma unroll
   for (int ii = 1; ii < N; ++ii) {
-    qk_vec = fma(q[ii], k[ii], qk_vec);
+    qk_vec = vllm::fma(q[ii], k[ii], qk_vec);
   }
 
   // Finalize the reduction across lanes.

Original file line number	Diff line number	Diff line change
`@@ -34,7 +34,7 @@ inline __device__ float qk_dot_(const Vec (&q)[N], const Vec (&k)[N]) {`
`34`	`34`	`A_vec qk_vec = mul<A_vec, Vec, Vec>(q[0], k[0]);`
`35`	`35`	`#pragma unroll`
`36`	`36`	`for (int ii = 1; ii < N; ++ii) {`
`37`		`- qk_vec = fma(q[ii], k[ii], qk_vec);`
	`37`	`+ qk_vec = vllm::fma(q[ii], k[ii], qk_vec);`
`38`	`38`	`}`
`39`	`39`
`40`	`40`	`// Finalize the reduction across lanes.`