commit 865d8f079c06456ad24e4aede11954812c04abd4
author Ruihang Lai <ruihangl@cs.cmu.edu> Thu Sep 11 12:53:39 2025 -0400
committer GitHub <noreply@github.com> Thu Sep 11 12:53:39 2025 -0400
tree b9930a84cf8cd79a35d9b30bad4847469c27a838
parent b07c6c1e7502536db6c6f8c8696cc4c7f6bc46a1

[CUDA] Support NVTX in CUDA 13 (#18300)

This PR adds the support of NVTX for CUDA 13. The change is because
that starting CUDA 13, the nvtx functions are moved to the lirbary
of `libnvtx3interop.so`, and the previous nvToolsExt library no
longer exists.

To ensure compatibility with both CUDA 12 and 13, we add
libnvtx3interop.so to the library lookup list.

cmake/utils/FindCUDA.cmake

diff --git a/cmake/utils/FindCUDA.cmake b/cmake/utils/FindCUDA.cmake
index 2036c7c..c4c18ee 100644
--- a/cmake/utils/FindCUDA.cmake
+++ b/cmake/utils/FindCUDA.cmake
@@ -101,7 +101,7 @@
         PATH_SUFFIXES lib lib64 targets/x86_64-linux/lib targets/x86_64-linux/lib/stubs lib64/stubs lib/x86_64-linux-gnu
         NO_DEFAULT_PATH)
       find_library(CUDA_NVTX_LIBRARY
-        NAMES nvToolsExt nvTools nvtoolsext nvtools nvtx NVTX
+        NAMES nvToolsExt nvTools nvtoolsext nvtools nvtx NVTX nvtx3interop
         PATHS "${CUDA_CUDART_LIBRARY_DIR}" "${CUDA_TOOLKIT_ROOT_DIR}" ENV LD_LIBRARY_PATH
         PATH_SUFFIXES "lib64" "common/lib64" "common/lib" "lib"
         DOC "Location of the CUDA Toolkit Extension (NVTX) library"