DeepSpeed
e9d5d416 - deepcompile: Record graph order using OrderedDict (#7563)

Commit
101 days ago
deepcompile: Record graph order using OrderedDict (#7563) On clear, GraphOrder does not clears ordered_frames. That may confuses subsequent passes after the first iteration. Use an OrderedDict to record the mapping from frame IDs to other graph-related information. Also fix the type annotation of graph_order which is a list of (int , bool) tuples instead of a list of int. Signed-off-by: Junjie Mao <junjie.mao@linux.alibaba.com>
Author
Parents
Loading