[MPS] Support 64-bit indexing for pooling operations
- Dominant language
- Python
- Stars
- 103k
- Forks
- 29.5k
- PR merge metrics
- PR metrics pending
Description
### 🚀 The feature, motivation and pitch
MPS pooling operations do not support tensors requiring 64-bit indexing. [PR #194460](https://github.com/pytorch/pytorch/pull/194460) adds `NotImplementedError` guards for the affected forward and backward paths. This issue is to track adding full 64-bit indexing support on MPS.
### Alternatives
_No response_
### Additional context
_No response_
cc @mikaylagawarecki @kulinseth @malfet @DenisVieriu97 @jhavukainen @aditvenk
Contributor guide
Research direction
Start by reviewing PR #194460 and the NotImplementedError guards it adds for the affected MPS pooling forward and backward paths. Trace how those paths handle tensors requiring 64-bit indexing, then verify that both forward and backward pooling support those tensors without the guards.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Active
- Clarity
- Mostly clear
- Newbie friendliness
- 48/100