Skip to content

Commit 8d0dbb2

Browse files
kaushikmitrclaude
andcommitted
Parameterize inference-perf workload profiles
Replace hardcoded load stages and conversation_replay settings in the agentic_code_generation and guide_predicted-latency-routing profiles with ${NUM_REQUESTS}, ${CONCURRENCY_LEVEL}/${CONCURRENCY}, and ${SEED} environment variables so the profiles can be driven dynamically. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com> Signed-off-by: Kaushik Mitra <kaushikmitra@google.com>
1 parent 978befa commit 8d0dbb2

2 files changed

Lines changed: 8 additions & 16 deletions

File tree

workload/profiles/inference-perf/agentic_code_generation.yaml.in

Lines changed: 5 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -3,16 +3,8 @@ load:
33
num_workers: 8
44
worker_max_concurrency: 1000
55
stages:
6-
- num_requests: 100
7-
concurrency_level: 5
8-
- num_requests: 200
9-
concurrency_level: 10
10-
- num_requests: 400
11-
concurrency_level: 20
12-
- num_requests: 600
13-
concurrency_level: 30
14-
- num_requests: 800
15-
concurrency_level: 40
6+
- num_requests: ${NUM_REQUESTS}
7+
concurrency_level: ${CONCURRENCY_LEVEL}
168
api:
179
type: completion
1810
streaming: true
@@ -23,9 +15,9 @@ server:
2315
data:
2416
type: conversation_replay
2517
conversation_replay:
26-
seed: 42
27-
max_model_len: 262144 # Update based on the model's context length
28-
num_conversations: 40
18+
seed: ${SEED}
19+
max_model_len: 262144
20+
num_conversations: ${CONCURRENCY_LEVEL}
2921
shared_system_prompt_len: 3000
3022
dynamic_system_prompt_len:
3123
type: lognormal

workload/profiles/inference-perf/guide_predicted-latency-routing_1.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@ load:
33
num_workers: 6
44
request_timeout: 300.0
55
stages:
6-
- num_requests: 192
7-
concurrency_level: 32
6+
- num_requests: ${NUM_REQUESTS}
7+
concurrency_level: ${CONCURRENCY}
88
api:
99
type: completion
1010
streaming: true
@@ -18,7 +18,7 @@ tokenizer:
1818
data:
1919
type: conversation_replay
2020
conversation_replay:
21-
seed: ${CONCURRENCY} # unique seed per concurrency level
21+
seed: ${SEED} # unique seed per concurrency level
2222
num_conversations: ${CONCURRENCY} # equals concurrency_level above
2323
shared_system_prompt_len: 3000
2424
dynamic_system_prompt_len:

0 commit comments

Comments
 (0)