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.

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

[REQUEST] Enable both CPU and NVMe for optimizer #5508

Open
shanhx2000 opened this issue May 8, 2024 · 0 comments
Open

[REQUEST] Enable both CPU and NVMe for optimizer #5508

shanhx2000 opened this issue May 8, 2024 · 0 comments
Labels
enhancement New feature or request

Comments

@shanhx2000
Copy link

Describe the solution you'd like
Dear developers, I am trying to test the optimizer offloading function with deepspeed. I notice that when nvme offloading is enabled, all optimizers will go to NVMe, which misses the chances to put some in CPU. Although there is a configurable variable "ratio" in "offload_optimizer", it will result in some optimizers on GPU and some on CPU, leading an error when specifying "device" as "nvme".

An example expectation is:
0.05 optimizer states on GPU
0.55 optimizer states on CPU
0.4 optimizer states on NVMe

@shanhx2000 shanhx2000 added the enhancement New feature or request label May 8, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant