update zluda

This commit is contained in:
Seunghoon Lee
2024-08-24 21:00:45 +09:00
parent c1285c6de6
commit ab9a4d37cc
2 changed files with 3 additions and 2 deletions
+2 -1
View File
@@ -1,4 +1,5 @@
import torch
from modules import rocm
_topk = torch.topk
@@ -14,6 +15,6 @@ def jit_script(f, *_, **__): # experiment / provide dummy graph
def do_hijack():
torch.version.hip = "5.7"
torch.version.hip = rocm.version
torch.topk = topk
torch.jit.script = jit_script
+1 -1
View File
@@ -26,7 +26,7 @@ def install(zluda_path: os.PathLike) -> None:
default_hash = None
if rocm.version == "6.1":
default_hash = 'd7714d84c0c13bbf816eaaac32693e4e75e58a87'
default_hash = '2f2e38a8adebb456ad75390e60871f2c8ba18fa7'
elif rocm.version == "5.7":
default_hash = '11cc5844514f93161e0e74387f04e2c537705a82'
else: