Method __call__

python/tvm/script/parser/core/parser.py:142–170

(self, *args, **kwargs)

        raise RuntimeError(f"cannot find macro definition for {self.func.__name__}")

    def __call__(self, *args, **kwargs):
        param_binding = inspect.signature(self.func).bind(*args, **kwargs)
        param_binding.apply_defaults()
        local_vars = param_binding.arguments
        parser = self._find_parser_def()

        with parser.with_diag_source(self.source):
            if self.hygienic:
                saved_var_table = parser.var_table
                parser.var_table = VarTable()

                with parser.var_table.with_frame():
                    for k, v in self.closure_vars.items():
                        parser.var_table.add(k, v)
                    for k, v in local_vars.items():
                        parser.var_table.add(k, v)

                    parse_result = self.parse_macro(parser)

                parser.var_table = saved_var_table

            else:
                with parser.var_table.with_frame():
                    for k, v in local_vars.items():
                        parser.var_table.add(k, v)

                    parse_result = self.parse_macro(parser)

        return parse_result


class VarTableFrame: