Merge pull request #266 from NVIDIA/nvcxx-stdpar-no-cudacc

nvc++ and __CUDACC__
NVIDIA · Mar 1, 2021 · 7cdf6df · 7cdf6df
2 parents fef1b9c + 5867442
commit 7cdf6df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cub/util_compiler.cuh b/cub/util_compiler.cuh
@@ -63,7 +63,7 @@
 #endif // CUB_HOST_COMPILER
 
 // figure out which device compiler we're using
-#if defined(__CUDACC__)
+#if defined(__CUDACC__) || defined(__NVCOMPILER_CUDA__)
 #  define CUB_DEVICE_COMPILER CUB_DEVICE_COMPILER_NVCC
 #elif CUB_HOST_COMPILER == CUB_HOST_COMPILER_MSVC
 #  define CUB_DEVICE_COMPILER CUB_DEVICE_COMPILER_MSVC