Do not query NVRTC for cuda runtime header (#2102)

This may fail as discussed in #2095 Fixes 2095
NVIDIA · Jul 30, 2024 · 1e67aa7 · 1e67aa7
1 parent 15e2ce0
commit 1e67aa7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cub/cub/detail/detect_cuda_runtime.cuh b/cub/cub/detail/detect_cuda_runtime.cuh
@@ -44,7 +44,10 @@
 #  pragma system_header
 #endif // no system header
 
-#include <cuda_runtime_api.h>
+// CUDA headers might not be present when using NVRTC, see NVIDIA/cccl#2095 for detail
+#if !defined(_CCCL_COMPILER_NVRTC)
+#  include <cuda_runtime_api.h>
+#endif // !_CCCL_COMPILER_NVRTC
 
 #ifdef DOXYGEN_SHOULD_SKIP_THIS // Only parse this during doxygen passes: