fix cpu test

pytorch · Jul 20, 2023 · d635b44 · d635b44
1 parent cf7cafa
commit d635b44
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/inductor/test_torchinductor.py b/test/inductor/test_torchinductor.py
@@ -2014,7 +2014,6 @@ 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
@@ -2023,7 +2022,10 @@ 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),)
+        self.common(
+            fn,
+            (torch.randn(10), grad_output, lambd),
+        )
 
     def test_inductor_assert(self):
         @torch._dynamo.optimize("inductor", dynamic=True)