mirror of
https://github.com/vladmandic/automatic
synced 2026-09-17 08:19:11 +02:00
update zluda
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user