search

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.46k stars 467 forks source link

[hard] Op info test for `repeat_interleave` #7997

Open ManfeiBai opened 2 weeks ago

ManfeiBai commented 2 weeks ago

Fix the Op info test for repeat_interleave

  1. Find the lines of repeat_interleave in test_ops.py and remove repeat_interleave from skip_list
  2. Run op_info test with pytest test/test_ops.py
  3. Fix the failure.

Please refer to this guide as guide to fix:

Also refer to these PRs:

ManfeiBai commented 2 weeks ago

failed to gain repeats and dim arg in @op(torch.ops.aten. repeat_interleave.Tensor)