Skip to content

Commit e47036b

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

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

CMakeLists.txt

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -527,7 +527,12 @@ function(MFC_SETUP_TARGET)
527527

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

0 commit comments

Comments
 (0)