Update more CUDA compiler ID checks.

NVIDIA · Jun 6, 2023 · 1d94385 · 1d94385
1 parent 9d4a366
commit 1d94385
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/cmake/CubBuildCompilerTargets.cmake b/cmake/CubBuildCompilerTargets.cmake
@@ -83,7 +83,7 @@ function(cub_build_compiler_targets)
     endif()
   endif()
 
-  if ("NVCXX" STREQUAL "${CMAKE_CUDA_COMPILER_ID}")
+  if ("NVHPC" STREQUAL "${CMAKE_CUDA_COMPILER_ID}")
     list(APPEND cxx_compile_options -Mnodaz)
     # TODO: Managed memory is currently not supported on windows with WSL
     list(APPEND cxx_compile_options -gpu=nomanaged)

diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
@@ -1,4 +1,4 @@
-if ("NVCXX" STREQUAL "${CMAKE_CUDA_COMPILER_ID}")
+if ("NVHPC" STREQUAL "${CMAKE_CUDA_COMPILER_ID}")
   # NVBugs 200770766
   set(CUB_SEPARATE_CATCH2 ON)
 else()
@@ -152,7 +152,7 @@ function(cub_add_test target_name_var test_name test_src cub_target)
         PROPERTY POSITION_INDEPENDENT_CODE ON
       )
 
-      if ("NVCXX" STREQUAL "${CMAKE_CUDA_COMPILER_ID}")
+      if ("NVHPC" STREQUAL "${CMAKE_CUDA_COMPILER_ID}")
         target_link_options(${config_c2h_target} PRIVATE "-cuda")
         target_compile_options(${config_c2h_target} PRIVATE "-fPIC")
       endif()
@@ -195,7 +195,7 @@ function(cub_add_test target_name_var test_name test_src cub_target)
         target_include_directories(${config_c2run_target} PRIVATE
           "${CUB_SOURCE_DIR}/test"
         )
-        if ("NVCXX" STREQUAL "${CMAKE_CUDA_COMPILER_ID}")
+        if ("NVHPC" STREQUAL "${CMAKE_CUDA_COMPILER_ID}")
           target_link_options(${config_c2run_target} PRIVATE "-cuda")
         endif()