Skip to content
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.

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

Already on GitHub? Sign in to your account

[MPS] Add support for aten::repeat_interleave.self_Tensor for MPS backend #89738

Open
chenxingqiang opened this issue Nov 28, 2022 · 1 comment
Labels
module: mps Related to Apple Metal Performance Shaders framework triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module

Comments

@chenxingqiang
Copy link

chenxingqiang commented Nov 28, 2022

馃殌 The feature, motivation and pitch

detectron2/modeling/poolers.py:66: UserWarning: The operator 'aten::repeat_interleave.self_Tensor' is not currently supported on the MPS backend and will fall back to run on the CPU. This may have performance implications. (Triggered internally at /Users/runner/work/_temp/anaconda/conda-bld/pytorch_1666646703877/work/aten/src/ATen/mps/MPSFallback.mm:11.)
indices = torch.repeat_interleave(

Alternatives

No response

Additional context

No response

cc @kulinseth @albanD @malfet @DenisVieriu97 @razarmehr @abhudev

@albanD albanD added triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module module: mps Related to Apple Metal Performance Shaders framework labels Nov 28, 2022
@Raman-Kumar
Copy link
Contributor

I have already contributed a PR to PyTorch to add support for a torch.median() op for MPS
should I try this one?

or

if any new contributors want to work on this I will guide them.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
module: mps Related to Apple Metal Performance Shaders framework triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module
Projects
None yet
Development

No branches or pull requests

3 participants