From dd458f3212dd4ddf0e1a7907bbf539b660e70d21 Mon Sep 17 00:00:00 2001 From: Ke Bao Date: Sun, 16 Aug 2026 01:57:40 +0800 Subject: [PATCH] Fix dsv4 kl test timeout (#34963) --- .github/workflows/pr-test-extra.yml | 1 + .../unified_radix_tree/test_unified_radix_cache_kl_dsv4.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/pr-test-extra.yml b/.github/workflows/pr-test-extra.yml index 5b2349fc6..fc15908c1 100644 --- a/.github/workflows/pr-test-extra.yml +++ b/.github/workflows/pr-test-extra.yml @@ -205,6 +205,7 @@ jobs: caller_inputs: ${{ toJson(inputs) }} partitions: ${{ needs.check-changes.outputs.partitions }} run_timeout_minutes: '60' + timeout_per_file: '3600' rust_ext_artifact: ${{ needs.rust-ext-build.outputs.artifact_name }} secrets: inherit diff --git a/test/registered/radix_cache/unified_radix_tree/test_unified_radix_cache_kl_dsv4.py b/test/registered/radix_cache/unified_radix_tree/test_unified_radix_cache_kl_dsv4.py index 7e840e94a..b097d33e4 100644 --- a/test/registered/radix_cache/unified_radix_tree/test_unified_radix_cache_kl_dsv4.py +++ b/test/registered/radix_cache/unified_radix_tree/test_unified_radix_cache_kl_dsv4.py @@ -22,7 +22,7 @@ DSV4_FLASH_MODEL = "sgl-project/DeepSeek-V4-Flash-FP8" DSV4_DSPARK_MODEL = "deepseek-ai/DeepSeek-V4-Flash-DSpark" DSV4_FLASH_LAUNCH_TIMEOUT = 3600 -register_cuda_ci(est_time=1500, stage="extra-b", runner_config="4-gpu-h100") +register_cuda_ci(est_time=2400, stage="extra-b", runner_config="4-gpu-h100") def _assert_dsv4_decode_cached_tokens(result, history_len, output_len, label):