Jee Jee Li 7555d6b34a

[Bugfix] Fix test_mixtral_moe (#24371)

2025-09-06 09:32:03 -07:00

..

attention

[Attention] FlashAttn MLA (#14258)

2025-09-04 02:47:59 -07:00

core

[Kernel] Add cuda kernel for gpt_oss activation (#22951)

2025-08-17 05:03:24 +00:00

mamba

[Mamba] - refactor: Renamed mamba_attn to mamba2_attn (#22818)

2025-08-15 06:38:05 +00:00

moe

[Bugfix] Fix test_mixtral_moe (#24371)

2025-09-06 09:32:03 -07:00

quantization

[Bugfix][Misc] Fix silu_and_mul_nvfp4_quant issue and extract common utils for nvfp4 kernel source files (#23727)

2025-09-04 14:25: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

[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

[Attention] Blackwell FP8 MLA support with CUTLASS_MLA backend (#23289)

2025-09-03 14:05:24 -04:00

test_flex_attention.py

Updates to Flex + VLLm integration (#21416)

2025-08-25 09:32:42 -04:00

test_fused_quant_activation.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

test_onednn.py

[CPU] Refactor CPU unquantized linear (#24150)

2025-09-04 14:28:45 +08: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

[Doc]: fix typos in Python comments (#24093)

2025-09-02 21:05:45 -07:00