mirror of https://github.com/hpcaitech/ColossalAI
c2fdc6a011
* derive compute pattern from dist spec * polish code |
||
---|---|---|
.. | ||
components_to_test | ||
test_amp | ||
test_comm | ||
test_config | ||
test_context | ||
test_data | ||
test_data_pipeline_tensor_parallel | ||
test_engine | ||
test_gemini | ||
test_layers | ||
test_moe | ||
test_optimizer | ||
test_tensor | ||
test_trainer | ||
test_utils | ||
test_zero | ||
__init__.py |