From 55558f10a7972d7724ad39fd5ed12d8ba6cc299d Mon Sep 17 00:00:00 2001
From: zhangbo9674 <82555433+zhangbo9674@users.noreply.github.com>
Date: Thu, 11 Jan 2024 10:08:45 +0800
Subject: [PATCH] [PIR] Open test_case ut (#60721)

* fix

* fix
---
 test/legacy_test/test_case.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/legacy_test/test_case.py b/test/legacy_test/test_case.py
index d1f67f1cd70cda..1a5cf3e459e6b9 100644
--- 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(