Fix ROCm compile

turboderp · Feb 24, 2024 · 2586566 · 2586566
1 parent 67af1d1
commit 2586566
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/exllamav2/exllamav2_ext/cuda/q_mlp.cu b/exllamav2/exllamav2_ext/cuda/q_mlp.cu
@@ -55,7 +55,7 @@ __device__ __forceinline__ half gelu(half x)
 
 __device__ __forceinline__ half2 gelu(half2 x)
 {
-    return __halves2half2(gelu(x.x), gelu(x.y));
+    return __halves2half2(gelu(__low2half(x)), gelu(__high2half(x)));
 }
 
 typedef void (*fp_silu_mul_kernel)