flybird11111 46e091651b [shardformer] hybridparallelplugin support gradients accumulation. (#5246) ...

* support gradients acc

fix

fix

fix

fix

fix

fix

fix

fix

fix

fix

fix

fix

fix

* fix

fix

* fix

fix

fix

2024-01-17 15:22:33 +08:00

..

test_3d_plugin.py

[shardformer] hybridparallelplugin support gradients accumulation. (#5246)

2024-01-17 15:22:33 +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

[workflow] fixed oom tests (#5275)

2024-01-16 18:55:13 +08:00

test_low_level_zero_plugin.py

[workflow] fixed oom tests (#5275)

2024-01-16 18:55:13 +08:00

test_torch_ddp_plugin.py

[workflow] fixed oom tests (#5275)

2024-01-16 18:55:13 +08:00

test_torch_fsdp_plugin.py

[workflow] fixed oom tests (#5275)

2024-01-16 18:55:13 +08:00