microsoft
/
DeepSpeed
зеркало из https://github.com/microsoft/DeepSpeed.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность
DeepSpeed/op_builder
История
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
..
cpu AIO CPU Locked Tensor (#6592) 2024-10-09 21:07:31 +00:00
hpu Rearrange inference OPS and stop using builder.load (#5490) 2024-10-09 01:22:28 +00:00
mlu Fix torch include in `op_builder/mlu/fused_adam.py` and update no-torch workflow triggers (#6584) 2024-09-27 13:32:48 -07:00
npu Op_builder->is_compatible quite warning (#6093) 2024-09-05 17:03:34 +00:00
xpu [XPU] Support DeepNVMe new code structure (#6532) 2024-09-26 20:39:59 +00:00
__init__.py add type checker ignore to resolve that pylance can't resolved noqa annotation (#4102) 2023-08-08 13:30:23 +00:00
all_ops.py Make op builder detection adapt to accelerator change (#5206) 2024-03-12 20:48:29 +00:00
async_io.py Safe usage of popen (#6490) 2024-09-04 21:06:04 +00:00
builder.py AIO CPU Locked Tensor (#6592) 2024-10-09 21:07:31 +00:00
cpu_adagrad.py CPUAdam fp16 and bf16 support (#5409) 2024-05-20 12:50:20 +00:00
cpu_adam.py CPUAdam fp16 and bf16 support (#5409) 2024-05-20 12:50:20 +00:00
cpu_lion.py CPUAdam fp16 and bf16 support (#5409) 2024-05-20 12:50:20 +00:00
evoformer_attn.py add pip install cutlass version check (#6393) 2024-08-22 17:18:17 +00:00
fp_quantizer.py Allow triton==3.0.x for fp_quantizer (#6447) 2024-08-28 18:04:40 +00:00
fused_adam.py Update DeepSpeed copyright license to Apache 2.0 (#3111) 2023-03-30 17:14:38 -07:00
fused_lamb.py Update DeepSpeed copyright license to Apache 2.0 (#3111) 2023-03-30 17:14:38 -07:00
fused_lion.py feat: add Lion optimizer (#4331) 2023-10-05 22:32:14 +00:00
gds.py Handle an edge case where `CUDA_HOME` is not defined on ROCm systems (#6488) 2024-09-04 22:28:13 +00:00
inference_core_ops.py Op_builder->is_compatible quite warning (#6093) 2024-09-05 17:03:34 +00:00
inference_cutlass_builder.py Op_builder->is_compatible quite warning (#6093) 2024-09-05 17:03:34 +00:00
quantizer.py Enable quantizer op on ROCm (#4114) 2024-01-10 23:48:54 +00:00
ragged_ops.py Op_builder->is_compatible quite warning (#6093) 2024-09-05 17:03:34 +00:00
ragged_utils.py Op_builder->is_compatible quite warning (#6093) 2024-09-05 17:03:34 +00:00
random_ltd.py ROCm 6.0 prep changes (#4537) 2023-10-20 19:05:54 +00:00
sparse_attn.py Op_builder->is_compatible quite warning (#6093) 2024-09-05 17:03:34 +00:00
spatial_inference.py Op_builder->is_compatible quite warning (#6093) 2024-09-05 17:03:34 +00:00
stochastic_transformer.py Update DeepSpeed copyright license to Apache 2.0 (#3111) 2023-03-30 17:14:38 -07:00
transformer.py ROCm 6.0 prep changes (#4537) 2023-10-20 19:05:54 +00:00
transformer_inference.py Op_builder->is_compatible quite warning (#6093) 2024-09-05 17:03:34 +00:00