zluda minor fixes

This commit is contained in:
Seunghoon Lee
2024-05-21 10:58:42 +09:00
parent 8206af49bf
commit 9267b9b4e2
2 changed files with 6 additions and 2 deletions
+5 -2
View File
@@ -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({
+1
View File
@@ -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