Fix ROCm compile

turboderp · turboderp · commit 2586566840c7 · 2024-02-24T08:00:30.000+01:00
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)

Original file line number	Diff line number	Diff line change
`@@ -55,7 +55,7 @@ __device__ __forceinline__ half gelu(half x)`
`55`	`55`
`56`	`56`	`__device__ __forceinline__ half2 gelu(half2 x)`
`57`	`57`	`{`
`58`		`- return __halves2half2(gelu(x.x), gelu(x.y));`
	`58`	`+ return __halves2half2(gelu(__low2half(x)), gelu(__high2half(x)));`
`59`	`59`	`}`
`60`	`60`
`61`	`61`	`typedef void (*fp_silu_mul_kernel)`