Fix nvjitlink inclusion in 12.1 wheels (#1397)

atalman · web-flow · commit 843d6e9b43ec · 2023-04-27T17:24:13.000-07:00
* Fix nvjitlink inclusion in 12.1 wheels

* Fix typo
diff --git a/windows/internal/copy.bat b/windows/internal/copy.bat
@@ -18,3 +18,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
+)

Original file line number	Diff line number	Diff line change
`@@ -18,3 +18,8 @@ copy "%libuv_ROOT%\bin\uv.dll" pytorch\torch\lib`
`18`	`18`	`if exist "C:\Windows\System32\zlibwapi.dll" (`
`19`	`19`	`copy "C:\Windows\System32\zlibwapi.dll" pytorch\torch\lib`
`20`	`20`	`)`
	`21`	`+`
	`22`	`+::copy nvJitLink dll is requires for cuda 12+`
	`23`	`+if exist "%CUDA_PATH%\bin\nvJitLink_.dll" (`
	`24`	`+ copy "%CUDA_PATH%\bin\nvJitLink_.dll" pytorch\torch\lib`
	`25`	`+)`