Skip to content

Commit 45e9600

Browse files
authored
Relax tolerance for test_input_float16_acc_float16_dynamic_shape (#399)
1 parent 19f23b2 commit 45e9600

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_dot.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ def run_kernel():
150150
torch.testing.assert_close(result, expected, atol=5e-3, rtol=0.5)
151151
elif input_dtype == torch.float16 and acc_dtype == torch.float16:
152152
# Use higher tolerance when accumulator is float16 due to precision limits
153-
torch.testing.assert_close(result, expected, atol=5e-3, rtol=0.5)
153+
torch.testing.assert_close(result, expected, atol=1e-2, rtol=0.5)
154154
elif input_dtype == torch.float32:
155155
# Use higher tolerance for TF32 mode
156156
torch.testing.assert_close(result, expected, atol=1e-1, rtol=1e-1)

0 commit comments

Comments
 (0)