mirror of https://github.com/hpcaitech/ColossalAI
30dd13c450
* 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 * add unet test * add bench * update bench * update bench * init * support vit * move to cpu * add cpu benchmark |
||
---|---|---|
.. | ||
benchmark_autochunk_diffuser.py | ||
test_autochunk_diffuser_utils.py | ||
test_autochunk_unet.py |