[Decomposition] std.correction (#108733)

Summary: Include decomp in core_aten_decompositions Decomp: https://www.internalfb.com/code/fbsource/[e69bf00ff87a55c9a30bd7905881661ff05fa211]/fbcode/caffe2/torch/_refs/__init__.py?lines=2398 Differential Revision: D48940402 Pull Request resolved: #108733 Approved by: https://github.com/larryliu0820, https://github.com/SherlockNoMad
pytorch · Sep 18, 2023 · af93b29 · af93b29
1 parent a683bc5
commit af93b29
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/test/expect/HasDecompTest.test_aten_core_operators.expect b/test/expect/HasDecompTest.test_aten_core_operators.expect
@@ -486,7 +486,6 @@ aten::sqrt_
 aten::squeeze
 aten::squeeze.dim
 aten::squeeze.dims
-aten::std.correction
 aten::std.correction_out
 aten::std_mean.correction
 aten::sub.Scalar

diff --git a/torch/_decomp/__init__.py b/torch/_decomp/__init__.py
@@ -356,6 +356,7 @@ def core_aten_decompositions() -> Dict[torch._ops.OperatorBase, Callable]:
             aten.special_entr,
             aten.special_log_ndtr,
             aten.special_xlog1py,
+            aten.std.correction,
             aten.stack,
             aten.t,
             aten.tanh_backward,