Michael Goin d2b0e97ea6

[CI Perf] Prune tests in tests/kernels/moe/ (#22939) ...

Signed-off-by: mgoin <mgoin64@gmail.com>

2025-08-14 21:33:42 -06:00

..

attention

[Misc] parametrize 'dtype' in test_flash_mla (#22641)

2025-08-12 16:31:48 -04:00

core

Revert "[Kernel] Add cuda kernel for gpt_oss activation" (#22948)

2025-08-14 17:38:10 -07:00

mamba

[Bugfix] Mamba2 SSD varlen bug fix initstates decay, improve test, assert chunk pwr 2 (#21783)

2025-08-12 02:04:37 -07:00

moe

[CI Perf] Prune tests in tests/kernels/moe/ (#22939)

2025-08-14 21:33:42 -06:00

quantization

[CI Perf] Prune tests in tests/kernels/quantization/ (#22942)

2025-08-14 21:25:34 -06: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

[Refactor] Remove Duplicate per_block_cast_to_fp8, Remove Dependencies of DeepGEMM (#21787)

2025-08-01 01:13:27 +00: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

Support encoder_only attention for FlexAttention (#22273)

2025-08-06 18:37:14 -07:00

test_fused_quant_activation.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

test_shuffle_rows.py

[Bugfix] Fix CUDA arch flags for MoE permute (#21426)

2025-07-24 03:23:59 -07:00

test_triton_flash_attention.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

utils.py

[Kernel] [Quantization] Add MXFP4 and bias support for marlin kernel (#22428)

2025-08-14 11:23:22 -07:00