add slope == 0 case into standard leaky relu nn test

ghstack-source-id: d7b20a65968c320da2d4de1e76d1578bf06d4eea
Pull Request resolved: #37559

test/test_autograd.py

@@ -5624,7 +5624,7 @@ def test_ctc_loss_cudnn(self, device):
         self.assertEqual(grad_cudnn, grad_native, atol=1e-4)
 
     @skipCUDAIfRocm
-    def test_leaky_relu_inplace_with_zero_or_neg_slope(self, device):
+    def test_leaky_relu_inplace_with_neg_slope(self, device):
         a = torch.tensor([-1., 1.], device=device, requires_grad=True)
         b = torch.nn.functional.leaky_relu_(a.clone(), -2)
         with self.assertRaisesRegex(RuntimeError, "call out-of-place version"):
@@ -5635,10 +5635,13 @@ def test_leaky_relu_inplace_with_zero_or_neg_slope(self, device):
         with self.assertRaisesRegex(RuntimeError, "call out-of-place version"):
             b.backward(torch.ones(2, device=device))
 
+    @skipCUDAIfRocm
+    def test_leaky_relu_inplace_with_zero_slope(self, device):
         a = torch.tensor([-2., 0., 2.], device=device, requires_grad=True)
         b = torch.nn.functional.leaky_relu_(a.clone(), 0.0)
         b.backward(torch.ones(3, device=device))
-        self.assertEqual(a.grad, torch.tensor([0., 0., 1.], device=device))
+        expected = torch.tensor([0., 0., 1.], device=device)
+        self.assertEqual(a.grad, expected)
 
     @onlyCUDA
     def test_free_unneeded_tensor(self, device):

torch/testing/_internal/common_nn.py

@@ -225,6 +225,14 @@ def get_weight(m):
         check_inplace=True,
         desc='with_negval'
     ),
+    dict(
+        module_name='LeakyReLU',
+        constructor_args=(0.0,),
+        cpp_constructor_args='torch::nn::LeakyReLUOptions().negative_slope(0.0)',
+        input_fn=lambda: torch.randn(10, 10),
+        check_inplace=True,
+        desc='with_zero_negval'
+    ),
     dict(
         module_name='LogSigmoid',
         input_size=(2, 3, 4),