3 people 4fb56914c5 [perf] Add fused MLA QKV + strided layernorm (#21116) ...

Signed-off-by: Mickael Seznec <mickael@mistral.ai>
Co-authored-by: mgoin <mgoin64@gmail.com>

2025-07-22 07:07:44 -07:00

..

attention

[Misc] Add sliding window to flashinfer test (#21282)

2025-07-21 08:37:49 -07:00

core

[perf] Add fused MLA QKV + strided layernorm (#21116)

2025-07-22 07:07:44 -07:00

mamba

[Kernel] Triton implementation of causal-conv1d for Mamba-based models (#18218)

2025-07-09 12:53:55 -07:00

moe

Revert "[Performance] Performance improvements in non-blockwise fp8 CUTLASS MoE (#20762) (#21334)

2025-07-21 21:49:01 -07:00

quantization

[Perf] Use Triton instead of Torch for DeepGEMM Per Token Group Quant (#20841)

2025-07-12 19:38:45 -07:00

__init__.py

[CI/Build] Move test_utils.py to tests/utils.py (#4425)

2024-05-13 23:50:09 +09:00

allclose_default.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

quant_utils.py

[Kernel] Enable fp8 support for pplx and BatchedTritonExperts. (#18864)

2025-07-03 14:55:40 -07:00

test_apply_repetition_penalties.py

[BUG] Fix #20484. Support empty sequence in cuda penalty kernel (#20491)

2025-07-05 19:38:02 -07:00

test_cutlass_mla_decode.py

[NVIDIA] Add Cutlass MLA backend (#17625)

2025-06-03 21:40:26 -07:00

test_flex_attention.py

[Misc] Add SPDX-FileCopyrightText (#20428)

2025-07-04 07:40:42 +00:00

test_fused_quant_activation.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_triton_flash_attention.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

utils.py

[Misc] Add unit tests for MoE ModularKernel combinations + Profiling utility (#20449)

2025-07-11 07:51:46 -07:00