diff --git a/test/legacy_test/test_polar.py b/test/legacy_test/test_polar.py index 5336fe38037a4f..46136b4e2b7624 100644 --- a/test/legacy_test/test_polar.py +++ b/test/legacy_test/test_polar.py @@ -19,6 +19,7 @@ import paddle from paddle.base import core +from paddle.pir_utils import test_with_pir_api np.random.seed(10) @@ -37,6 +38,7 @@ def setUp(self): if core.is_compiled_with_cuda(): self.place.append(paddle.CUDAPlace(0)) + @test_with_pir_api def test_api_static(self): paddle.enable_static()