增加pir模式下op名称检测 (#66382)

PaddlePaddle · Jul 24, 2024 · e356e2e · e356e2e
1 parent 6885613
commit e356e2e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test/deprecated/legacy_test/test_nn_sigmoid_op.py b/test/deprecated/legacy_test/test_nn_sigmoid_op.py
@@ -50,7 +50,12 @@ def check_static_api(self, place):
         exe = paddle.static.Executor(place)
         out = exe.run(main_program, feed={'x': self.x}, fetch_list=[y])
         np.testing.assert_allclose(out[0], self.y, rtol=1e-05)
-        self.assertTrue(y.name.startswith("api_sigmoid"))
+
+        if paddle.framework.in_pir_mode():
+            y_name = y.get_defining_op().name()
+            self.assertTrue(y_name.startswith("pd_op.sigmoid"))
+        else:
+            self.assertTrue(y.name.startswith("api_sigmoid"))
 
     def check_dynamic_api(self, place):
         paddle.disable_static(place)