github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/components_to_test/utils/__init__.py

3 lines
96 B
Python
Raw Normal View History

[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