Skip to content

Commit eda3cab

Browse files
ader47F.Liu
andauthored
[UT] add pcp&dcp UT for mla_cp (#4953)
### What this PR does / why we need it? Add UT of mla_cp, which include: - test_compute_prefill_context_with_dcp_pcp - test_reorg_kvcache_with_dcp_pcp - test_out_lse_reshape - test_npu_attention_update_with_dcp_pcp - test_attention_with_mask_and_nomask_with_dcp_pcp - test_process_attn_out_lse_with_dcp_pcp - test_forward_prefill_cp_with_dcp_pcp - vLLM version: v0.12.0 - vLLM main: vllm-project/vllm@ad32e3e --------- Signed-off-by: F.Liu <[email protected]> Co-authored-by: F.Liu <[email protected]>
1 parent 724d043 commit eda3cab

File tree

1 file changed

+666
-0
lines changed

1 file changed

+666
-0
lines changed

0 commit comments

Comments
 (0)