From 8c24bbcecb94c8857b92433b434d6a2e97b110f7 Mon Sep 17 00:00:00 2001 From: Lans1ot <47025645+Lans1ot@users.noreply.github.com> Date: Wed, 24 Jul 2024 17:57:54 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0pir=E6=A8=A1=E5=BC=8F?= =?UTF-8?q?=E4=B8=8Bop=E5=90=8D=E7=A7=B0=E6=A3=80=E6=B5=8B=20(#66382)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- test/deprecated/legacy_test/test_nn_sigmoid_op.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/test/deprecated/legacy_test/test_nn_sigmoid_op.py b/test/deprecated/legacy_test/test_nn_sigmoid_op.py index ed80a30a837bfc..ddc744544e0c06 100644 --- 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)