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

Remove wrong dependency between TritonGPU and NVGPU dialect (#2276)

Browse Source
This commit is contained in:
Thomas Raoux
2023-09-11 16:30:13 -07:00
committed by GitHub
parent ec4a968d44
commit a9db6b94b9
7 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/test/unit/operators/test_cross_entropy.py
View File

@@ -36,5 +36,7 @@ def test_op(M, N, dtype, mode):
x.grad = None
th_y.backward(dy)
th_dx = x.grad.clone()
torch.testing.assert_close(th_dx, tt_dx)
if dtype == 'float16':
torch.testing.assert_close(th_dx, tt_dx, rtol=0.001, atol=0.001)
else:
torch.testing.assert_close(th_dx, tt_dx)