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
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