fix build on SM 5.2 (NVIDIA#1664)

MaxAkaAltmer · Jul 31, 2024 · fbd116c · fbd116c
1 parent 5b283c8
commit fbd116c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/cutlass/functional.h b/include/cutlass/functional.h
@@ -234,7 +234,7 @@ template <>
 struct inverse_square_root<half_t> {
   CUTLASS_HOST_DEVICE
   half_t operator()(half_t const &lhs) const {
-#if defined(__CUDA_ARCH__)
+#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ > 520
     auto result = hrsqrt(reinterpret_cast<__half const &>(lhs));
     return reinterpret_cast<half_t const &>(result);
 #else