From b7f4da04e198b3ae48907fdf991fc30110a0c16f Mon Sep 17 00:00:00 2001
From: Ruihang Lai <ruihangl@cs.cmu.edu>
Date: Wed, 10 Sep 2025 15:35:10 -0400
Subject: [PATCH] [CUDA] Support NVTX in CUDA 13

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 | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cmake/utils/FindCUDA.cmake b/cmake/utils/FindCUDA.cmake
index 2036c7c32994..c4c18eef0f80 100644
--- a/cmake/utils/FindCUDA.cmake
+++ b/cmake/utils/FindCUDA.cmake
@@ -101,7 +101,7 @@ macro(find_cuda use_cuda use_cudnn)
         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"