From f775de18841fa02533e751eda520fab141eadb4f Mon Sep 17 00:00:00 2001 From: Stan Gerald Moore Date: Wed, 15 Feb 2023 10:02:25 -0700 Subject: [PATCH] Fix Kokkos compile error with NVIDIA Hopper GPU --- lib/kokkos/core/src/Cuda/Kokkos_Cuda_Half_Conversion.hpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lib/kokkos/core/src/Cuda/Kokkos_Cuda_Half_Conversion.hpp b/lib/kokkos/core/src/Cuda/Kokkos_Cuda_Half_Conversion.hpp index 8c40ebd60d..f25d7bd124 100644 --- a/lib/kokkos/core/src/Cuda/Kokkos_Cuda_Half_Conversion.hpp +++ b/lib/kokkos/core/src/Cuda/Kokkos_Cuda_Half_Conversion.hpp @@ -290,7 +290,8 @@ KOKKOS_INLINE_FUNCTION // Go in this branch if CUDA version is >= 11.0.0 and less than 11.1.0 or if the // architecture is not Ampere #if CUDA_VERSION >= 11000 && \ - (CUDA_VERSION < 11010 || !defined(KOKKOS_ARCH_AMPERE)) + (CUDA_VERSION < 11010 || \ + !(defined(KOKKOS_ARCH_AMPERE) || defined(KOKKOS_ARCH_HOPPER))) KOKKOS_INLINE_FUNCTION bhalf_t cast_to_bhalf(bhalf_t val) { return val; }