diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py
index dec3f776a54d9b..853af843a8120a 100644
--- a/python/paddle/tensor/manipulation.py
+++ b/python/paddle/tensor/manipulation.py
@@ -1580,7 +1580,7 @@ def rot90(x, k=1, axes=[0, 1], name=None):
     """
 
     helper = LayerHelper("rot90", **locals())
-    check_type(x, 'X', (Variable, paddle.pir.OpResult), 'rot90')
+    check_type(x, 'X', (Variable, paddle.pir.Value), 'rot90')
     dtype = helper.input_dtype('x')
     check_dtype(
         dtype,
diff --git a/test/legacy_test/test_softmax2d.py b/test/legacy_test/test_softmax2d.py
index 4d57793744993f..59eca6214a788b 100644
--- a/test/legacy_test/test_softmax2d.py
+++ b/test/legacy_test/test_softmax2d.py
@@ -19,6 +19,7 @@
 
 import paddle
 from paddle.base import core
+from paddle.pir_utils import test_with_pir_api
 
 
 class TestSoftmax2DAPI(unittest.TestCase):
@@ -32,6 +33,7 @@ def setUp(self):
             else paddle.CPUPlace()
         )
 
+    @test_with_pir_api
     def test_static_api(self):
         paddle.enable_static()
         with paddle.static.program_guard(paddle.static.Program()):
@@ -108,6 +110,7 @@ def setUp(self):
             else paddle.CPUPlace()
         )
 
+    @test_with_pir_api
     def test_static_error(self):
         paddle.enable_static()
         with paddle.static.program_guard(paddle.static.Program()):