Class ASTPrinter

tests/python/tirx-transform/test_tir_functor.py:65–78

Print tirx AST in structured format. The shape of Node is ignored.

@tirx.functor.visitor
class ASTPrinter(PyStmtExprVisitor):
    """Print tirx AST in structured format. The shape of Node is ignored."""

    def __init__(self) -> None:
        super().__init__()
        self.log = ASTLog()

    def visit_var_(self, op: Var) -> None:
        self.log.add("Stmt: Var")
        super().visit_var_(op)

    def visit_add_(self, op: Add) -> None:
        self.log.add("Stmt: Add")
        super().visit_add_(op)


@tirx.functor.visitor