.. |
cpu
|
a1f98bdc70
AIO CPU Locked Tensor (#6592)
|
1 week ago |
hpu
|
645639bcf8
Rearrange inference OPS and stop using builder.load (#5490)
|
1 week ago |
mlu
|
8cded575a9
Fix torch include in `op_builder/mlu/fused_adam.py` and update no-torch workflow triggers (#6584)
|
3 weeks ago |
npu
|
4f803852ac
Op_builder->is_compatible quite warning (#6093)
|
1 month ago |
xpu
|
d45cfd3455
[XPU] Support DeepNVMe new code structure (#6532)
|
3 weeks ago |
__init__.py
|
57a27b0803
add type checker ignore to resolve that pylance can't resolved noqa annotation (#4102)
|
1 year ago |
all_ops.py
|
c08e69f212
Make op builder detection adapt to accelerator change (#5206)
|
7 months ago |
async_io.py
|
662a421b05
Safe usage of popen (#6490)
|
1 month ago |
builder.py
|
a1f98bdc70
AIO CPU Locked Tensor (#6592)
|
1 week ago |
cpu_adagrad.py
|
69af361167
CPUAdam fp16 and bf16 support (#5409)
|
5 months ago |
cpu_adam.py
|
69af361167
CPUAdam fp16 and bf16 support (#5409)
|
5 months ago |
cpu_lion.py
|
69af361167
CPUAdam fp16 and bf16 support (#5409)
|
5 months ago |
evoformer_attn.py
|
51da191eb5
add pip install cutlass version check (#6393)
|
2 months ago |
fp_quantizer.py
|
4864991f53
Allow triton==3.0.x for fp_quantizer (#6447)
|
1 month ago |
fused_adam.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
fused_lamb.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
fused_lion.py
|
8e64c3b550
feat: add Lion optimizer (#4331)
|
1 year ago |
gds.py
|
10ba3dde84
Handle an edge case where `CUDA_HOME` is not defined on ROCm systems (#6488)
|
1 month ago |
inference_core_ops.py
|
4f803852ac
Op_builder->is_compatible quite warning (#6093)
|
1 month ago |
inference_cutlass_builder.py
|
4f803852ac
Op_builder->is_compatible quite warning (#6093)
|
1 month ago |
quantizer.py
|
f2d4845f07
Enable quantizer op on ROCm (#4114)
|
9 months ago |
ragged_ops.py
|
4f803852ac
Op_builder->is_compatible quite warning (#6093)
|
1 month ago |
ragged_utils.py
|
4f803852ac
Op_builder->is_compatible quite warning (#6093)
|
1 month ago |
random_ltd.py
|
e238351101
ROCm 6.0 prep changes (#4537)
|
1 year ago |
sparse_attn.py
|
4f803852ac
Op_builder->is_compatible quite warning (#6093)
|
1 month ago |
spatial_inference.py
|
4f803852ac
Op_builder->is_compatible quite warning (#6093)
|
1 month ago |
stochastic_transformer.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
transformer.py
|
e238351101
ROCm 6.0 prep changes (#4537)
|
1 year ago |
transformer_inference.py
|
4f803852ac
Op_builder->is_compatible quite warning (#6093)
|
1 month ago |