github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[DOCS] fixed typo triton.testing.allclose -> torch.allclose in MatMul tutorial (#1460)

Browse Source
This commit is contained in:
Kern Handa
2023-03-31 17:06:46 -07:00
committed by GitHub
parent cc4aa1ebbc
commit 2c0417da96
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tutorials/03-matrix-multiplication.py
View File

@@ -293,7 +293,7 @@ triton_output = matmul(a, b, activation=None)
torch_output = torch.matmul(a, b)
print(f"triton_output={triton_output}")
print(f"torch_output={torch_output}")
if triton.testing.allclose(triton_output, torch_output):
if torch.allclose(triton_output, torch_output, atol=1e-2, rtol=0):
print("✅ Triton and Torch match")
else:
print("❌ Triton and Torch differ")