github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-16 07:28:45 +00:00
Code Issues Packages Projects Releases Wiki Activity
2,563 Commits 60 Branches 51 Tags
caa4433072aab219418205bf4d338fd1aa355d42
Commit Graph

4 Commits

Author SHA1 Message Date
Hongxin Liu
50793b35f4 [gemini] accelerate inference (#3641) 
* [gemini] support don't scatter after inference

* [chat] update colossalai strategy

* [chat] fix opt benchmark

* [chat] update opt benchmark

* [gemini] optimize inference

* [test] add gemini inference test

* [chat] fix unit test ci

* [chat] fix ci

* [chat] fix ci

* [chat] skip checkpoint test
 2023-04-26 16:32:40 +08:00
HELSON
17a3c685b0 [zero] fix unit-tests (#2039) 2022-11-30 10:40:31 +08:00
Jiarui Fang
28aa9a4294 [Gemini] more rigorous unit tests for run_fwd_bwd (#2034) 2022-11-29 09:26:06 +08:00
Jiarui Fang
2e9cbfca12 [Gemini] add unitests to check gemini correctness (#2015) 2022-11-24 16:51:45 +08:00