fix ci coverage

PaddlePaddle · Jul 24, 2023 · 0920620 · 0920620
1 parent 5c50cfe
commit 0920620
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/test/ir/new_ir/test_special_op_translator.py b/test/ir/new_ir/test_special_op_translator.py
@@ -159,5 +159,19 @@ def test_normal_attribute(self):
         _ = paddle.fluid.core.translate_newirprogram(main_program.desc)
 
 
+class TestArgmaxOpTranscriber(unittest.TestCase):
+    def test_op(self):
+        place = core.Place()
+        place.set_place(paddle.CPUPlace())
+        new_scope = paddle.static.Scope()
+        main_program = paddle.static.Program()
+        with paddle.static.scope_guard(new_scope):
+            with paddle.static.program_guard(main_program):
+                x = paddle.to_tensor([2, 3, 4], 'float64')
+                y = paddle.argmax(x)
+
+        _ = paddle.fluid.core.translate_newirprogram(main_program.desc)
+
+
 if __name__ == "__main__":
     unittest.main()