ColossalAI

Commit Graph

Author	SHA1	Message	Date
Xuanlei Zhao	30dd13c450	[autochunk] support complete benchmark (#3121 ) * 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	2 years ago
Xuanlei Zhao	10c61de2f7	[autochunk] support vit (#3084 ) support vit for autochunk * support some new ops for vit * fix some bugs * add test for vit	2 years ago

Author

SHA1

Message

Date

Xuanlei Zhao

30dd13c450

[autochunk] support complete benchmark (#3121 )

* 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

Xuanlei Zhao

10c61de2f7

[autochunk] support vit (#3084 )

support vit for autochunk
* support some new ops for vit
* fix some bugs
* add test for vit

2 Commits (develop)