pytorch
b794fd19 - [MPS] Add scatter gather kernels (support up to 5 dimensions) (#94663)

Commit
1 year ago
[MPS] Add scatter gather kernels (support up to 5 dimensions) (#94663) Add scatter gather kernels (support up to 5 dimensions) - Fixes int64 issues for `mH`, `mT`, `T`, `H` on Monterey Pull Request resolved: https://github.com/pytorch/pytorch/pull/94663 Approved by: https://github.com/kulinseth
Author
Committer
Parents
Loading