rasmith 7618dc973d

[CI/Build] Make test_mha_attn.py run on correct platform only and check for flash_attn_varlen_func in layer.py (#29145)

2025-12-09 20:18:17 +00:00

..

attention

[CI/Build] Make test_mha_attn.py run on correct platform only and check for flash_attn_varlen_func in layer.py (#29145)

2025-12-09 20:18:17 +00:00

core

[Performance] Fused blockwise quant RMS norm (#27883)

2025-12-07 16:38:04 +00:00

mamba

Add SpecDec support to selective_state_update (#29488)

2025-12-08 16:45:18 -05:00

moe

[Kernel][MoE] optimize moe_align_block_size (#29642)

2025-12-07 01:58:47 -08:00

quantization

[Kernel]Support W4A8 Grouped GEMM on Hopper (#29691)

2025-12-08 19:29:06 -08:00

__init__.py

…

allclose_default.py

…

quant_utils.py

[CI/Test] Fix FP8 per-tensor quant test reference scale shape (#30352)

2025-12-09 12:52:20 -06:00

test_apply_repetition_penalties.py

…

test_cache_kernels.py

[Bugfix][cache_kernels]: Fix OOB in cache_kernels.cu (#28760)

2025-11-20 02:52:02 -08:00

test_fla_layernorm_guard.py

…

test_flex_attention.py

…

test_fused_quant_activation.py

…

test_onednn.py

…

test_shuffle_rows.py

…

test_top_k_per_row.py

[DeepSeek v3.2] Make top-k work for any logit values. (#27568)

2025-12-08 06:55:58 -08:00

utils.py

[Feat] Support non-gated activations in NVFP4 modelopt path (#29004)

2025-11-30 11:02:40 -05:00