diff --git a/cli/zluda-python.py b/cli/zluda-python.py index 958f2121c..31ee96362 100644 --- a/cli/zluda-python.py +++ b/cli/zluda-python.py @@ -26,8 +26,11 @@ class Interpreter: if __name__ == '__main__': sys.path.append(os.getcwd()) - from modules.zluda_installer import find, load - load(find()) + from modules import zluda_installer + zluda_path = zluda_installer.get_path() + zluda_installer.install(zluda_path) + zluda_installer.make_copy(zluda_path) + zluda_installer.load(zluda_path) import torch interpreter = Interpreter({ diff --git a/modules/zluda.py b/modules/zluda.py index b181f60a4..55f83ff30 100644 --- a/modules/zluda.py +++ b/modules/zluda.py @@ -20,6 +20,7 @@ def test(device: DeviceLikeType) -> Union[Exception, None]: ten2 = torch.randn((4, 8,), device=device) out = torch.mm(ten1, ten2) assert out.sum().is_nonzero() + return None except Exception as e: return e