DeepSpeed icon indicating copy to clipboard operation
DeepSpeed copied to clipboard

[REQUEST] Enable both CPU and NVMe for optimizer

Open shanhx2000 opened this issue 9 months ago • 0 comments

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 avatar May 08 '24 17:05 shanhx2000