Method check_marker

test/tensorrt/tensorrt_test_base.py:343–357

(self, expected_result)

            )

    def check_marker(self, expected_result):
        paddle.framework.set_flags({"FLAGS_trt_min_group_size": 1})
        with paddle.pir_utils.IrGuard():
            main_program, startup_program, fetch_list = (
                self.create_fake_program()
            )
            main_program = run_pir_pass(
                main_program,
                disable_passes=self.disable_passes,
            )
            marker_result = False
            for op in main_program.global_block().ops:
                if op.name() == self.target_marker_op:
                    marker_result = op.attrs().get("__l_trt__", False)
            self.assertEqual(marker_result, expected_result)