From 0c017db9163854880b7ee6983e7e9a2ccfb4200e Mon Sep 17 00:00:00 2001 From: Cheng Wan <54331508+ch-wan@users.noreply.github.com> Date: Sat, 16 May 2026 16:43:12 -0700 Subject: [PATCH] Update kl_div_thres to 0.02 in swa_radix_cache (#25497) --- test/registered/radix_cache/test_swa_radix_cache_kl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/registered/radix_cache/test_swa_radix_cache_kl.py b/test/registered/radix_cache/test_swa_radix_cache_kl.py index 795ad9d9e..97afbd45b 100644 --- a/test/registered/radix_cache/test_swa_radix_cache_kl.py +++ b/test/registered/radix_cache/test_swa_radix_cache_kl.py @@ -11,7 +11,7 @@ register_cuda_ci(est_time=151, stage="base-b", runner_config="1-gpu-large") class TestSWARadixCacheKL(KLDivergenceMixin, DefaultServerBase): model = MODEL - kl_div_thres = 0.002 + kl_div_thres = 0.02 # it was 0.002 kl_div_decode_max_new_tokens = 2048 other_args = [ "--tp-size",