Skip to content

Commit 5fa5e4c

Browse files
authored
Update test_quant_api.py (#2469)
1 parent 09f0d6c commit 5fa5e4c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/quantization/test_quant_api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -307,7 +307,7 @@ def api(model):
307307
res = m2(*example_inputs)
308308

309309
# TODO: figure out why ROCm has a larger error
310-
atol, rtol = (1e-2, 1e-2) if torch.version.hip else (1e-7, 1e-5)
310+
atol, rtol = (1e-2, 1e-2) if torch.version.hip else (None, None)
311311
torch.testing.assert_close(ref, res.cpu(), atol=atol, rtol=rtol)
312312

313313
@unittest.skipIf(

0 commit comments

Comments
 (0)