-
Notifications
You must be signed in to change notification settings - Fork 51
Open
Labels
Description
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.