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),