ColossalAI/colossalai/autochunk
oahzxl ecccc91f21
[autochunk] support autochunk on evoformer (#2497)
2023-01-19 11:41:00 +08:00
..
autochunk_codegen.py [autochunk] support autochunk on evoformer (#2497) 2023-01-19 11:41:00 +08:00
estimate_memory.py [autochunk] support autochunk on evoformer (#2497) 2023-01-19 11:41:00 +08:00
reorder_graph.py add doc 2023-01-10 09:59:47 +08:00
search_chunk.py [autochunk] support autochunk on evoformer (#2497) 2023-01-19 11:41:00 +08:00
select_chunk.py rename trace_index to trace_indice 2023-01-09 17:25:13 +08:00
trace_flow.py [autochunk] support autochunk on evoformer (#2497) 2023-01-19 11:41:00 +08:00
trace_indice.py [autochunk] support autochunk on evoformer (#2497) 2023-01-19 11:41:00 +08:00
utils.py [autochunk] support autochunk on evoformer (#2497) 2023-01-19 11:41:00 +08:00