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

2 lines
116 B
Python
Raw Normal View History

[Gemini] add an inline_op_module to common test models and polish unitests. (#2004)
2022-11-23 08:55:54 +00:00
from . import bert, gpt, inline_op_model, nested_model, no_leaf_module, repeated_computed_layer, resnet, simple_net