[CI] Fix stage-b-test-1-gpu-large (0) timeout by reordering LoRA tests and using tokenizer from cache (#22292)

Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Alex Nails
2026-04-07 20:00:44 -07:00
committed by GitHub
co-authored by Claude Opus 4.6
parent 86e4542f35
commit 493ec91cbe
2 changed files with 19 additions and 5 deletions
+15 -1
View File
@@ -1129,12 +1129,26 @@ def run_bench_serving(
other_args=other_server_args,
)
# Resolve tokenizer to local snapshot path when available, so the benchmark
# client's AutoTokenizer.from_pretrained uses the local path directly instead
# of calling the HF Hub API (which can stall for minutes in CI).
bench_tokenizer = tokenizer
if bench_tokenizer is None:
try:
from sglang.srt.utils import find_local_repo_dir
local_dir = find_local_repo_dir(model, revision=None)
if local_dir and os.path.isdir(local_dir):
bench_tokenizer = local_dir
except Exception:
pass
# Run benchmark
args = get_benchmark_args(
base_url=base_url,
dataset_name=dataset_name,
dataset_path=dataset_path,
tokenizer=tokenizer,
tokenizer=bench_tokenizer,
num_prompts=num_prompts,
random_input_len=random_input_len,
random_output_len=random_output_len,
@@ -141,7 +141,7 @@ class TestBenchServing1GPUPart1(CustomTestCase):
self.assertLess(res["median_ttft_ms"], 86)
self.assertLess(res["median_itl_ms"], 10)
def test_lora_online_latency(self):
def test_online_lora_latency(self):
if is_in_amd_ci():
pass
@@ -149,14 +149,14 @@ class TestBenchServing1GPUPart1(CustomTestCase):
if is_in_ci():
write_github_step_summary(
f"### test_lora_online_latency\n"
f"### test_online_lora_latency\n"
f"median_e2e_latency_ms: {res['median_e2e_latency_ms']:.2f} ms\n"
f"median_ttft_ms: {res['median_ttft_ms']:.2f} ms\n"
)
self.assertLess(res["median_e2e_latency_ms"], 2400)
self.assertLess(res["median_ttft_ms"], 58)
def test_lora_online_latency_with_concurrent_adapter_updates(self):
def test_online_lora_latency_with_concurrent_adapter_updates(self):
if is_in_amd_ci():
pass
@@ -164,7 +164,7 @@ class TestBenchServing1GPUPart1(CustomTestCase):
if is_in_ci():
write_github_step_summary(
f"### test_lora_online_latency\n"
f"### test_online_lora_latency_with_concurrent_adapter_updates\n"
f"median_e2e_latency_ms: {res['median_e2e_latency_ms']:.2f} ms\n"
f"median_ttft_ms: {res['median_ttft_ms']:.2f} ms\n"
)