[router] Fix ci nvcc not found error (#11411)

This commit is contained in:
Keyang Ru
2025-10-10 15:43:16 -07:00
committed by GitHub
parent 55b14656e6
commit a0835c3a62
4 changed files with 7 additions and 6 deletions

View File

@@ -692,7 +692,7 @@ def pytest_configure(config):
@pytest.fixture(scope="session")
def e2e_model() -> str:
# Always use the default test model
return DEFAULT_MODEL_NAME_FOR_TEST
return os.getenv("E2E_PRIMARY_MODEL", DEFAULT_MODEL_NAME_FOR_TEST)
@pytest.fixture

View File

@@ -253,10 +253,10 @@ def test_pd_genai_bench(e2e_model: str, pd_cluster, genai_bench_runner):
model_path=e2e_model,
experiment_folder=policy_label,
thresholds={
"ttft_mean_max": 12,
"e2e_latency_mean_max": 15,
"input_throughput_mean_min": 400,
"output_throughput_mean_min": 20,
"ttft_mean_max": 13,
"e2e_latency_mean_max": 16,
"input_throughput_mean_min": 350,
"output_throughput_mean_min": 18,
"gpu_util_p50_min": 99,
},
kill_procs=pd_cluster.workers,