AIbin bf7e2424d0 [Optimization][Feature]Supports multiple batches of DSK-DSA. (#6930) ...

* support DSA_MUTI_BATCH

* update test topk

* update dsk-dsa

2026-03-20 15:59:22 +08:00

..

test_air_top_p_sampling.py

…

test_build_sampling_params.py

[Optimization][Speculative Decoding]Fuse padding sampling params (#6765)

2026-03-12 05:05:15 -07:00

test_cutlass_fp8_fp8_fp8_dual_gemm_fused.py

…

test_cutlass_scaled_mm.py

…

test_depermute_prefill_combine.py

[Feature] Support EP prefill with num_worst_tokens (#6574)

2026-03-11 17:09:07 +08:00

test_deqant_int8_cpp_extension.py

…

test_dequant.py

…

test_draft_model_postprocess.py

…

test_draft_model_preprocess.py

[Speculative Decoding] Fix attn_mask_offset for multi-step MTP in mixed and PD-split modes (#5738)

2025-12-25 01:54:59 -08:00

test_draft_model_set_value_by_flags.py

…

test_draft_model_update.py

[MTP] refactor MTP pre_process (#6358)

2026-02-09 10:47:15 +08:00

test_dsmla_writecache.py

Dsa clean code，add dsk_attn_write_cache baseline (#6855)

2026-03-16 11:01:14 +08:00

test_dynamic_per_token_scaled_fp8_quant.py

[Docs] Add License in Unittest (#4957)

2025-11-12 10:44:09 +08:00

test_eagle_get_hidden_states.py

…

test_eagle_get_self_hidden_states.py

…

test_flash_mask_attn.py

fix fa4 test (#6408)

2026-02-10 10:57:21 +08:00

test_fp8_fp8_half_cuda_core_gemm.py

…

test_fused_get_rotary_embedding.py

…

test_fused_hadamard_quant_fp8.py

[Docs] Add License in Unittest (#4957)

2025-11-12 10:44:09 +08:00

test_fused_masked_swiglu_quant.py

[Optimize] optimize mask_quant & swiglu (#6222)

2026-02-02 13:52:38 +08:00

test_fused_moe.py

[BugFix] fix wint2 (#6109)

2026-01-20 21:46:21 +08:00

test_fused_neox_rope_embedding.py

[Feature] Optim PaddleOCR-VL (#4873)

2025-11-07 14:56:44 +08:00

test_fused_rotary_position_encoding.py

…

test_gelu_tanh.py

[Feature] Optim PaddleOCR-VL (#4873)

2025-11-07 14:56:44 +08:00

test_get_padding_offset.py

remove speculate_get_padding_offset op (#6308)

2026-02-03 15:18:12 +08:00

test_get_position_ids_and_mask_encoder_batch.py

…

test_get_token_penalty_multi_scores.py

[Feature] GPU Memory Optimization and Retirement of V0 Scheduler (#6407)

2026-02-28 15:07:43 +08:00

test_group_swiglu_with_masked.py

…

test_hybrid_mtp_ngram.py

[Docs] Add License in Unittest (#4957)

2025-11-12 10:44:09 +08:00

test_machete_mm.py

…

test_moe_redundant_topk_select.py

…

test_moe_top_k_select.py

…

test_ngram_match.py

[Feature] GPU Memory Optimization and Retirement of V0 Scheduler (#6407)

2026-02-28 15:07:43 +08:00

test_noaux_tc_redundant.py

[Feature] Support noaux for eplb (#5143)

2025-11-21 14:10:32 +08:00

test_noaux_tc.py

[Docs] Add License in Unittest (#4957)

2025-11-12 10:44:09 +08:00

test_per_token_quant.py

[Feature] Support EP prefill with num_worst_tokens (#6574)

2026-03-11 17:09:07 +08:00

test_permute_prefill_masked_gemm.py

[BugFix] Fix ep compatibility issues & Optimize permute operator (#6821)

2026-03-17 10:32:11 +08:00

test_pre_cache_len_concat.py

[Others] Remove useless code (#5404)

2025-12-08 13:59:46 +08:00

test_qk_rmsnorm_fused.py

[Optimization] Support FA2/FA3/FA4 with attn_mask_q (#6354)

2026-02-05 14:39:00 +08:00

test_radix_topk_accuracy.py

[Optimization][Feature]Supports multiple batches of DSK-DSA. (#6930)

2026-03-20 15:59:22 +08:00

test_reasoning_phase_token_constraint.py

[Feature] GPU Memory Optimization and Retirement of V0 Scheduler (#6407)

2026-02-28 15:07:43 +08:00

test_rebuild_padding.py

[MTP] refactor MTP pre_process (#6358)

2026-02-09 10:47:15 +08:00

test_rejection_top_p_sampling.py

…

test_scaled_gemm_f8_i4_f16.py

…

test_set_value_by_flags_and_idx.py

[Feature] GPU Memory Optimization and Retirement of V0 Scheduler (#6407)

2026-02-28 15:07:43 +08:00

test_share_external_data.py

…

test_speculate_get_seq_lens_output.py

…

test_speculate_get_target_logits.py

…

test_speculate_get_token_penalty_multi_scores.py

[Feature] GPU Memory Optimization and Retirement of V0 Scheduler (#6407)

2026-02-28 15:07:43 +08:00

test_speculate_insert_first_token.py

…

test_speculate_pre_process.py

[Optimization][Speculative Decoding]Fuse padding sampling params (#6765)

2026-03-12 05:05:15 -07:00

test_speculate_set_stop_value_multi_seqs.py

[Speculative Decoding] Fix speculate stop_seqs and fix accept_num in eos branch (#6825)

2026-03-12 23:48:24 -07:00

test_speculate_update.py

[Others] remove stop_nums (#6182)

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

test_speculate_verify.py

[MTP] refactor MTP pre_process (#6358)

2026-02-09 10:47:15 +08:00

test_speculative_schedule_cache.py

[Others] remove stop_nums (#6182)

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

test_split_fuse.py

…

test_stop_generation_multi_ends.py

[Feature] GPU Memory Optimization and Retirement of V0 Scheduler (#6407)

2026-02-28 15:07:43 +08:00

test_swap_layout.py

[CI] Add unit test for swap_layout && remove unit test of splitwise_scheduler (#6250)

2026-01-28 19:20:20 +08:00

test_token_penalty.py

…

test_top_k_renorm_probs.py

…

test_top_p_candidates.py

[Docs] Add License in Unittest (#4957)

2025-11-12 10:44:09 +08:00

test_tree_mask.py

add more cases for attention unit test (#5931)

2026-01-15 19:52:35 +08:00

test_tritonmoe_preprocess.py

…

test_unified_update_model_status.py

[Speculative Decoding] Unify Spec and non-spec branch (#6685)

2026-03-10 23:58:44 -07:00

test_update_attn_mask.py

[Speculative Decoding] Support suffix decoding (#6403)

2026-02-26 11:42:05 +08:00

test_update_inputs_v1.py

[Model Runner] Refactor execute_model for GPU async scheduling (#6176)

2026-01-28 14:19:33 +08:00

test_verify_draft_tokens.py

[Speculative Decoding] Fix speculate stop_seqs and fix accept_num in eos branch (#6825)

2026-03-12 23:48:24 -07:00

test_w4afp8_gemm.py

fix w4afp8 (#5634)

2025-12-22 13:39:41 +08:00

test_wfp8afp8_sparse_gemm.py

…