diff --git a/test/expect/HasDecompTest.test_aten_core_operators.expect b/test/expect/HasDecompTest.test_aten_core_operators.expect index 53c5cf70d248c..06010b0b1d081 100644 --- a/test/expect/HasDecompTest.test_aten_core_operators.expect +++ b/test/expect/HasDecompTest.test_aten_core_operators.expect @@ -517,7 +517,6 @@ aten::unfold aten::uniform aten::uniform.out aten::uniform_ -aten::unsafe_split.Tensor aten::unsafe_split_with_sizes aten::unsafe_split_with_sizes.out aten::unsqueeze diff --git a/torch/_decomp/__init__.py b/torch/_decomp/__init__.py index 247f8eabd8e14..7d4d8a5df97ef 100644 --- a/torch/_decomp/__init__.py +++ b/torch/_decomp/__init__.py @@ -371,6 +371,7 @@ def core_aten_decompositions() -> Dict[torch._ops.OperatorBase, Callable]: aten.unfold_backward, aten.unfold_copy, aten._unsafe_index, + aten.unsafe_split.Tensor, aten.upsample_bilinear2d, aten.upsample_nearest2d_backward, aten.view_as_complex,