DeepSpeed/op_builder/cpu
Joe Mayer a1f98bdc70
AIO CPU Locked Tensor (#6592)
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>
2024-10-09 21:07:31 +00:00
..
__init__.py AIO CPU Locked Tensor (#6592) 2024-10-09 21:07:31 +00:00
async_io.py AIO CPU Locked Tensor (#6592) 2024-10-09 21:07:31 +00:00
builder.py CPUAdam fp16 and bf16 support (#5409) 2024-05-20 12:50:20 +00:00
comm.py Op_builder->is_compatible quite warning (#6093) 2024-09-05 17:03:34 +00:00
cpu_adam.py [CPU] FusedAdam and CPU training support (#3991) 2023-07-25 13:32:02 +00:00
fused_adam.py [CPU] FusedAdam and CPU training support (#3991) 2023-07-25 13:32:02 +00:00
no_impl.py [CPU] Support Intel CPU inference (#3041) 2023-05-16 11:59:22 -04:00