ColossalAI/colossalai/autochunk
oahzxl 72341e65f4
[auto-chunk] support extramsa (#3) (#2504)
2023-01-20 10:13:03 +08:00
..
autochunk_codegen.py [autochunk] support autochunk on evoformer (#2497) 2023-01-19 11:41:00 +08:00
estimate_memory.py [auto-chunk] support extramsa (#3) (#2504) 2023-01-20 10:13:03 +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 [auto-chunk] support extramsa (#3) (#2504) 2023-01-20 10:13:03 +08:00
trace_indice.py [auto-chunk] support extramsa (#3) (#2504) 2023-01-20 10:13:03 +08:00
utils.py [auto-chunk] support extramsa (#3) (#2504) 2023-01-20 10:13:03 +08:00