DeepSpeed/accelerator
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 [NPU] Add NPU support for unit test (#4569) 2023-11-13 20:36:12 +00:00
abstract_accelerator.py assumption of torch.initial_seed function accepting seed arg in DeepSpeedAccelerator abstract class is incorrect (#5569) 2024-06-12 09:32:17 -07:00
cpu_accelerator.py AIO CPU Locked Tensor (#6592) 2024-10-09 21:07:31 +00:00
cuda_accelerator.py Fixed Windows inference build. (#5609) 2024-06-24 13:39:18 -07:00
hpu_accelerator.py HPU: add required ENV vars to acccelerator init (#6495) 2024-09-05 15:43:08 +00:00
mlu_accelerator.py [Accelerator] Cambricon MLU support (#6472) 2024-09-26 13:10:52 +00:00
mps_accelerator.py assumption of torch.initial_seed function accepting seed arg in DeepSpeedAccelerator abstract class is incorrect (#5569) 2024-06-12 09:32:17 -07:00
npu_accelerator.py assumption of torch.initial_seed function accepting seed arg in DeepSpeedAccelerator abstract class is incorrect (#5569) 2024-06-12 09:32:17 -07:00
real_accelerator.py [Accelerator] Cambricon MLU support (#6472) 2024-09-26 13:10:52 +00:00
xpu_accelerator.py [XPU]Use host time to replace xpu time when IPEX version slower than 2.5. (#5796) 2024-07-25 15:03:07 -07:00