diff --git a/modules/zluda_hijacks.py b/modules/zluda_hijacks.py index 77744c8f0..642c7530d 100644 --- a/modules/zluda_hijacks.py +++ b/modules/zluda_hijacks.py @@ -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 diff --git a/modules/zluda_installer.py b/modules/zluda_installer.py index 7c20d3d79..20ecb233d 100644 --- a/modules/zluda_installer.py +++ b/modules/zluda_installer.py @@ -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: