add test_with_pir_api => TestDiffOpFp16.test_fp16_with_gpu

PaddlePaddle · Nov 20, 2023 · d399662 · d399662
1 parent 546d936
commit d399662
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/legacy_test/test_diff_op.py b/test/legacy_test/test_diff_op.py
@@ -242,6 +242,7 @@ def set_args(self):
 
 
 class TestDiffOpFp16(TestDiffOp):
+    @test_with_pir_api
     def test_fp16_with_gpu(self):
         paddle.enable_static()
         if paddle.base.core.is_compiled_with_cuda():