From 47f74f2265221ae8511044ea7be4a8f686c5c375 Mon Sep 17 00:00:00 2001
From: patientx <patient.x@gmail.com>
Date: Sun, 26 Jan 2025 15:28:35 +0300
Subject: [PATCH] corrected the copying part , now nightly builds correctly
 copied

---
 patchzluda2.bat | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

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