DeepSpeed
d9225000 - ZeRO-3: build the partial-offload device string from the public accelerator API (#8994)

Commit
5 days ago
ZeRO-3: build the partial-offload device string from the public accelerator API (#8994) `get_accelerator()._name` is the accelerator internal backend name, which maybe not the torch device name like in KLXPU, and can fail test `tests/unit/v1/zero/test_zero.py::TestZero3OffloadPinnedCopyBack::test[0.5]`. `device_name()` is a better choice here. The PR has been validated on KLXPU. Signed-off-by: guhangsong <guhangsong@outlook.com>
Author
Parents
Loading