wwwip

pytorch · Dec 25, 2024 · 9e6a077 · 9e6a077
1 parent b0dd312
commit 9e6a077
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/.github/scripts/utils_cuda.bash b/.github/scripts/utils_cuda.bash
@@ -22,9 +22,11 @@ __set_cuda_symlinks_envvars () {
     print_exec ln -sf ${conda_prefix}/lib/libnvToolsExt.so.1 ${conda_prefix}/lib/libnvToolsExt.so
     print_exec ln -sf ${conda_prefix}/targets/x86_64-linux/lib/libnvToolsExt.so.1 ${conda_prefix}/targets/x86_64-linux/lib/libnvToolsExt.so
 
-    for file in ${conda_prefix}/nsight-compute-2024.3.1/host/target-linux-x64/nvtx/include/nvtx3/*; do
-      cp ${file} ${conda_prefix}/include/
-    done
+    cp ${conda_prefix}/nsight-compute-2024.3.1/host/target-linux-x64/nvtx/include/nvtx3/nvToolsExt.h ${conda_prefix}/include/
+    cp ${conda_prefix}/nsight-compute-2024.3.1/host/target-linux-x64/nvtx/include/nvtx3/nvToolsExtOpenCL.h ${conda_prefix}/include/
+    cp ${conda_prefix}/nsight-compute-2024.3.1/host/target-linux-x64/nvtx/include/nvtx3/nvToolsExtSync.h ${conda_prefix}/include/
+    cp ${conda_prefix}/nsight-compute-2024.3.1/host/target-linux-x64/nvtx/include/nvtx3/nvToolsExtCudaRt.h ${conda_prefix}/include/
+    cp ${conda_prefix}/nsight-compute-2024.3.1/host/target-linux-x64/nvtx/include/nvtx3/nvToolsExtCuda.h ${conda_prefix}/include/
   fi
 
   # if [ -f ${CONDA_PREFIX}/targets/x86_64-linux/lib/libnvToolsExt.so.1 ] && [ ! -f ${CONDA_PREFIX}/targets/x86_64-linux/lib/libnvToolsExt.so ]; then