DeepSpeed/accelerator
Yizhou Wang 11bbf45af5
[XPU] host timer check version from Torch 2.5 to Torch 2.6 (#6633)
Elapsed time would be supported in Torch 2.6.

Co-authored-by: Masahiro Tanaka <81312776+tohtana@users.noreply.github.com>
2024-10-22 06:53:15 +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] host timer check version from Torch 2.5 to Torch 2.6 (#6633) 2024-10-22 06:53:15 +00:00