Skip to content

Fix test_scan_xla_patched_linear on TPU CI #8747

@tengyifei

Description

@tengyifei
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)

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions