Xin Yang b1c4f0b265 [Kernel] Optimize grouped topk kernel (#34206) ...

Signed-off-by: Xin Yang <xyangx@amazon.com>

2026-02-20 01:34:45 -08:00

..

attention

Add unit tests for fp8 output fusion of triton_attn (#34228)

2026-02-18 06:22:49 -05:00

core

[Bugfix] Fix quant RMS norm fusion for quantization with TMA-aligned scales (#33255)

2026-02-17 23:35:04 -08:00

helion

[Kernel] [Helion] [4/N] Add silu_mul_fp8 Helion kernel (#33373)

2026-02-12 18:13:12 -08:00

mamba

[Chore] Migrate V0 attention utils (#31891)

2026-01-07 13:44:36 +00:00

moe

[Kernel] Optimize grouped topk kernel (#34206)

2026-02-20 01:34:45 -08:00

quantization

[CI][AMD][BugFix] Use torch.testing.assert_close instead of assert torch.allclose in test_rocm_skinny_gemms.py (#34181)

2026-02-18 23:10:19 +00: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

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

quant_utils.py

[Perf] Create TMA-aligned input scale tensor for DeepGemm on Hopper (#32619)

2026-01-22 15:47:04 -05:00

test_apply_repetition_penalties.py

[Platform] Deprecate seed_everything (#31659)

2026-01-04 18:34:04 -08:00

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

[Platform] Deprecate seed_everything (#31659)

2026-01-04 18:34:04 -08:00

test_flex_attention.py

[Tests] Standardize RNG seed utility across test files (#32982)

2026-01-24 06:47:14 +00:00

test_fused_quant_activation.py

[Misc] Fix Current vLLM config is not set. warnings, assert to avoid issues in the future (#31747)

2026-01-08 15:20:49 -08:00

test_onednn.py

[CPU] Refactor CPU attention backend (#27954)

2025-11-12 09:43:06 +08:00

test_shuffle_rows.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

test_top_k_per_row.py

[Perf][Kernel] Add faster topKperRow decode kernel for DeepSeek-V3.2 sparse attention (#33680)

2026-02-10 10:29:52 -05:00

utils.py

[Refactor] Replace activation: str with MoEActivation enum (#33843)

2026-02-11 17:29:32 -08:00