ColossalAI/colossalai/autochunk
oahzxl 19cc64b1d3 remove autochunk_available 2023-01-09 16:06:58 +08:00
..
autochunk_codegen.py remove autochunk_available 2023-01-09 16:06:58 +08:00
estimate_memory.py fix typo 2023-01-09 13:46:17 +08:00
reorder_graph.py code style 2023-01-06 17:31:59 +08:00
search_chunk.py fix typo 2023-01-09 13:46:17 +08:00
select_chunk.py fix typo 2023-01-09 13:46:17 +08:00
trace_flow.py code style 2023-01-06 17:31:59 +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