Skip to content

Commit 8b02247

Browse files
committed
Make nvtx future proof
1 parent 8fef22d commit 8b02247

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

CMakeLists.txt

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -526,8 +526,12 @@ function(MFC_SETUP_TARGET)
526526
endif()
527527

528528
if (CMAKE_Fortran_COMPILER_ID STREQUAL "NVHPC" OR CMAKE_Fortran_COMPILER_ID STREQUAL "PGI")
529-
find_package(CUDAToolkit REQUIRED)
530-
target_link_libraries(${a_target} PRIVATE CUDA::nvToolsExt)
529+
if (TARGET CUDA::nvToolsExt) # CUDA <= 12.8
530+
target_link_libraries(${a_target} PRIVATE CUDA::nvToolsExt)
531+
else() # CUDA >= 12.9
532+
target_link_libraries(${a_target} PRIVATE nvhpcwrapnvtx )
533+
endif()
534+
target_link_options(${a_target} PRIVATE "-cudalib=nvtx")
531535
endif()
532536
endforeach()
533537

0 commit comments

Comments
 (0)