Function write

imperative/python/megengine/xla/lower.py:120–127

(var_ids, hlo_nodes)

    # update env with var_ids and ir.Values
    def write(var_ids, hlo_nodes):
        assert isinstance(var_ids, (list, tuple))
        assert isinstance(hlo_nodes, (map, list, tuple))
        hlo_nodes = list(hlo_nodes)
        assert len(var_ids) == len(hlo_nodes), (len(var_ids), len(hlo_nodes))
        for vid, node in zip(var_ids, hlo_nodes):
            assert vid not in env
            env[vid] = node

    assert len(args) == len(trace_result.inputs)
    assert len(consts) == len(trace_result.consts)