ColossalAI

Commit Graph

Author	SHA1	Message	Date
Hongxin Liu	079bf3cb26	[misc] update pre-commit and run all files (#4752 ) * [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format	2023-09-19 14:20:26 +08:00
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	2023-03-13 17:42:37 +08:00
Xuanlei Zhao	10c61de2f7	[autochunk] support vit (#3084 ) support vit for autochunk * support some new ops for vit * fix some bugs * add test for vit	2023-03-10 10:23:26 +08:00

Author

SHA1

Message

Date

Hongxin Liu

079bf3cb26

[misc] update pre-commit and run all files (#4752 )

* [misc] update pre-commit

* [misc] run pre-commit

* [misc] remove useless configuration files

* [misc] ignore cuda for clang-format

2023-09-19 14:20:26 +08:00

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

2023-03-13 17:42:37 +08:00

Xuanlei Zhao

10c61de2f7

[autochunk] support vit (#3084 )

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

2023-03-10 10:23:26 +08:00

3 Commits (df5e9c53cf23d44656470cc319ee0b470c40712f)