This website requires JavaScript.
Explore
关于
Help
Register
Sign In
github
/
ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI
Watch
1
Star
0
Fork
You've already forked ColossalAI
0
Code
Issues
Projects
Releases
Wiki
Activity
d8ceeac14e
ColossalAI
/
tests
/
components_to_test
/
utils
/
__init__.py
3 lines
96 B
Python
Raw
Normal View
History
Unescape
Escape
[Gemini] add unitests to check gemini correctness (#2015)
2022-11-24 08:51:45 +00:00
from
.
dummy_data_generator
import
DummyDataGenerator
[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 08:32:40 +00:00
from
.
executor
import
run_fwd
,
run_fwd_bwd