Mohammad Miadh AngkadandMohammad Angkad 4c81cd1b09 [KDA] Fix missing beta sigmoid in PTX prefill (#40685) ...

Co-authored-by: Mohammad Angkad <mohammad.angkad@radixark.ai>

2026-09-21 23:24:38 -07:00

..

test_c4_v2.py

[AMD][DSV4] Reland unified-KV pool sizing and SWA ring accounting, fully gated (#38192)

2026-09-07 13:13:04 -07:00

test_c128_v2.py

[Kernel] Reclassify kernel tests by ops group + move helpers out of the package (RFC #29630) (#32128)

2026-07-23 12:18:27 +08:00

test_clamp_position.py

[Kernel] Add OOT dispatch for clamp position (#38687)

2026-09-15 10:22:21 +08:00

test_combine_topk_swa_indices.py

[diffusion] model: support qwen-image-2.1 (#39983)

2026-09-20 09:46:09 +08:00

test_concat_mla.py

[Kernel] Reclassify kernel tests by ops group + move helpers out of the package (RFC #29630) (#32128)

2026-07-23 12:18:27 +08:00

test_cute_dsl_fp8_paged_mqa_logits.py

[Cleanup] Deduplicate kernel tests, diffusion fixtures and benchmark helpers (#40265)

2026-09-19 19:45:38 +08:00

test_cutedsl_gdn.py

[CI][RFC] Replace black-jupyter with ruff-format (#37210)

2026-09-02 19:46:08 -07:00

test_dcp_lse_combine.py

[CI] Derive registered-test kind from the registry call instead of the path (#40294)

2026-09-20 02:01:22 -07:00

test_deepgemm_paged_mqa_logits.py

[Cleanup] Deduplicate kernel tests, diffusion fixtures and benchmark helpers (#40265)

2026-09-19 19:45:38 +08:00

test_deepseek_v4_compress_plan_bounds.py

[Test] Consolidate kernel tests under plural kernels tree (#39966)

2026-09-18 07:37:48 +08:00

test_deepseek_v4_compress_state_runtime_shapes.py

[Test] Consolidate kernel tests under plural kernels tree (#39966)

2026-09-18 07:37:48 +08:00

test_dense_prefill_indexer.py

[DeepSeek-V4.1] Bound dense prefill indexer memory (#40217)

2026-09-20 14:43:19 -07:00

test_dsa_indexer.py

chore: update CI test est_time values (#38238)

2026-09-06 17:49:41 -07:00

test_dsa_kpool_metadata_fusion.py

[Test] Consolidate kernel tests under plural kernels tree (#39966)

2026-09-18 07:37:48 +08:00

test_dsa_kpool_multi_pool.py

[CI] Derive registered-test kind from the registry call instead of the path (#40294)

2026-09-20 02:01:22 -07:00

test_dsa_metadata_replay.py

[Test] Consolidate kernel tests under plural kernels tree (#39966)

2026-09-18 07:37:48 +08:00

test_dsa_metadata.py

[CI] Give the kernel lane a 5090 suite and move kernel-only tests off the general lane (#40496)

2026-09-20 16:46:53 -07:00

test_dsa_multi_ctas_counter.py

[Fix] Don't free the multi-CTAs KV counter the decode graphs captured (#39175)

2026-09-21 12:58:48 -07:00

test_dsa_sink_pad_cache.py

Support Hy4-preview (#36805)

2026-09-04 18:03:49 -07:00

test_dsa_transform_index.py

[AMD] [GLM-5.3-Flash Day 0] Support non-2048 top-k widths in the DSA page-table transform (#39340)

2026-09-21 19:50:17 -07:00

test_dsv4_indexer_quant.py

[AMD] fix: use the hardware fp8 e4m3 convert on gfx950 (#37140)

2026-09-08 03:01:53 -07:00

test_dsv4_q_rope_store.py

[Test] Consolidate kernel tests under plural kernels tree (#39966)

2026-09-18 07:37:48 +08:00

test_dsv32_indexer_fusion.py

chore: update CI test est_time values (#38238)

2026-09-06 17:49:41 -07:00

test_flash_attention_3_only_qv.py

[Kernel] Reclassify kernel tests by ops group + move helpers out of the package (RFC #29630) (#32128)

2026-07-23 12:18:27 +08:00

test_flash_attention_4_sm120.py

[CI] Give the kernel lane a 5090 suite and move kernel-only tests off the general lane (#40496)

2026-09-20 16:46:53 -07:00

test_flash_attention_4.py

fix: support FA4 backend for GLM4.7-flash (#33436)

2026-08-09 20:05:38 +08:00

test_flash_mla_backends.py

[SM120] Use exact query-head widths for DeepSeek-V4 sparse MLA decode (#36655)

2026-09-10 15:40:17 -07:00

test_flashmla_sched_meta.py

[Test] Consolidate kernel tests under plural kernels tree (#39966)

2026-09-18 07:37:48 +08:00

test_fp4_indexer_hip.py

[AMD] Fix FP4 indexer OOR (#37660)

2026-09-03 01:49:46 -07:00

test_fp4_indexer.py

[DSV4.1] Reduce mHC, metadata and small-batch router overhead (#39704)

2026-09-19 19:51:21 +08:00

test_fp8_wo_a.py

[DSV4] Emit TMA-aligned UE8M0 scales for FP8 einsum (#34277)

2026-08-16 22:06:41 -07:00

test_fused_kda_conv_recurrent_verify.py

[Fix] Don't write conv state from the fused KDA verify kernel (#39524)

2026-09-21 18:33:03 -07:00

test_fused_metadata_copy.py

[Cleanup] Deduplicate kernel tests, diffusion fixtures and benchmark helpers (#40265)

2026-09-19 19:45:38 +08:00

test_fused_qk_rmsnorm_rope_gate.py

[rotary] Fix the fused Qwen3.5 RoPE kernel discarding mrope height and width (#34446)

2026-08-30 15:37:40 +08:00

test_fused_store_index_cache.py

[CI][RFC] Replace black-jupyter with ruff-format (#37210)

2026-09-02 19:46:08 -07:00

test_fused_verify_triton_gdn.py

[AMD] Tune Qwen3.5 TP4 GDN recurrent launch on gfx950 (#39987)

2026-09-21 13:20:54 -07:00

test_gdn_decode_fused_proj_conv.py

[CI] Give the kernel lane a 5090 suite and move kernel-only tests off the general lane (#40496)

2026-09-20 16:46:53 -07:00

test_hy4_hpc_gated_mla.py

Support Hy4-preview (#36805)

2026-09-04 18:03:49 -07:00

test_inkling_attn_prologue_tau.py

[Kernel] Classification cleanup: unify _jit_ naming, drop empty/model groups, add elementwise (RFC #29630) (#32148)

2026-07-23 13:47:02 +08:00

test_inkling_rel_proj.py

[Kernel] Add inventory guards and clean benchmark layout (#32788)

2026-07-30 09:03:24 +08:00

test_inkling_row_scale.py

[Kernel] Classification cleanup: unify _jit_ naming, drop empty/model groups, add elementwise (RFC #29630) (#32148)

2026-07-23 13:47:02 +08:00

test_kda_fused_decode.py

feat(kernels): port standalone Kimi K3 kernels (#32890)

2026-08-01 13:26:47 +08:00

test_kda_fused_verify_backend.py

[Test] Consolidate kernel tests under plural kernels tree (#39966)

2026-09-18 07:37:48 +08:00

test_kda_gate_beta_cumsum.py

[kimi k3][pd disagg] support pp prefill + dcp decode with dspark (#40045)

2026-09-20 00:15:28 -07:00

test_kda_helion.py

[Kernel] GLM 5.3 Flash related kernels (ported from #36507) (#37477)

2026-09-01 22:17:09 -07:00

test_kda_mtp_cutedsl_replayssm_ring.py

[CI] Derive registered-test kind from the registry call instead of the path (#40294)

2026-09-20 02:01:22 -07:00

test_kda_prefill.py

[KDA] Fix missing beta sigmoid in PTX prefill (#40685)

2026-09-21 23:24:38 -07:00

test_kda_replayssm_fold_batched.py

[CI] Derive registered-test kind from the registry call instead of the path (#40294)

2026-09-20 02:01:22 -07:00

test_kda_replayssm_fold.py

[CI] Derive registered-test kind from the registry call instead of the path (#40294)

2026-09-20 02:01:22 -07:00

test_kda_replayssm_ring_fused.py

[CI] Derive registered-test kind from the registry call instead of the path (#40294)

2026-09-20 02:01:22 -07:00

test_kda_replayssm_ring_ragged.py

[CI] Derive registered-test kind from the registry call instead of the path (#40294)

2026-09-20 02:01:22 -07:00

test_kda_track_state.py

[Test] Consolidate kernel tests under plural kernels tree (#39966)

2026-09-18 07:37:48 +08:00

test_lean_attention.py

[CI] Derive registered-test kind from the registry call instead of the path (#40294)

2026-09-20 02:01:22 -07:00

test_minimax_decode_topk_page_table.py

[CI][RFC] Replace black-jupyter with ruff-format (#37210)

2026-09-02 19:46:08 -07:00

test_minimax_decode_topk.py

[Kernel] Reclassify kernel tests by ops group + move helpers out of the package (RFC #29630) (#32128)

2026-07-23 12:18:27 +08:00

test_minimax_m3_qk_norm_rope.py

[Kernel] Reclassify kernel tests by ops group + move helpers out of the package (RFC #29630) (#32128)

2026-07-23 12:18:27 +08:00

test_minimax_qknorm_rope.py

[Kernel] Reclassify kernel tests by ops group + move helpers out of the package (RFC #29630) (#32128)

2026-07-23 12:18:27 +08:00

test_mla_kv_pack_quantize_fp8.py

[Kernel] Reclassify kernel tests by ops group + move helpers out of the package (RFC #29630) (#32128)

2026-07-23 12:18:27 +08:00

test_paged_mqa_metadata.py

[CI][RFC] Replace black-jupyter with ruff-format (#37210)

2026-09-02 19:46:08 -07:00

test_pos_enc.py

[CI][RFC] Replace black-jupyter with ruff-format (#37210)

2026-09-02 19:46:08 -07:00

test_q8kv8_sparse_prefill_backend.py

[Fix] Repair CI fixtures and ROCm speculative tree device checks (#40325)

2026-09-19 18:16:05 +08:00

test_qprep_bf16_fp8_sm90.py

[CI][RFC] Replace black-jupyter with ruff-format (#37210)

2026-09-02 19:46:08 -07:00

test_rope.py

[CI] Slim JIT kernel unit tests (#36887)

2026-08-29 07:26:49 +08:00

test_sm120_paged_mqa_logits.py

chore: update CI test est_time values (#38238)

2026-09-06 17:49:41 -07:00

test_sparse_mla_q8kv8_prefill_sm90.py

[DSA] Q8KV8 FP8 Sparse Prefill on GLM-5.2 & DeepSeek-V3.2: Q8-Path & Shared-Path Optimizations (#31888)

2026-07-30 15:15:11 +08:00

test_subblock_sage_fp8_sm90.py

[Test] Consolidate kernel tests under plural kernels tree (#39966)

2026-09-18 07:37:48 +08:00

test_subblock_sparse_sm90.py

[diffusion][Minimax H3]support subblock sparse attention on SM90 (#34680)

2026-08-19 10:31:45 +08:00

test_suffix_attention_merge.py

[Speculative Decoding] Add native UNO serving support (#37667)

2026-09-03 20:08:41 +08:00

test_topk_v2.py

[AMD] Enable GLM DSA prefill top-k to the v2 kernel (#37889)

2026-09-21 10:31:07 -07:00

test_wo_a_fused.py

[DSV4.1] Big fused wo_a quant (#39957)

2026-09-19 19:46:53 +08:00