| .. |
|
cpu
|
Op_builder->is_compatible quite warning (#6093)
|
2024-09-05 17:03:34 +00:00 |
|
hpu
|
CPUAdam fp16 and bf16 support (#5409)
|
2024-05-20 12:50:20 +00:00 |
|
npu
|
Op_builder->is_compatible quite warning (#6093)
|
2024-09-05 17:03:34 +00:00 |
|
xpu
|
Op_builder->is_compatible quite warning (#6093)
|
2024-09-05 17:03:34 +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
|
Avoid security issues of subprocess shell (#6498)
|
2024-09-11 20:07:06 +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
|
…
|
|
|
fused_lamb.py
|
…
|
|
|
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
|
…
|
|
|
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 |
659f6be105