зеркало из https://github.com/microsoft/DeepSpeed.git
a1f98bdc70
Restoring the functionality of the cpu locked tensor in the AIO library. Make async_io operator available for CPU accelerator, i.e., CPU only environment. --------- Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> |
||
---|---|---|
.. | ||
__init__.py | ||
async_io.py | ||
builder.py | ||
comm.py | ||
cpu_adam.py | ||
fused_adam.py | ||
no_impl.py |