diff --git a/patchzluda2.bat b/patchzluda2.bat index f6393c722..76912480c 100644 --- a/patchzluda2.bat +++ b/patchzluda2.bat @@ -84,9 +84,15 @@ if not exist "%torch_dir%" ( :: Copy necessary files echo Copying files to Torch library... -copy zluda\cublas.dll "%torch_dir%\cublas64_11.dll" /Y >nul 2>&1 -copy zluda\cusparse.dll "%torch_dir%\cusparse64_11.dll" /Y >nul 2>&1 -copy zluda\nvrtc.dll "%torch_dir%\nvrtc64_112_0.dll" /Y >nul 2>&1 +if "%target_dir%"=="zluda" ( + copy cublas.dll "%torch_dir%\cublas64_11.dll" /Y >nul 2>&1 + copy cusparse.dll "%torch_dir%\cusparse64_11.dll" /Y >nul 2>&1 + copy nvrtc.dll "%torch_dir%\nvrtc64_112_0.dll" /Y >nul 2>&1 +) else ( + copy zluda\cublas.dll "%torch_dir%\cublas64_11.dll" /Y >nul 2>&1 + copy zluda\cusparse.dll "%torch_dir%\cusparse64_11.dll" /Y >nul 2>&1 + copy zluda\nvrtc.dll "%torch_dir%\nvrtc64_112_0.dll" /Y >nul 2>&1 +) if "%target_dir%"=="zluda" ( cd .. @@ -98,4 +104,4 @@ echo * ZLUDA has been successfully patched from the URL: %zl% echo. echo Press any key to close. pause -exit /b \ No newline at end of file +exit /b