From d7b706be945bea1c55ce86ba9c6c6558c41c915d Mon Sep 17 00:00:00 2001 From: Alison Shao <54658187+alisonshao@users.noreply.github.com> Date: Mon, 5 Jan 2026 19:38:21 -0800 Subject: [PATCH] ci: migrate HiCache 1-GPU tests to test/registered/hicache/ (#16416) --- test/{srt => registered}/hicache/test_hicache_storage.py | 4 ++++ test/{srt => registered}/hicache/test_hicache_variants.py | 4 ++++ test/srt/run_suite.py | 2 -- 3 files changed, 8 insertions(+), 2 deletions(-) rename test/{srt => registered}/hicache/test_hicache_storage.py (92%) rename test/{srt => registered}/hicache/test_hicache_variants.py (97%) diff --git a/test/srt/hicache/test_hicache_storage.py b/test/registered/hicache/test_hicache_storage.py similarity index 92% rename from test/srt/hicache/test_hicache_storage.py rename to test/registered/hicache/test_hicache_storage.py index b4e905326..75da6e138 100644 --- a/test/srt/hicache/test_hicache_storage.py +++ b/test/registered/hicache/test_hicache_storage.py @@ -1,3 +1,7 @@ +from sglang.test.ci.ci_register import register_cuda_ci + +register_cuda_ci(est_time=96, suite="stage-b-test-small-1-gpu") + import time import unittest from types import SimpleNamespace diff --git a/test/srt/hicache/test_hicache_variants.py b/test/registered/hicache/test_hicache_variants.py similarity index 97% rename from test/srt/hicache/test_hicache_variants.py rename to test/registered/hicache/test_hicache_variants.py index b0b360629..ea20634c9 100644 --- a/test/srt/hicache/test_hicache_variants.py +++ b/test/registered/hicache/test_hicache_variants.py @@ -1,3 +1,7 @@ +from sglang.test.ci.ci_register import register_cuda_ci + +register_cuda_ci(est_time=368, suite="stage-b-test-small-1-gpu") + """ Consolidated HiCache variant tests. Tests HiCache with different configurations: standard, MLA, EAGLE, and page size variants. diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py index 9c0f8e06c..7793145cb 100644 --- a/test/srt/run_suite.py +++ b/test/srt/run_suite.py @@ -10,8 +10,6 @@ from sglang.test.ci.ci_utils import TestFile, run_unittest_files suites = { "per-commit-1-gpu": [ TestFile("debug_utils/test_tensor_dump_forward_hook.py", 9), - TestFile("hicache/test_hicache_storage.py", 96), - TestFile("hicache/test_hicache_variants.py", 368), TestFile("openai_server/basic/test_openai_embedding.py", 70), TestFile("openai_server/basic/test_openai_server.py", 184), TestFile("openai_server/basic/test_protocol.py", 3),