OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Remove aten.elu core ATen decomp because it is now core ATen (#149780)

Browse Source 
Per @larryliu0820.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/149780
Approved by: https://github.com/larryliu0820
This commit is contained in:
Scott Wolchok 2025-03-24 12:09:52 -07:00 committed by PyTorch MergeBot
parent 84684e9397
commit dc39e673e2
4 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/expect/HasDecompTest.test_aten_core_operators.expect
View File

@ -193,6 +193,8 @@ aten::div_.Scalar
aten::div_.Scalar_mode aten::div_.Scalar_mode
aten::div_.Tensor aten::div_.Tensor
aten::div_.Tensor_mode aten::div_.Tensor_mode
aten::elu
aten::elu.out
aten::embedding aten::embedding
aten::embedding.out aten::embedding.out
aten::empty_strided aten::empty_strided

1
test/export/test_export.py
View File

@ -13465,7 +13465,6 @@ class TestExportCustomClass(TorchTestCase):
) )
decomp_table = default_decompositions() decomp_table = default_decompositions()
del decomp_table[torch.ops.aten.elu.default]
ep = ep.run_decompositions( ep = ep.run_decompositions(
decomp_table=decomp_table, decomp_table=decomp_table,

1
torch/_decomp/__init__.py
View File

@ -332,7 +332,6 @@ def _core_aten_decompositions_post_autograd() -> dict[
aten.diagonal_copy, aten.diagonal_copy,
aten.dot, aten.dot,
aten.vdot, aten.vdot,
aten.elu,
aten.elu_, aten.elu_,
aten.elu_backward, aten.elu_backward,
aten._embedding_bag, aten._embedding_bag,

1
torch/_inductor/decomposition.py
View File

@ -61,6 +61,7 @@ inductor_decompositions = get_decompositions(
aten.bitwise_or_, aten.bitwise_or_,
aten.clamp_min_, aten.clamp_min_,
aten.dist, aten.dist,
aten.elu,
aten.empty_like, aten.empty_like,
aten.flip, aten.flip,
aten.gelu, aten.gelu,