[NPU]Add ZeRO-Infinity feature for NPU#4809
Conversation
|
Hi @tjruwase , thanks for your help launching CI and I have fixed the formatting failure issue. But the failure of |
|
Hi @tjruwase , would you please help me launch unit-tests again? Thank you. |
|
@misstek, can you please fix the formatting issues? https://github.com/microsoft/DeepSpeed/blob/master/CONTRIBUTING.md#contributing |
Hi, @tjruwase , the formatting failure has been fixed according to |
|
Hi @mrwyattii @jomayeri , since this pull request has passed all tests, could you please review it and add it to merge queue. Thank you. |
|
Hi @tjruwase , I noticed that this PR has been approved. Is it ready to be put into the merge queue? Thank you. |
Add ZeRO-Infinity feature for NPU devices. I add a new `async_io.py` in `op_builder/npu` and compilation preprocessing judgment in `deepspeed_aio_thread.cpp` specifically for NPU, which will be isolated from other devices such as the GPU and will not affect each other. See what we have already done in deepspeedai#4567 . --------- Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
Add ZeRO-Infinity feature for NPU devices.
I add a new
async_io.pyinop_builder/npuand compilation preprocessing judgment indeepspeed_aio_thread.cppspecifically for NPU, which will be isolated from other devices such as the GPU and will not affect each other.See what we have already done in #4567 .