ColossalAI/colossalai/autochunk
Xuanlei Zhao 2ca9728cbb
[autochunk] refactor chunk memory estimation (#2762)
* refact memory code

* dont log free var memory

* add memory align

* update chunk target

* update setting for new memory

* finish test

* update tracer

* update typo

* update test
2023-03-08 16:22:30 +08:00
..
autochunk_codegen.py [autochunk] refactor chunk memory estimation (#2762) 2023-03-08 16:22:30 +08:00
estimate_memory.py [autochunk] refactor chunk memory estimation (#2762) 2023-03-08 16:22:30 +08:00
reorder_graph.py [autochunk] support multi outputs chunk search (#2538) 2023-02-01 13:18:51 +08:00
search_chunk.py [autochunk] refactor chunk memory estimation (#2762) 2023-03-08 16:22:30 +08:00
select_chunk.py [autochunk] refactor chunk memory estimation (#2762) 2023-03-08 16:22:30 +08:00
trace_flow.py [autochunk] support diffusion for autochunk (#2621) 2023-02-07 16:32:45 +08:00
trace_indice.py [autochunk] refactor chunk memory estimation (#2762) 2023-03-08 16:22:30 +08:00
utils.py [autochunk] refactor chunk memory estimation (#2762) 2023-03-08 16:22:30 +08:00