Skip to content

Commit 1a79234

Browse files
committed
Add benchmark enable switch
1 parent bf024cd commit 1a79234

2 files changed

Lines changed: 5 additions & 1 deletion

File tree

tests/platforms/cuda.yaml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,5 +77,7 @@ a100:
7777
include: "*"
7878
# Exclude patterns: empty list or list paths to exclude
7979
exclude: []
80+
8081
benchmark:
81-
smoke: ["throughput", "latency", "serve"]
82+
enabled: true
83+
smoke: ["throughput", "latency", "serve"]

tests/run.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -338,6 +338,8 @@ def _discover_e2e_tests(self) -> list[TestCase]:
338338
def _discover_benchmark_tests(self) -> list[TestCase]:
339339
"""Benchmark smoke tests selected by platform YAML."""
340340
benchmark = self.config.get_benchmark_tests()
341+
if not benchmark.get("enabled", False):
342+
return []
341343
selected_smoke = benchmark.get("smoke", [])
342344

343345
if not selected_smoke:

0 commit comments

Comments
 (0)