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

[TESTING] triton.testing.allclose now uses torch.allclose (#1333)

Browse Source
This commit is contained in:
Philippe Tillet
2023-03-13 17:48:32 -07:00
committed by GitHub
parent 6539395337
commit dde34904d0
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/triton/testing.py
View File

@@ -106,10 +106,8 @@ def allclose(x, y, atol=0, rtol=1e-2):
return torch.sum(x ^ y) == 0
if x.dtype in [torch.int8, torch.int16, torch.int32, torch.int64]:
rtol = 0
diff = abs(x - y)
x_max = torch.max(x)
y_max = torch.max(y)
return torch.max(diff) <= atol + rtol * torch.max(x_max, y_max)
atol = 0
return torch.allclose(x, y, rtol=rtol, atol=atol)
def nvsmi(attrs):