Uploaded sanitized BC250/ROCm Repository.
This commit is contained in:
@@ -0,0 +1,7 @@
|
||||
import torch
|
||||
print("CUDA:", torch.cuda.is_available())
|
||||
t = torch.randn(4, 4, device="cuda")
|
||||
print("GPU tensor:", t.shape)
|
||||
r = t @ t.T
|
||||
print("matmul:", r.shape)
|
||||
print("OK")
|
||||
Reference in New Issue
Block a user