Skip to content

[ckpt] Add async ckpt api#6136

Merged
ver217 merged 12 commits intohpcaitech:feature/async-checkpointiofrom
wangbluo:async_ckpt
Nov 15, 2024
Merged

[ckpt] Add async ckpt api#6136
ver217 merged 12 commits intohpcaitech:feature/async-checkpointiofrom
wangbluo:async_ckpt

Conversation

@wangbluo
Copy link
Copy Markdown
Contributor

📝 What does this PR do?

Add async ckpt api for all the plugin, like save_model, save_optimizer.

@wangbluo wangbluo requested a review from a team as a code owner November 14, 2024 08:07
Comment thread colossalai/checkpoint_io/utils.py Outdated
Comment thread tests/test_checkpoint_io/test_low_level_zero_checkpoint_io.py Outdated
Comment thread tests/test_checkpoint_io/test_low_level_zero_checkpoint_io.py Outdated
@ver217 ver217 merged commit cb00f1f into hpcaitech:feature/async-checkpointio Nov 15, 2024
@wangbluo wangbluo deleted the async_ckpt branch November 18, 2024 07:08
flybird11111 pushed a commit to flybird11111/ColossalAI that referenced this pull request Nov 18, 2024
* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix
flybird11111 pushed a commit to flybird11111/ColossalAI that referenced this pull request Nov 18, 2024
* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix
ver217 pushed a commit that referenced this pull request Nov 19, 2024
* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix
flybird11111 pushed a commit to flybird11111/ColossalAI that referenced this pull request Nov 19, 2024
* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix
flybird11111 pushed a commit to flybird11111/ColossalAI that referenced this pull request Nov 20, 2024
* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants