Function _handle

exir/memory_planning.py:1305–1318

(
        submodule_node: torch.fx.Node,
        alloc_graph_input: bool = False,
    )

    bufsizes: list[int] = []

    def _handle(
        submodule_node: torch.fx.Node,
        alloc_graph_input: bool = False,
    ) -> None:
        current_bufsizes = _handle_submodule(
            algo,
            graph_module,
            alignment,
            submodule_node,
            graph_signature,
            alloc_graph_input=alloc_graph_input,
        )
        nonlocal bufsizes
        _merge_bufsizes(bufsizes, current_bufsizes)

    for cond_node in get_cond_nodes(graph_module):
        _handle(cast(torch.fx.Node, cond_node.args[1]))