-
Notifications
You must be signed in to change notification settings - Fork 568
Fix test_scan_xla_patched_linear on TPU CI #8747
Copy link
Copy link
Closed
Description
FAIL: test_scan_xla_patched_linear (__main__.ScanSpmdTest)
When we use scan to trace `XLAPatchedLinear` layers, the lowered HLO should
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/runner/_work/xla/xla/pytorch/xla/test/scan/test_scan_spmd.py", line 118, in test_scan_xla_patched_linear
torch.testing.assert_close(y_cpu, y_xla.cpu())
File "/home/runner/.local/lib/python3.10/site-packages/torch/testing/_comparison.py", line 1519, in assert_close
raise error_metas[0].to_error(msg)
AssertionError: Tensor-likes are not close!
Mismatched elements: 9 / 7680 (0.1%)
Greatest absolute difference: 0.00018310546875 at index (1, 0, 4, 101) (up to 1e-05 allowed)
Greatest relative difference: 0.265625 at index (1, 0, 4, 65) (up to 0.016 allowed)
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels