-
Notifications
You must be signed in to change notification settings - Fork 24.9k
Open
Labels
featureA request for a proper, new feature.A request for a proper, new feature.module: mpsRelated to Apple Metal Performance Shaders frameworkRelated to Apple Metal Performance Shaders frameworktrackerA tracking issueA tracking issuetriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Description
This issue will list the most requested ops for the MPS backend, that haven't been implemented yet, taken from the comments to #77764 and #141287.
The script that produced this data is provided in this gist. The number of votes is computed as the number of unique users requesting the ops plus the number of thumbs up that their comments received.
op | votes | users | thumbs_up | suffixes |
---|---|---|---|---|
224 | 75 | 148 | .Tensor_Scalar_out, .Tensor_Tensor_out | |
57 | 15 | 42 | .out | |
49 | 20 | 29 | .out | |
48 | 23 | 25 | ||
grid_sampler_3d | 47 | 29 | 18 | |
linalg_eig | 38 | 9 | 29 | .out |
grid_sampler_2d_backward | 35 | 18 | 17 | |
linalg_qr | 33 | 23 | 10 | .out |
25 | 11 | 14 | ||
_linalg_eigh | 24 | 11 | 13 | .eigenvalues |
native_dropout | 23 | 13 | 10 | |
_embedding_bag | 22 | 16 | 6 | |
21 | 13 | 8 | .Scalar, .Tensor | |
linalg_matrix_exp | 18 | 9 | 9 | |
18 | 5 | 13 | ||
nanmedian | 16 | 9 | 7 | .dim_values |
_linalg_eigvals | 13 | 5 | 8 | |
unique_dim | 12 | 6 | 6 | |
10 | 4 | 6 | .out | |
linalg_householder_product | 8 | 6 | 2 | |
8 | 5 | 3 | .int_Scalar | |
kthvalue | 8 | 5 | 3 | .values |
multilabel_margin_loss_forward | 5 | 2 | 3 | |
linalg_lu | 5 | 2 | 3 | .out |
5 | 1 | 4 | .vec | |
4 | 4 | 0 | .Scalar | |
_embedding_bag_forward_only | 4 | 2 | 2 | |
logcumsumexp | 4 | 2 | 2 | |
cholesky_inverse | 4 | 2 | 2 | .out |
_cdist_backward | 3 | 3 | 0 | |
_amp_foreach_non_finite_check_and_unscale_ | 3 | 3 | 0 | |
3 | 3 | 0 | .out | |
_embedding_bag_dense_backward | 3 | 1 | 2 | |
_embedding_bag_per_sample_weights_backward | 3 | 1 | 2 | |
_segment_reduce_backward | 3 | 1 | 2 | |
3 | 1 | 2 | .grad_input | |
cholesky_solve | 3 | 1 | 2 | .out |
3 | 1 | 2 | .out | |
3 | 1 | 2 | .out | |
grid_sampler_3d_backward | 3 | 1 | 2 | |
3 | 1 | 2 | .out | |
igamma | 3 | 1 | 2 | .out |
igammac | 3 | 1 | 2 | .out |
log_normal_ | 3 | 1 | 2 | |
3 | 1 | 2 | ||
mvlgamma | 3 | 1 | 2 | .out |
segment_reduce | 3 | 1 | 2 |
andre-brainn, hvaara, digantdesai, semenko and dego1985hvaara, andre-brainn, exdysa and dego1985andre-brainnandre-brainn
Metadata
Metadata
Assignees
Labels
featureA request for a proper, new feature.A request for a proper, new feature.module: mpsRelated to Apple Metal Performance Shaders frameworkRelated to Apple Metal Performance Shaders frameworktrackerA tracking issueA tracking issuetriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module