mirror of https://github.com/hpcaitech/ColossalAI
152239bbfa
* [gemini] fix nvme optimizer init * [gemini] gemini supports lazy init * [gemini] add init example * [gemini] add fool model * [zero] update gemini ddp * [zero] update init example * add chunk method * add chunk method * [lazyinit] fix lazy tensor tolist * [gemini] fix buffer materialization * [misc] remove useless file * [booster] update gemini plugin * [test] update gemini plugin test * [test] fix gemini plugin test * [gemini] fix import * [gemini] fix import * [lazyinit] use new metatensor * [lazyinit] use new metatensor * [lazyinit] fix __set__ method |
||
|---|---|---|
| .. | ||
| components_to_test | ||
| kit | ||
| test_amp | ||
| test_analyzer | ||
| test_auto_parallel | ||
| test_autochunk | ||
| test_booster | ||
| test_checkpoint_io | ||
| test_cluster | ||
| test_comm | ||
| test_config | ||
| test_context | ||
| test_data | ||
| test_data_pipeline_tensor_parallel | ||
| test_ddp | ||
| test_device | ||
| test_engine | ||
| test_fx | ||
| test_layers | ||
| test_moe | ||
| test_ops | ||
| test_optimizer | ||
| test_pipeline | ||
| test_tensor | ||
| test_trainer | ||
| test_utils | ||
| test_zero | ||
| __init__.py | ||