Skip to content

Commit ac430ab

Browse files
authored
Increase tolerance in test_scan_xla_patched_linear (#8749)
1 parent 0797642 commit ac430ab

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/scan/test_scan_spmd.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ def forward(self, x: torch.Tensor):
115115
y_xla = model(x)
116116

117117
torch_xla.sync()
118-
torch.testing.assert_close(y_cpu, y_xla.cpu())
118+
torch.testing.assert_close(y_cpu, y_xla.cpu(), atol=1e-3, rtol=1e-3)
119119

120120
def check_dots_in_model(self, model, x, expect_pattern):
121121
# Trace the model to get the HLO.

0 commit comments

Comments
 (0)