digger yu 32f81f14d4 [NFC] fix typo colossalai/amp auto_parallel autochunk (#3756)

2023-05-19 13:50:00 +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

[NFC] fix typo colossalai/amp auto_parallel autochunk (#3756)

2023-05-19 13:50:00 +08:00

trace_indice.py

[NFC] fix typo colossalai/amp auto_parallel autochunk (#3756)

2023-05-19 13:50:00 +08:00

utils.py

[autochunk] support vit (#3084)

2023-03-10 10:23:26 +08:00