Skip to content

Commit 2e34b72

Browse files
author
cloudforge1
committed
bench: remove env-gate from benchmark groups, cut NUM_ITERS to 1000
Benchmark groups 1-5 now run unconditionally in CI (~9s total). Env-gates moved to separate PR PaddlePaddle#7170.
1 parent 3ddae7f commit 2e34b72

1 file changed

Lines changed: 1 addition & 5 deletions

File tree

tests/spec_decode/test_benchmark_ngram_kernel.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040

4141
MAX_NGRAM_SIZE = 3
4242
MAX_DRAFT_TOKENS = 10
43-
NUM_ITERS = 10000
43+
NUM_ITERS = 1000
4444
WARMUP = 5
4545

4646

@@ -206,10 +206,6 @@ def _print_table(title, header, rows):
206206
print(f"{'=' * 80}")
207207

208208

209-
@unittest.skipUnless(
210-
os.environ.get("RUN_NGRAM_BENCHMARKS", "0") == "1",
211-
"Set RUN_NGRAM_BENCHMARKS=1 to run multi-group profiling (slow)",
212-
)
213209
class TestNgramBenchmarkGroups(unittest.TestCase):
214210
"""Multi-dimension benchmark matching NKNaN's 5-group methodology."""
215211

0 commit comments

Comments
 (0)