mirror of https://github.com/hpcaitech/ColossalAI
6ba8364881
* add alphafold benchmark * renae alphafold test * rename tests * rename diffuser * renme * rename * update transformer * update benchmark * update benchmark * update bench memory * update transformer benchmark * rename * support diffuser * support unet metainfo prop * fix bug and simplify code * update linear and support some op * optimize max region search, support conv * update unet test * support some op * support groupnorm and interpolate * update flow search * add fix dim in node flow * fix utils * rename * support diffusion * update diffuser * update chunk search * optimize imports * import * finish autochunk |
||
|---|---|---|
| .. | ||
| autochunk_codegen.py | ||
| estimate_memory.py | ||
| reorder_graph.py | ||
| search_chunk.py | ||
| select_chunk.py | ||
| trace_flow.py | ||
| trace_indice.py | ||
| utils.py | ||