ColossalAI/tests/test_pipeline/test_schedule
flybird11111 17062c83b9
[hotfix] fix hybrid checkpointio for sp+dp (#6184)
* Update hybrid_parallel_plugin.py

* Update hybrid_parallel_plugin.py

* Update hybrid_parallel_plugin.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update build_on_pr.yml

* Update test_zerobubble_pp.py

* fix

* fix

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* fix

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2025-02-06 17:21:04 +08:00
..
test_interleaved.py [Feature] Zigzag Ring attention (#5905) 2024-08-16 13:56:38 +08:00
test_oneF_oneB.py [Feature] Zigzag Ring attention (#5905) 2024-08-16 13:56:38 +08:00
test_pipeline_schedule_utils.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_zerobubble_pp.py [hotfix] fix hybrid checkpointio for sp+dp (#6184) 2025-02-06 17:21:04 +08:00