[PIR] Open test_case ut (#60721)

* fix * fix
PaddlePaddle · Jan 11, 2024 · 55558f1 · 55558f1
1 parent 57fff3a
commit 55558f1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/legacy_test/test_case.py b/test/legacy_test/test_case.py
@@ -169,14 +169,14 @@ def fn_3():
             np.testing.assert_allclose(res[4], 2, rtol=1e-05)
             self.assertEqual(res[4].shape, ())
 
-    # Todo(zhangbo): grad_list can not find dx in oir mode
-    # @test_with_pir_api
+    @test_with_pir_api
     def test_0d_tensor_backward(self):
         main_program = paddle.static.Program()
         startup_program = paddle.static.Program()
         with paddle.static.program_guard(main_program, startup_program):
             x = paddle.full(shape=[], dtype='float32', fill_value=-2.0)
             x.stop_gradient = False
+            x.persistable = True
             pred = paddle.full(shape=[], dtype='bool', fill_value=0)
             # pred is False, so out = -x
             out = paddle.static.nn.case(