MPS operator coverage tracking issue (2.6+ version) #141287
Description
🐛 Describe the bug
This issue is to have a centralized place to list and track work on adding support to new ops for the MPS backend.
Most requested ops (extracted from comments to this issue and #77764)
PyTorch MPS Ops Project : Project to track all the ops for MPS backend. There are a very large number of operators in pytorch and so they are not all yet implemented. We will be prioritizing adding new operators based on user feedback. If possible, please also provide link to the network or use-case where this op is getting used.
As Ops are requested we will add " To Triage" pool. If we have 3+ requests for an operation and given its complexity/need the operation will be moved "To be implemented" pool. If you want to work on adding support for such op, feel free to comment below to get assigned one. Please avoid pickup up an op that is already being worked on tracked in "In progress" pool.
Link to the wiki for details on how to add these ops and example PRs.
MPS operators coverage matrix - The matrix covers most of the supported operators but is not exhaustive. Please look at the In vx.x.x column, if the box is green, it means that the op implementation is included in the latest release; on the other hand, if the box is yellow, it means the op implementation is in the nightly and has not yet included in the latest release. Before you comment below, please take a look at this matrix to make sure the operator you're requesting has not been implemented in nightly. More details can be found on the readme.
This is a spiritual successor of #77764, but hopefully it will not receive requests for the OPs that has been added already since 1.13 but until 2.6
| Op Name | Times requested | Added by PR | Availability |
|---|---|---|---|
aten::angle |
2 | #143449 | Nightly |
aten::_upsample_bilinear2d_aa.out |
4 | #145581 | Nightly |
aten::linalg_cholesky_ex.L |
3 | #146799 | Nightly |
aten::_linalg_det.result |
2 | #146279 | Nightly |
aten::_linalg_solve_ex.result |
4 | #146531 | Nightly |
aten::_upsample_bicubic2d_aa.out |
7 | #149378 | Nightly |