Skip to content

Most requested ops for the MPS backend #154052

@manuelcandales

Description

@manuelcandales

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
isin 224 75 148 .Tensor_Scalar_out, .Tensor_Tensor_out
index_copy 57 15 42 .out
_upsample_bicubic2d_aa 49 20 29 .out
max_pool3d_with_indices 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
masked_scatter_ 25 11 14
_linalg_eigh 24 11 13 .eigenvalues
native_dropout 23 13 10
_embedding_bag 22 16 6
__rshift__ 21 13 8 .Scalar, .Tensor
linalg_matrix_exp 18 9 9
max_unpool2d 18 5 13
nanmedian 16 9 7 .dim_values
_linalg_eigvals 13 5 8
unique_dim 12 6 6
avg_pool3d 10 4 6 .out
linalg_householder_product 8 6 2
index_fill_ 8 5 3 .int_Scalar
kthvalue 8 5 3 .values
multilabel_margin_loss_forward 5 2 3
linalg_lu 5 2 3 .out
upsample_nearest3d 5 1 4 .vec
__lshift__ 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
upsample_trilinear3d 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
avg_pool3d_backward 3 1 2 .grad_input
cholesky_solve 3 1 2 .out
cummax 3 1 2 .out
cummin 3 1 2 .out
grid_sampler_3d_backward 3 1 2
hardshrink 3 1 2 .out
igamma 3 1 2 .out
igammac 3 1 2 .out
log_normal_ 3 1 2
max_unpool3d 3 1 2
mvlgamma 3 1 2 .out
segment_reduce 3 1 2

cc @kulinseth @albanD @malfet @DenisVieriu97 @jhavukainen

Metadata

Metadata

Assignees

No one assigned

    Labels

    featureA request for a proper, new feature.module: mpsRelated to Apple Metal Performance Shaders frameworktrackerA tracking issuetriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions