[Decomposition] clamp_min (#108717)

Summary: bypass-github-pytorch-ci-checks Decomp already exists so just add it to core_aten_decompositions https://www.internalfb.com/code/fbsource/[abda43a5a268e83fef6d62b49531a390ce915ad2]/fbcode/caffe2/torch/_refs/__init__.py?lines=1846 Test Plan: Phabricator + OSS Tests Reviewed By: SS-JIA Differential Revision: D48880080
pytorch · Sep 17, 2023 · ed05c11 · ed05c11
1 parent 75b954b
commit ed05c11
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/test/expect/HasDecompTest.test_aten_core_operators.expect b/test/expect/HasDecompTest.test_aten_core_operators.expect
@@ -139,10 +139,6 @@ aten::clamp_max.Tensor_out
 aten::clamp_max.out
 aten::clamp_max_
 aten::clamp_max_.Tensor
-aten::clamp_min
-aten::clamp_min.Tensor
-aten::clamp_min.Tensor_out
-aten::clamp_min.out
 aten::clamp_min_
 aten::clamp_min_.Tensor
 aten::clone

diff --git a/torch/_decomp/__init__.py b/torch/_decomp/__init__.py
@@ -227,6 +227,7 @@ def core_aten_decompositions() -> Dict[torch._ops.OperatorBase, Callable]:
             aten.binary_cross_entropy_with_logits,
             aten.celu,
             aten.celu_,
+            aten.clamp_min,
             aten.col2im,
             aten.count_nonzero,
             aten.cudnn_batch_norm,

diff --git a/torch/_inductor/decomposition.py b/torch/_inductor/decomposition.py
@@ -71,6 +71,7 @@
 decomps_to_exclude = [
     aten._unsafe_index,
     aten._scaled_dot_product_flash_attention.default,  # See comments in torch/_decomp/decompositions.py
+    aten.clamp_min,
 ]
 
 remove_decompositions(decompositions, decomps_to_exclude)