mirror of
https://github.com/vladmandic/automatic
synced 2026-09-18 16:54:33 +02:00
zluda minor fixes
This commit is contained in:
+5
-2
@@ -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({
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user