Carlos-Francisco Méndez-Cruz / curso-alianza-2024

Go to a project
Toggle navigation Toggle navigation pinning
cmendezc
f11ed5a1 1 parent cd2dc1bb

NPU examples

Inline Side-by-side
Showing 1 changed file with 13 additions and 0 deletions
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