fixed __CUDA_ARCH__ check for compat with cuda 12.2 (#855)

* fixed __CUDA_ARCH__ check for compat with cuda 12.2 * Update src/tensor_ops/utilities/compatibility.cuh Co-authored-by: Corey Lowman <clowman1993@gmail.com> --------- Co-authored-by: Corey Lowman <clowman1993@gmail.com>
coreylowman · Aug 22, 2023 · 47855ea · 47855ea
1 parent 923c430
commit 47855ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/tensor_ops/utilities/compatibility.cuh b/src/tensor_ops/utilities/compatibility.cuh
@@ -14,7 +14,7 @@
 // }
 // #endif
 
-#if __CUDA_ARCH__ < 800
+#if (__CUDACC_VER_MAJOR__ < 12 || __CUDACC_VER_MINOR__ < 2) && __CUDA_ARCH__ < 800
 __device__ __forceinline__ __half __hmax_nan(__half a, __half b) {
     return __hisnan(a) ? a : (__hisnan(b) ? b : __hmax(a, b));
 }