ColossalAI/colossalai/autochunk
oahzxl a6cdbf9161 seperate trace flow 2023-01-06 17:24:23 +08:00
..
autochunk_codegen.py seperate trace flow 2023-01-06 17:24:23 +08:00
estiamte_memory.py rename 2023-01-06 17:09:37 +08:00
reorder_graph.py ad reorder graph 2023-01-06 17:13:18 +08:00
search_chunk.py seperate trace flow 2023-01-06 17:24:23 +08:00
select_chunk.py seperate trace flow 2023-01-06 17:24:23 +08:00
trace_flow.py seperate trace flow 2023-01-06 17:24:23 +08:00
trace_index.py seperate trace flow 2023-01-06 17:24:23 +08:00
utils.py take apart chunk code gen 2023-01-06 14:14:45 +08:00