Add empty/empty_like to core aten decomps (pytorch#105158)

Fixes pytorch#104871 Pull Request resolved: pytorch#105158 Approved by: https://github.com/SherlockNoMad
harborn · Jul 15, 2023 · 5cd861f · 5cd861f
1 parent 1152e86
commit 5cd861f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/test/expect/HasDecompTest.test_aten_core_operators.expect b/test/expect/HasDecompTest.test_aten_core_operators.expect
@@ -185,8 +185,6 @@ aten::div_.Tensor
 aten::div_.Tensor_mode
 aten::embedding
 aten::embedding.out
-aten::empty_like
-aten::empty_like.out
 aten::empty_strided
 aten::empty_strided.out
 aten::eq.Scalar

diff --git a/torch/_decomp/__init__.py b/torch/_decomp/__init__.py
@@ -211,6 +211,7 @@ def core_aten_decompositions() -> Dict[OpOverload, Callable]:
             aten.elu_backward,
             aten._embedding_bag,
             aten.embedding_dense_backward,
+            aten.empty_like,
             aten._euclidean_dist.default,
             aten.expand_as,
             aten.eye,