From 60aaa73a4e5b4de008d47a265c35326a5a62d277 Mon Sep 17 00:00:00 2001 From: XieYunshen <1084314248@qq.com> Date: Tue, 2 Sep 2025 14:30:46 +0800 Subject: [PATCH 1/4] reopen ut --- tests/cov_pytest.ini | 6 ------ tests/operators/test_tree_mask.py | 2 +- 2 files changed, 1 insertion(+), 7 deletions(-) diff --git a/tests/cov_pytest.ini b/tests/cov_pytest.ini index 363132aa3f..bc221a8474 100644 --- a/tests/cov_pytest.ini +++ b/tests/cov_pytest.ini @@ -6,7 +6,6 @@ addopts = --ignore=tests/layers/test_append_attention.py --ignore=tests/layers/test_attention.py --ignore=tests/operators/test_rejection_top_p_sampling.py - --ignore=tests/operators/test_perchannel_gemm.py --ignore=tests/operators/test_scaled_gemm_f8_i4_f16.py --ignore=tests/operators/test_topp_sampling.py --ignore=tests/operators/test_stop_generation.py @@ -17,8 +16,3 @@ addopts = --ignore=tests/graph_optimization/test_cuda_graph_dynamic_subgraph.py --ignore=tests/graph_optimization/test_cuda_graph_spec_decode --ignore=tests/layers/test_quant_layer.py - --ignore=tests/operators/test_token_penalty.py - --ignore=tests/operators/test_split_fuse.py - --ignore=tests/operators/test_flash_mask_attn.py - --ignore=tests/operators/test_w4afp8_gemm.py - --ignore=tests/operators/test_tree_mask.py diff --git a/tests/operators/test_tree_mask.py b/tests/operators/test_tree_mask.py index 10e55a4b19..79b19c43f7 100644 --- a/tests/operators/test_tree_mask.py +++ b/tests/operators/test_tree_mask.py @@ -271,7 +271,7 @@ def run_append_c16_attention(self, q_len, kv_len, prefill=False, attn_mask=None) paddle.device.synchronize() e_time = time.time() print(f"mean infer time: {np.mean((e_time - s_time) * 1000 / self.run_time):.2f}") - return out[0].reshape([token_num, self.num_q_head, self.head_dim]) + return out.reshape([token_num, self.num_q_head, self.head_dim]) def test_naive_speculative_decoding(self): prefill_len = 8192 From 637940c68b5909150431cc34c6c1edd3abd98739 Mon Sep 17 00:00:00 2001 From: XieYunshen <1084314248@qq.com> Date: Tue, 2 Sep 2025 14:39:02 +0800 Subject: [PATCH 2/4] update --- tests/cov_pytest.ini | 13 ------------- 1 file changed, 13 deletions(-) diff --git a/tests/cov_pytest.ini b/tests/cov_pytest.ini index bc221a8474..6688d05bf5 100644 --- a/tests/cov_pytest.ini +++ b/tests/cov_pytest.ini @@ -3,16 +3,3 @@ addopts = --ignore=tests/ci_use --ignore=tests/ce - --ignore=tests/layers/test_append_attention.py - --ignore=tests/layers/test_attention.py - --ignore=tests/operators/test_rejection_top_p_sampling.py - --ignore=tests/operators/test_scaled_gemm_f8_i4_f16.py - --ignore=tests/operators/test_topp_sampling.py - --ignore=tests/operators/test_stop_generation.py - --ignore=tests/operators/test_air_topp_sampling.py - --ignore=tests/operators/test_fused_moe.py - --ignore=tests/operators/test_stop_generation_multi_ends.py - --ignore=tests/graph_optimization/test_cuda_graph.py - --ignore=tests/graph_optimization/test_cuda_graph_dynamic_subgraph.py - --ignore=tests/graph_optimization/test_cuda_graph_spec_decode - --ignore=tests/layers/test_quant_layer.py From 2f1e491c36319ddf63bd134a5f01c467e5d98737 Mon Sep 17 00:00:00 2001 From: XieYunshen <1084314248@qq.com> Date: Tue, 2 Sep 2025 18:36:39 +0800 Subject: [PATCH 3/4] update --- tests/cov_pytest.ini | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/cov_pytest.ini b/tests/cov_pytest.ini index 6688d05bf5..4e0c31e584 100644 --- a/tests/cov_pytest.ini +++ b/tests/cov_pytest.ini @@ -3,3 +3,5 @@ addopts = --ignore=tests/ci_use --ignore=tests/ce + --ignore=tests/operators/test_fused_moe.py + --ignore=tests/operators/test_w4afp8_gemm.py From 51ded9fa007fa26d4da7994500275f095d5fa32f Mon Sep 17 00:00:00 2001 From: XieYunshen <1084314248@qq.com> Date: Tue, 2 Sep 2025 18:46:35 +0800 Subject: [PATCH 4/4] update ci dockerfile --- .github/workflows/_unit_test_coverage.yml | 5 +---- scripts/unittest_requirement.txt | 2 ++ 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/.github/workflows/_unit_test_coverage.yml b/.github/workflows/_unit_test_coverage.yml index 132d0ab156..31ed7fa59a 100644 --- a/.github/workflows/_unit_test_coverage.yml +++ b/.github/workflows/_unit_test_coverage.yml @@ -168,10 +168,7 @@ jobs: python -m pip install --pre paddlepaddle-gpu -i https://www.paddlepaddle.org.cn/packages/nightly/cu126/ pip config set global.extra-index-url https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple - python -m pip install coverage - python -m pip install diff-cover - python -m pip install pytest-cov - python -m pip install jsonschema aistudio_sdk==0.3.5 + python -m pip install -r scripts/unittest_requirement.txt python -m pip install ${fd_wheel_url} rm -rf fastdeploy # coverage subprocess use diff --git a/scripts/unittest_requirement.txt b/scripts/unittest_requirement.txt index 52b0c03f7d..43894ff72e 100644 --- a/scripts/unittest_requirement.txt +++ b/scripts/unittest_requirement.txt @@ -5,3 +5,5 @@ anyio coverage diff-cover partial_json_parser +jsonschema +aistudio_sdk==0.3.5