From 6030e950169c284ae55cfcab9144da5babff685d Mon Sep 17 00:00:00 2001
From: Seunghoon Lee <lshqqytiger@naver.com>
Date: Fri, 10 Jan 2025 15:52:07 +0900
Subject: [PATCH] add zluda nightly flag

---
 modules/zluda_installer.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/modules/zluda_installer.py b/modules/zluda_installer.py
index 3593f2641..a3cc6d7ab 100644
--- a/modules/zluda_installer.py
+++ b/modules/zluda_installer.py
@@ -34,10 +34,11 @@ def install(zluda_path: os.PathLike) -> None:
         __initialize(zluda_path)
         return
 
-    commit = os.environ.get("ZLUDA_HASH", "1b6e012d8f2404840b524e2abae12cb91e1ac01d")
-    if rocm.version == "6.1":
-        commit = "c0804ca624963aab420cb418412b1c7fbae3454b"
-    urllib.request.urlretrieve(f'https://github.com/lshqqytiger/ZLUDA/releases/download/rel.{commit}/ZLUDA-windows-rocm{rocm.version[0]}-amd64.zip', '_zluda')
+    platform = "windows"
+    commit = os.environ.get("ZLUDA_HASH", "d60bddbc870827566b3d2d417e00e1d2d8acc026")
+    if os.environ.get("ZLUDA_NIGHTLY", "0") == "1":
+        platform = "nightly-" + platform
+    urllib.request.urlretrieve(f'https://github.com/lshqqytiger/ZLUDA/releases/download/rel.{commit}/ZLUDA-{platform}-rocm{rocm.version[0]}-amd64.zip', '_zluda')
     with zipfile.ZipFile('_zluda', 'r') as archive:
         infos = archive.infolist()
         for info in infos: