cmcamdy 5a67a6d960 [XPU] support kernel for mtp(base) (#4748) ...

* [XPU] support kernel for mtp(base)

* [XPU] support kernel for mtp(base)

* format

* format

* format

* fix gather next token

* fix step && add test

* fix

* mv pre/post process

* add adjust batch / gather next token for mtp

* fix code style

* fix mtp kenrel name

* fix mtp kernel test

* mv xpu pre/post process

* mv xpu pre/post process

2025-11-27 15:05:44 +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_block_attn_prefix_cache.py

fix typos (#4093)

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

test_draft_model_postprocess.py

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

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

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

test_eagle_get_hidden_states.py

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

test_eagle_get_self_hidden_states.py

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +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

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

test_get_token_penalty_multi_scores.py

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

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] add ep custom ops (#3911)

2025-09-10 12:22:50 +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

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

test_speculate_clear_accept_nums.py

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

test_speculate_get_output_padding_offset.py

add dtype int32 (#3692)

2025-08-29 14:56:35 +08:00

test_speculate_get_padding_offset.py

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

test_speculate_get_seq_lens_output.py

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

test_speculate_get_token_penalty_multi_scores.py

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

test_speculate_rebuild_append_padding.py

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

test_speculate_set_stop_value_multi_seqs.py

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

test_speculate_set_value_by_flags.py

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +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

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

test_step.py

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

test_stop_generation_multi_ends.py

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

test_token_repetition_penalty.py

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

test_update_inputs.py

[Feature][XPU] add custom kernels for mtp (#3537)

2025-08-25 10:14:17 +08:00

test_weight_only_linear.py

fix typos (#4093)

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

test_weight_quantize_xpu.py

fix typos (#3684)

2025-09-01 17:50:17 +08:00