ColossalAI/tests/test_booster/test_plugin
flybird11111 3e02154710
[gemini] gemini support extra-dp (#5043)
* support ddp

* fix

* fix

* fix

fix

* support ddp

* fix

* fix

* fix

fix

* simplify tests

* fix

* fix

* fix

fix

fix

* fix
2023-11-16 21:03:04 +08:00
..
test_3d_plugin.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_dp_plugin_base.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_gemini_plugin.py [gemini] gemini support extra-dp (#5043) 2023-11-16 21:03:04 +08:00
test_low_level_zero_plugin.py [gemini] support gradient accumulation (#4869) 2023-10-17 14:07:21 +08:00
test_torch_ddp_plugin.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_torch_fsdp_plugin.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00