Add Lowering for softshrink_backward

pytorch · Jul 19, 2023 · e9d4e13 · e9d4e13
1 parent 4448c78
commit e9d4e13
Show file tree

Hide file tree

Showing 2 changed files with 176 additions and 477 deletions.
diff --git a/test/inductor/test_torchinductor.py b/test/inductor/test_torchinductor.py
@@ -2014,6 +2014,16 @@ def fn(a):
         with self.assertRaisesRegex(RuntimeError, ""):
             fn(torch.randn(1, 5))
 
+    def test_softshrink_backward(self):
+        grad_output = torch.randn(1)
+        lambd = 0.5
+
+        def fn(a, grad_output, lambd):
+            a = a.cos()
+            return torch.ops.aten.softshrink_backward(grad_output, a, lambd)
+
+        self.common(fn, (torch.randn(1), grad_output, lambd))
+
     def test_inductor_assert(self):
         @torch._dynamo.optimize("inductor", dynamic=True)
         def fn(a):