ColossalAI/colossalai/autochunk
digger yu 7f8203af69
fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)
2023-05-24 09:01:50 +08:00
..
autochunk_codegen.py [NFC] fix typo applications/ and colossalai/ (#3735) 2023-05-15 11:46:25 +08:00
estimate_memory.py [doc] Fix typo under colossalai and doc(#3618) 2023-04-26 11:38:43 +08:00
reorder_graph.py [autochunk] support multi outputs chunk search (#2538) 2023-02-01 13:18:51 +08:00
search_chunk.py [doc] Fix typo under colossalai and doc(#3618) 2023-04-26 11:38:43 +08:00
select_chunk.py [autochunk] refactor chunk memory estimation (#2762) 2023-03-08 16:22:30 +08:00
trace_flow.py fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808) 2023-05-24 09:01:50 +08:00
trace_indice.py fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808) 2023-05-24 09:01:50 +08:00
utils.py [autochunk] support vit (#3084) 2023-03-10 10:23:26 +08:00