Check for __CUDA_ARCH_LIST__ for setting GT_CUDA_ARCH as well

GridTools · Apr 12, 2024 · 8f97c4b · 8f97c4b
1 parent c60293c
commit 8f97c4b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/include/gridtools/common/defs.hpp b/include/gridtools/common/defs.hpp
@@ -18,6 +18,8 @@ namespace gridtools {
 #define GT_CUDACC
 #ifdef __CUDA_ARCH__
 #define GT_CUDA_ARCH __CUDA_ARCH__
+#elif defined(__CUDA_ARCH_LIST__)
+#define GT_CUDA_ARCH __CUDA_ARCH_LIST__
 #endif
 #elif defined(__HIP__)
 #define GT_CUDACC