ColossalAI/colossalai/autochunk
oahzxl 7ab2db206f adapt new fx 2023-01-10 11:56:00 +08:00
..
autochunk_codegen.py adapt new fx 2023-01-10 11:56:00 +08:00
estimate_memory.py adapt new fx 2023-01-10 11:56:00 +08:00
reorder_graph.py add doc 2023-01-10 09:59:47 +08:00
search_chunk.py update doc 2023-01-10 10:44:07 +08:00
select_chunk.py rename trace_index to trace_indice 2023-01-09 17:25:13 +08:00
trace_flow.py rename function from index to indice 2023-01-09 17:34:30 +08:00
trace_indice.py add doc for trace indice 2023-01-09 17:59:52 +08:00
utils.py take apart chunk code gen 2023-01-06 14:14:45 +08:00