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