RuohengMa 12c76f8137 [XPU] add speculate_get_logits (#5497) ...

* [XPU] add speculate_step_system_cache

* [XPU] add speculate_step_system_cache

* [XPU] add speculate_get_logits

* delete context

* add ptr check

---------

Co-authored-by: cmcamdy <1027740945@qq.com>
Co-authored-by: YuBaoku <49938469+EmmonsCurse@users.noreply.github.com>

2025-12-12 15:38:30 +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

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

2025-12-12 10:31:20 +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_logits.py

[XPU] add speculate_get_logits (#5497)

2025-12-12 15:38:30 +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_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

[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