Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Why is CPU Checkpointing only available with partitioned activations? #541

Closed
sshleifer opened this issue Nov 19, 2020 · 1 comment
Closed

Why is CPU Checkpointing only available with partitioned activations? #541

sshleifer opened this issue Nov 19, 2020 · 1 comment

Comments

@sshleifer
Copy link

sshleifer commented Nov 19, 2020 •
edited
Loading

I am trying to better understand offloading activations to CPU (PA_TO_CPU) here https://github.com/microsoft/DeepSpeed/blob/master/deepspeed/runtime/activation_checkpointing/checkpointing.py#L730 and wondering why

"CPU Checkpointing/Contiguous Checkpointing is only available with partitioned activations."

It seems to me like one could be implemented without the other? Or am I missing something.
This is not an issue with the code, just a question to firm up my understanding.
Thanks in advance!

Tagging @jeffra based on git-blame.
@ghosthamlet ghosthamlet mentioned this issue Dec 1, 2020
Closed
@jomayeri
Copy link
Contributor

jomayeri commented Sep 9, 2024

Please try the latest version of DeepSpeed.

@jomayeri jomayeri closed this as completed Sep 9, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@sshleifer @jomayeri