diff --git a/test/expect/HasDecompTest.test_aten_core_operators.expect b/test/expect/HasDecompTest.test_aten_core_operators.expect index 60529dfcc63..374c9c2820f 100644 --- a/test/expect/HasDecompTest.test_aten_core_operators.expect +++ b/test/expect/HasDecompTest.test_aten_core_operators.expect @@ -193,6 +193,8 @@ aten::div_.Scalar aten::div_.Scalar_mode aten::div_.Tensor aten::div_.Tensor_mode +aten::elu +aten::elu.out aten::embedding aten::embedding.out aten::empty_strided diff --git a/test/export/test_export.py b/test/export/test_export.py index cd9956701d4..1533538740a 100755 --- a/test/export/test_export.py +++ b/test/export/test_export.py @@ -13465,7 +13465,6 @@ class TestExportCustomClass(TorchTestCase): ) decomp_table = default_decompositions() - del decomp_table[torch.ops.aten.elu.default] ep = ep.run_decompositions( decomp_table=decomp_table, diff --git a/torch/_decomp/__init__.py b/torch/_decomp/__init__.py index f147f304d70..abb94b109cc 100644 --- a/torch/_decomp/__init__.py +++ b/torch/_decomp/__init__.py @@ -332,7 +332,6 @@ def _core_aten_decompositions_post_autograd() -> dict[ aten.diagonal_copy, aten.dot, aten.vdot, - aten.elu, aten.elu_, aten.elu_backward, aten._embedding_bag, diff --git a/torch/_inductor/decomposition.py b/torch/_inductor/decomposition.py index 74af73d47da..ddf044ebf1f 100644 --- a/torch/_inductor/decomposition.py +++ b/torch/_inductor/decomposition.py @@ -61,6 +61,7 @@ inductor_decompositions = get_decompositions( aten.bitwise_or_, aten.clamp_min_, aten.dist, + aten.elu, aten.empty_like, aten.flip, aten.gelu,