Function scan_macro

python/tvm/script/parser/core/entry.py:68–73

Generate the AST, and the source code for __repr__.

(program: Any | str, extra_vars: dict[str, Any] | None = None)

def scan_macro(program: Any | str, extra_vars: dict[str, Any] | None = None) -> Any:
    """Generate the AST, and the source code for __repr__."""
    # The AST will be converted into TIR at the time of expansion.
    source = Source(program)
    closure_vars = extra_vars or _default_globals()
    return source, closure_vars


def parse(