BUG: add torch.repeat #305

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Merged

ev-br merged 1 commit into data-apis:main from ev-br:torch_repeat

Apr 15, 2025

Member

ev-br commented Apr 15, 2025

closes gh-292

Array API compatible repeat function is spelled repeat_interleave in pytorch (https://pytorch.org/docs/stable/generated/torch.repeat_interleave.html#torch.repeat_interleave)

test_repeat will have to remain xfailed because of repeat_interleave_cpu is not implemented for short integers pytorch/pytorch#151311.
that said, I ran locally

$ ARRAY_API_TESTS_SKIP_DTYPES=uint8,uint16,uint32,uint64,int8,int16 ARRAY_API_TESTS_MODULE=array_api_compat.torch pytest array_api_tests/test_manipulation_functions.py::test_repeat --max-examples=10_000

and it passed.


          BUG: add torch.repeat

00e7cce

ev-br force-pushed the torch_repeat branch from c631fa8 to 00e7cce Compare

April 15, 2025 10:50

ev-br added this to the 1.12 milestone

crusaderky approved these changes

View reviewed changes

ev-br merged commit 16978e6 into data-apis:main

23 checks passed

Member Author

ev-br commented Apr 15, 2025

Merged as approved, thanks Guido.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet