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 |
||
---|---|---|
.. | ||
_C | ||
amp | ||
auto_parallel | ||
autochunk | ||
builder | ||
cli | ||
communication | ||
context | ||
device | ||
engine | ||
fx | ||
gemini | ||
kernel | ||
logging | ||
nn | ||
pipeline | ||
registry | ||
tensor | ||
testing | ||
trainer | ||
utils | ||
zero | ||
__init__.py | ||
constants.py | ||
core.py | ||
global_variables.py | ||
initialize.py |