make pascal compile (#42)

turboderp · Jun 8, 2023 · ab81db1 · ab81db1
1 parent e449281
commit ab81db1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/exllama_ext/cuda_compat.cuh b/exllama_ext/cuda_compat.cuh
@@ -45,9 +45,12 @@ __device__ __forceinline__ void atomicAdd_half2(half2* address, half2 val)
 #if __CUDA_ARCH__ < 700 || defined(USE_ROCM)
 
 __device__ __forceinline__ void atomicAdd(half* address, half val) { atomicAdd_half(address, val); }
+
+#if __CUDA_ARCH__ < 600 || defined(USE_ROCM)
 __device__ __forceinline__ void atomicAdd(half2* address, half2 val) { atomicAdd_half2(address, val); }
+#endif
 
 #endif
 #endif
 
-#endif
+#endif