mirror of https://github.com/hpcaitech/ColossalAI
![]() * 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 |
||
---|---|---|
.. | ||
autochunk_codegen.py | ||
estimate_memory.py | ||
reorder_graph.py | ||
search_chunk.py | ||
select_chunk.py | ||
trace_flow.py | ||
trace_indice.py | ||
utils.py |