NPU examples

cmendezc
Commit f11ed5a173325c17b03d06fbec2403af72281597 f11ed5a1 1 parent cd2dc1bb
Showing 1 changed file with 13 additions and 0 deletions
ejemplo-npu/quick_verification_torch1.11.py
--- a/ejemplo-npu/quick_verification_torch1.11.py 0 → 100644
View file @f11ed5a
+++ b/ejemplo-npu/quick_verification_torch1.11.py 0 → 100644
View file @f11ed5a
+import torch
+import torch_npu
+
+print("NPU available?", torch.npu.is_available())
+print("Current device:", torch.npu.current_device())
+device = torch.device("npu:0")
+# x = torch.randn(2, 2).npu()
+# y = torch.randn(2, 2).npu()
+x = torch.randn(2, 2).to(device)
+y = torch.randn(2, 2).to(device)
+z = x.mm(y)
+
+print(z)
\ No newline at end of file