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