Jiajun Ji 4ff3f4212f [XPU] Add update_attn_mask_offsets op for xpu. (#6556) ...

* add update_attn_mask_offsets op for xpu.

* format code style.

* format codes with pre-commit.

2026-03-03 18:00:05 +08:00

..

test_adjust_batch_and_gather_next_token.py

[XPU] support kernel for mtp(base) (#4748)

2025-11-27 15:05:44 +08:00

test_adjust_batch_and_recover_batch_sequence.py

[XPU] add more type for recover batch sequence (#6142)

2026-01-23 15:16:05 +08:00

test_block_attn_prefix_cache.py

fix typos (#4093)

2025-09-15 18:33:30 +08:00

test_draft_model_postprocess.py

…

test_draft_model_preprocess.py

[XPU] support kernel for mtp(base) (#4748)

2025-11-27 15:05:44 +08:00

test_draft_model_update.py

…

test_eagle_get_hidden_states.py

[XPU] support get hidden state for mix (#5513)

2025-12-12 10:31:20 +08:00

test_eagle_get_self_hidden_states.py

…

test_fused_noaux_tc.py

[XPU] support noaux_tc (#6326)

2026-02-05 12:04:16 +08:00

test_fused_rms_norm.py

[xpu] add ep custom ops (#3911)

2025-09-10 12:22:50 +08:00

test_get_infer_param.py

[xpu] add ep custom ops (#3911)

2025-09-10 12:22:50 +08:00

test_get_padding_offset.py

…

test_get_token_penalty_multi_scores.py

…

test_moe_ep_combine.py

fix typos (#4093)

2025-09-15 18:33:30 +08:00

test_moe_ep_dispatch.py

fix typos (#4093)

2025-09-15 18:33:30 +08:00

test_moe_expert_ffn.py

[XPU] refine moe_expert_ffn ut (#5743)

2025-12-25 10:35:24 +08:00

test_moe_redundant_topk_select.py

[xpu] add ep custom ops (#3911)

2025-09-10 12:22:50 +08:00

test_moe_topk_select.py

[xpu] add ep custom ops (#3911)

2025-09-10 12:22:50 +08:00

test_read_data_ipc.py

[xpu] add ep custom ops (#3911)

2025-09-10 12:22:50 +08:00

test_set_data_ipc.py

[xpu] add ep custom ops (#3911)

2025-09-10 12:22:50 +08:00

test_set_get_data_ipc.py

[xpu] add ep custom ops (#3911)

2025-09-10 12:22:50 +08:00

test_set_value_by_flags_and_idx.py

…

test_speculate_clear_accept_nums.py

…

test_speculate_get_logits.py

[XPU] add speculate_get_logits (#5497)

2025-12-12 15:38:30 +08:00

test_speculate_get_output_padding_offset.py

…

test_speculate_get_padding_offset.py

…

test_speculate_get_seq_lens_output.py

…

test_speculate_get_token_penalty_multi_scores.py

[XPU] Speculate Decoding + PD, benchmark fix (#6036)

2026-01-15 19:19:03 +08:00

test_speculate_rebuild_append_padding.py

…

test_speculate_schedule_cache.py

[XPU] Speculate Decoding + PD, benchmark fix (#6036)

2026-01-15 19:19:03 +08:00

test_speculate_set_stop_value_multi_seqs.py

…

test_speculate_set_value_by_flags.py

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

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

test_speculate_step_system_cache.py

[XPU] add speculate_step_system_cache (#5397)

2025-12-09 14:40:11 +08:00

test_speculate_step.py

[XPU] support kernel for mtp(base) (#4748)

2025-11-27 15:05:44 +08:00

test_speculate_update_v3.py

[XPU] support kernel for mtp(base) (#4748)

2025-11-27 15:05:44 +08:00

test_speculate_verify.py

[XPU] modify speculate_verify (#5522)

2025-12-23 14:50:30 +08:00

test_step.py

…

test_stop_generation_multi_ends.py

…

test_token_repetition_penalty.py

…

test_update_attn_mask_xpu.py

[XPU] Add update_attn_mask_offsets op for xpu. (#6556)

2026-03-03 18:00:05 +08:00

test_update_inputs.py

…

test_weight_only_linear.py

fix typos (#4093)

2025-09-15 18:33:30 +08:00

test_weight_quantize_xpu.py

…