actually add the core decomp

pytorch · Apr 27, 2023 · 784b1a0 · 784b1a0
1 parent 3c95f13
commit 784b1a0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torch/_decomp/__init__.py b/torch/_decomp/__init__.py
@@ -184,6 +184,8 @@ def core_aten_decompositions() -> Dict[OpOverload, Callable]:
             aten.addcmul_,
             aten.addr,
             aten.aminmax,
+            aten.arange.default,
+            aten.arange.start,
             aten.avg_pool2d_backward,
             aten.binary_cross_entropy,
             aten.binary_cross_entropy_backward,