Function read

imperative/python/megengine/xla/lower.py:111–117

(var_ids)

    # read ir.Values from env according to var_ids
    def read(var_ids):
        assert isinstance(var_ids, (list, tuple))
        ret = []
        for vid in var_ids:
            assert isinstance(vid, int)
            ret.append(env[vid])
        return ret

    # update env with var_ids and ir.Values
    def write(var_ids, hlo_nodes):