Skip to content

Commit baffdde

Browse files
authored
chore: publish benchmarking results in json file (#3110)
Signed-off-by: Sahithi Chigurupati <[email protected]>
1 parent 6735f46 commit baffdde

File tree

1 file changed

+6
-1
lines changed
  • components/backends/trtllm/performance_sweeps/scripts

1 file changed

+6
-1
lines changed

components/backends/trtllm/performance_sweeps/scripts/bench.sh

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,8 +163,10 @@ python3 ${SCRIPTS_DIR}/scripts/bench/benchmark_serving.py \
163163
--host ${hostname} \
164164
--port ${port}
165165

166+
mkdir -p ${log_path}/results
166167
echo "Starting benchmark..."
167168
for concurrency in ${concurrency_list}; do
169+
original_concurrency=${concurrency}
168170
concurrency=$((concurrency * num_gen_servers))
169171
num_prompts=$((concurrency * multi_round))
170172
echo "Benchmarking with concurrency ${concurrency} ... ${num_prompts} prompts"
@@ -186,7 +188,10 @@ for concurrency in ${concurrency_list}; do
186188
--percentile-metrics ttft,tpot,itl,e2el \
187189
--max-concurrency "$concurrency" \
188190
--host ${hostname} \
189-
--port ${port}
191+
--port ${port} \
192+
--save-result \
193+
--result-dir "${log_path}/results" \
194+
--result-filename "results_concurrency_${original_concurrency}_gpus_${total_gpus}.json"
190195

191196
echo "Benchmark with concurrency ${concurrency} done"
192197
do_get_logs ${log_path}/output_workers.log ${log_path}/concurrency_${concurrency}

0 commit comments

Comments
 (0)