[Config] Round 5.2: the per-model declarations get their own modules (#37087)
Co-authored-by: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 5
parent
7e751153eb
commit
e51a3ae65e
@@ -1200,6 +1200,9 @@ class TestDerivedPredicatesAgreeAcrossTiers(_IsolatedServerArgs):
|
||||
def test_activation_reserve_matches_the_member(self):
|
||||
from types import SimpleNamespace
|
||||
|
||||
from sglang.srt.arg_groups.overrides import (
|
||||
pre_capture_activation_reserve_mb_of,
|
||||
)
|
||||
from sglang.srt.runtime_context import pre_capture_activation_reserve_mb
|
||||
|
||||
graph = SimpleNamespace(decode=SimpleNamespace(max_bs=64))
|
||||
@@ -1231,7 +1234,7 @@ class TestDerivedPredicatesAgreeAcrossTiers(_IsolatedServerArgs):
|
||||
args = _FakeResolvedArgs(cuda_graph_config=graph, **case)
|
||||
get_context().set_server_args(args)
|
||||
self.assertEqual(
|
||||
ServerArgs.pre_capture_activation_reserve_mb(args, gpu_mem),
|
||||
pre_capture_activation_reserve_mb_of(args, gpu_mem),
|
||||
pre_capture_activation_reserve_mb(gpu_mem),
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user