From 87abd9a6a2b200158d57a32d9eb9c10e6d0e6410 Mon Sep 17 00:00:00 2001 From: atalman Date: Thu, 27 Apr 2023 17:19:58 -0700 Subject: [PATCH 1/2] Fix nvjitlink inclusion in 12.1 wheels --- windows/internal/copy.bat | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/windows/internal/copy.bat b/windows/internal/copy.bat index cd88eca20..67d766f5f 100755 --- a/windows/internal/copy.bat +++ b/windows/internal/copy.bat @@ -7,6 +7,7 @@ copy "%CUDA_PATH%\bin\cusolver*64_*.dll*" pytorch\torch\lib copy "%CUDA_PATH%\bin\cudnn*64_*.dll*" pytorch\torch\lib copy "%CUDA_PATH%\bin\nvrtc*64_*.dll*" pytorch\torch\lib +copy "%CUDA_PATH%\bin\nvrtc*64_*.dll*" pytorch\torch\lib copy "%CUDA_PATH%\extras\CUPTI\lib64\cupti64_*.dll*" pytorch\torch\lib copy "C:\Program Files\NVIDIA Corporation\NvToolsExt\bin\x64\nvToolsExt64_1.dll*" pytorch\torch\lib @@ -18,3 +19,8 @@ copy "%libuv_ROOT%\bin\uv.dll" pytorch\torch\lib if exist "C:\Windows\System32\zlibwapi.dll" ( copy "C:\Windows\System32\zlibwapi.dll" pytorch\torch\lib ) + +::copy nvJitLink dll is requires for cuda 12+ +if exist "%CUDA_PATH%\bin\nvJitLink_*.dll*" ( + copy "%CUDA_PATH%\bin\nvJitLink_*.dll*" pytorch\torch\lib +) From d8148777421058544aaf47ead48ae1167baa1093 Mon Sep 17 00:00:00 2001 From: atalman Date: Thu, 27 Apr 2023 17:21:14 -0700 Subject: [PATCH 2/2] Fix typo --- windows/internal/copy.bat | 1 - 1 file changed, 1 deletion(-) diff --git a/windows/internal/copy.bat b/windows/internal/copy.bat index 67d766f5f..490d9593a 100755 --- a/windows/internal/copy.bat +++ b/windows/internal/copy.bat @@ -7,7 +7,6 @@ copy "%CUDA_PATH%\bin\cusolver*64_*.dll*" pytorch\torch\lib copy "%CUDA_PATH%\bin\cudnn*64_*.dll*" pytorch\torch\lib copy "%CUDA_PATH%\bin\nvrtc*64_*.dll*" pytorch\torch\lib -copy "%CUDA_PATH%\bin\nvrtc*64_*.dll*" pytorch\torch\lib copy "%CUDA_PATH%\extras\CUPTI\lib64\cupti64_*.dll*" pytorch\torch\lib copy "C:\Program Files\NVIDIA Corporation\NvToolsExt\bin\x64\nvToolsExt64_1.dll*" pytorch\torch\lib