You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_booster/test_plugin
digger yu 9c2feb2f0b
fix some typo with colossalai/device colossalai/tensor/ etc. (#4171)
1 year ago
..
test_3d_plugin.py [plugin] add 3d parallel plugin (#4295) 1 year ago
test_dp_plugin_base.py [CI] fix typo with tests/ etc. (#3727) 2 years ago
test_gemini_plugin.py [shardformer]fix gpt2 double head (#4663) 1 year ago
test_low_level_zero_plugin.py [zero] refactor low level zero for shard evenly (#4030) 1 year ago
test_torch_ddp_plugin.py fix some typo with colossalai/device colossalai/tensor/ etc. (#4171) 1 year ago
test_torch_fsdp_plugin.py fix some typo with colossalai/device colossalai/tensor/ etc. (#4171) 1 year ago