diff --git a/python/paddle/fluid/tests/unittests/xpu/test_kldiv_loss_op_xpu.py b/python/paddle/fluid/tests/unittests/xpu/test_kldiv_loss_op_xpu.py
index d7eb1c50ef94b5fc7bdddf2cbeca769725f3ef00..b3e3d7e5a058dd0bba1f71fd8d3e52a9595cbdfe 100644
--- a/python/paddle/fluid/tests/unittests/xpu/test_kldiv_loss_op_xpu.py
+++ b/python/paddle/fluid/tests/unittests/xpu/test_kldiv_loss_op_xpu.py
@@ -80,7 +80,7 @@ class XPUTestKLDivLossOp(XPUOpTestWrapper):
                 ['X'],
                 'Loss',
                 no_grad_set={"Target"},
-                check_eager=True,
+                check_dygraph=True,
             )
 
         def initTestCase(self):
diff --git a/python/paddle/fluid/tests/unittests/xpu/test_temporal_shift_op_xpu.py b/python/paddle/fluid/tests/unittests/xpu/test_temporal_shift_op_xpu.py
index 5af40a0c40f467684a6e4e431b9e262e80c85ef1..4a1967326504fe32317b41c3be5d5ba808ac2050 100644
--- a/python/paddle/fluid/tests/unittests/xpu/test_temporal_shift_op_xpu.py
+++ b/python/paddle/fluid/tests/unittests/xpu/test_temporal_shift_op_xpu.py
@@ -83,10 +83,10 @@ class XPUTestTemporalShiftOp(XPUOpTestWrapper):
             self.python_out_sig = ["Out"]
 
         def test_check_output(self):
-            self.check_output(check_dygraph=True)
+            self.check_output(check_dygraph=False)
 
         def test_check_grad(self):
-            self.check_grad(['X'], 'Out', check_dygraph=True)
+            self.check_grad(['X'], 'Out', check_dygraph=False)
 
         def initTestCase(self):
             self.x_shape = (6, 4, 4, 4)