| .. |
|
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] [DeepNVMe] use same cpu_op_desc_t with cuda (#6645)
|
2024-10-22 14:45:05 +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
|
Add support for H100/sm_90 arch compilation (#6669)
|
2024-10-28 03:39:51 +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 |
229960a5e9