Skip to content

[Help wanted][aten] Fix the broken PT2 ATen operators #270

@xuzhao9

Description

@xuzhao9

We are working on adding auto-generated ATen operator benchmarks, the first step is to include 180 ATen operators from torchbench, timm, and huggingface.

Broken operators:

Operator Name Status
aten._cudnn_rnn.default partial
aten._cudnn_rnn_backward.default partial
aten._embedding_bag.default partial
aten._embedding_bag_per_sample_weights_backward.default partial
aten._fft_c2c.default partial
aten._log_softmax_backward_data.default partial
aten._softmax_backward_data.default partial
aten._sparse_coo_tensor_with_dims_and_tensors.default partial
aten._to_copy.default partial
aten.col2im_backward.default fail
aten.embedding.default partial
aten.embedding_dense_backward.default partial
aten.gather.default partial
aten.im2col_backward.default fail
aten.index_add.default partial
aten.index_add_.default partial
aten.index_select.default partial
aten.new_empty.default partial
aten.new_full.default partial
aten.new_ones.default partial
aten.nll_loss_backward.default partial
aten.nll_loss_forward.default partial
aten.scatter.src partial
aten.scatter_add.default partial
aten.scatter_add_.default partial
aten.sum.SymInt fail
aten.unsqueeze_.default partial
aten.upsample_bilinear2d_backward.vec fail
aten.upsample_nearest2d_backward.vec fail

In the testing, some of the aten operators will fail, with PT1 or PT2.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions