first fix the UT

PaddlePaddle · Nov 15, 2023 · 29eaaa1 · 29eaaa1
1 parent 341afb7
commit 29eaaa1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/test/legacy_test/test_zero_dim_tensor.py b/test/legacy_test/test_zero_dim_tensor.py
@@ -830,6 +830,7 @@ def test_setitem(self):
         np.testing.assert_allclose(out[1, 2, 3, 4], np.array(10))
         self.assertEqual(x.grad.shape, [2, 3, 4, 5])
         x_grad_expected = np.ones((2, 3, 4, 5)) * 2
+        x_grad_expected[1, 2, 3, 4] = 0
         np.testing.assert_allclose(x.grad, x_grad_expected)
 
         # case2: 0-D Tensor indice in some axis
@@ -847,6 +848,7 @@ def test_setitem(self):
         self.assertEqual(out.shape, x.shape)
         np.testing.assert_allclose(out[1, 1], np.ones((4, 5)) * 0.5)
         x_grad_expected = np.ones((2, 3, 4, 5))
+        x_grad_expected[1, 1] = 0
         np.testing.assert_allclose(x.grad, x_grad_expected)
 
         # case3：0-D Tensor indice in some axis, value is a Tensor

diff --git a/test/xpu/test_zero_dim_tensor_xpu.py b/test/xpu/test_zero_dim_tensor_xpu.py
@@ -441,6 +441,7 @@ def test_setitem(self):
         np.testing.assert_allclose(out[1, 2, 3, 4], np.array(10))
         self.assertEqual(x.grad.shape, [2, 3, 4, 5])
         x_grad_expected = np.ones((2, 3, 4, 5)) * 2
+        x_grad_expected[1, 2, 3, 4] = 0
         np.testing.assert_allclose(x.grad, x_grad_expected)
 
         # case2: 0-D Tensor indice in some axis