wwwip

pytorch · Dec 25, 2024 · c9db629 · c9db629
1 parent 6747d9b
commit c9db629
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.github/scripts/utils_cuda.bash b/.github/scripts/utils_cuda.bash
@@ -21,9 +21,9 @@ __set_cuda_symlinks_envvars () {
     # if [ -f ${conda_prefix}/lib/libnvToolsExt.so.1 ] && [ ! -f ${conda_prefix}/lib/libnvToolsExt.so ]; then
     echo "[INSTALL] Fixing file placements for NVTX library ..."
     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
-    print_exec cp -r ${conda_prefix}/nsight-compute*/host/target-linux-x64/nvtx/include/nvtx3/* ${conda_prefix}/include
-    print_exec cp -r ${conda_prefix}/nsight-compute*/host/target-linux-x64/nvtx/include/nvtx3/* ${conda_prefix}/targets/x86_64-linux/include
+    print_exec ln -sf ${conda_prefix}/targets/${MACHINE_NAME_LC}-linux/lib/libnvToolsExt.so.1 ${conda_prefix}/targets/${MACHINE_NAME_LC}-linux/lib/libnvToolsExt.so
+    print_exec cp -r ${conda_prefix}/nsight-compute*/host/target-linux-${CONDA_ARCHNAME}/nvtx/include/nvtx3/* ${conda_prefix}/include
+    print_exec cp -r ${conda_prefix}/nsight-compute*/host/target-linux-${CONDA_ARCHNAME}/nvtx/include/nvtx3/* ${conda_prefix}/targets/x86_64-linux/include
   fi
 
   # The symlink appears to be missing when we attempt to run FBGEMM_GPU on the