Improve macro name KOKKOS_IMPL_{ARCH_NVIDIA_GPU_AMPERE_PLUS -> NVIDIA…

…_GPU_ARCH_SUPPORT_BHALF}
masterleinad · Mar 8, 2023 · b10f35e · b10f35e
1 parent 1d26ca8
commit b10f35e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/core/src/Cuda/Kokkos_Cuda_Half_Conversion.hpp b/core/src/Cuda/Kokkos_Cuda_Half_Conversion.hpp
@@ -264,12 +264,12 @@ KOKKOS_INLINE_FUNCTION
 #if !defined(KOKKOS_ARCH_KEPLER) && !defined(KOKKOS_ARCH_MAXWELL) && \
     !defined(KOKKOS_ARCH_PASCAL) && !defined(KOKKOS_ARCH_VOLTA) &&   \
     !defined(KOKKOS_ARCH_TURING75)
-#define KOKKOS_IMPL_ARCH_NVIDIA_GPU_AMPERE_PLUS
+#define KOKKOS_IMPL_NVIDIA_GPU_ARCH_SUPPORT_BHALF
 #endif
 
 #if CUDA_VERSION >= 11000 && \
     (CUDA_VERSION < 11010 || \
-     !defined(KOKKOS_IMPL_ARCH_NVIDIA_GPU_AMPERE_PLUS))
+     !defined(KOKKOS_IMPL_NVIDIA_GPU_ARCH_SUPPORT_BHALF))
 KOKKOS_INLINE_FUNCTION
 bhalf_t cast_to_bhalf(bhalf_t val) { return val; }
 
@@ -396,7 +396,7 @@ KOKKOS_INLINE_FUNCTION
 }
 #endif  // CUDA_VERSION >= 11000 && CUDA_VERSION < 11010
 
-#if CUDA_VERSION >= 11010 && defined(KOKKOS_IMPL_ARCH_NVIDIA_GPU_AMPERE_PLUS)
+#if CUDA_VERSION >= 11010 && defined(KOKKOS_IMPL_NVIDIA_GPU_ARCH_SUPPORT_BHALF)
 KOKKOS_INLINE_FUNCTION
 bhalf_t cast_to_bhalf(bhalf_t val) { return val; }
 KOKKOS_INLINE_FUNCTION
@@ -479,7 +479,7 @@ KOKKOS_INLINE_FUNCTION
 }
 #endif  // CUDA_VERSION >= 11010
 
-#undef KOKKOS_IMPL_ARCH_NVIDIA_GPU_AMPERE_PLUS
+#undef KOKKOS_IMPL_NVIDIA_GPU_ARCH_SUPPORT_BHALF
 }  // namespace Experimental
 
 #if (CUDA_VERSION >= 11000)