Function visit_while

python/tvm/tirx/script/parser/parser.py:299–313

The while visiting method for tirx. Parameters ---------- self : Parser The visiting parser. node : doc.While The doc AST while node.

(self: Parser, node: doc.While)

@dispatch.register(token="tirx", type_name="While")
def visit_while(self: Parser, node: doc.While) -> None:
    """The while visiting method for tirx.

    Parameters
    ----------
    self : Parser
        The visiting parser.

    node : doc.While
        The doc AST while node.
    """
    with self.var_table.with_frame():
        cond = self.eval_expr(node.test)
        with T.While(cond):
            self.visit_body(node.body)


@dispatch.register(token="tirx", type_name="Break")