Function get_func

python/tvm/tirx/script/parser/parser.py:677–687

()

    definition_depth = len(self.var_table.frames)

    def get_func():
        func_ast = from_doc(node)
        module_ast = ast.Module(body=[func_ast], type_ignores=[])
        ast.fix_missing_locations(module_ast)
        # set the filename to the source name, so that the error message can be reported correctly
        code_obj = compile(module_ast, filename=self.diag.source.source_name, mode="exec")
        namespace = self.var_table.get()
        exec(code_obj, namespace)  # pylint: disable=exec-used
        func_name = func_ast.name
        func = namespace[func_name]
        return func, func_name

    func, func_name = get_func()
    wrapper = inline(func, definition_depth=definition_depth, defining_var_table=self.var_table)