Skip to content

BUG: add torch.repeat #305

New issue

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

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

Already on GitHub? # to your account

Merged
merged 1 commit into from
Apr 15, 2025
Merged

BUG: add torch.repeat #305

merged 1 commit into from
Apr 15, 2025

Conversation

ev-br
Copy link
Member

@ev-br 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)

$ 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.

@ev-br ev-br added this to the 1.12 milestone Apr 15, 2025
@ev-br ev-br merged commit 16978e6 into data-apis:main Apr 15, 2025
23 checks passed
@ev-br
Copy link
Member Author

ev-br commented Apr 15, 2025

Merged as approved, thanks Guido.

# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

AttributeError: module 'array_api_compat.torch' has no attribute 'repeat'
2 participants