Skip to content

Commit ce29b20

Browse files
committed
Benchmark aime25
Signed-off-by: Huy Do <huydhn@gmail.com>
1 parent 82fa731 commit ce29b20

3 files changed

Lines changed: 11 additions & 8 deletions

File tree

.github/scripts/gpt-oss/run_accuracy_checks.sh

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -41,55 +41,57 @@ fi
4141

4242
pushd vllm-benchmarks/gpt-oss
4343
mkdir -p /tmp/gpqa_openai
44+
mkdir -p /tmp/aime25_openai
4445

4546
if [[ "${DEVICE_NAME}" == "rocm" ]]; then
4647
# Not sure why this is needed on ROCm
4748
pushd gpt_oss
4849
# Low
4950
OPENAI_API_KEY="" python3 -mevals --base-url http://localhost:8000/v1 \
5051
--model $MODEL \
51-
--eval gpqa \
52+
--eval aime25 \
5253
--reasoning-effort low \
5354
--n-threads $(expr $(nproc) / 2)
5455

5556
# Mid
5657
OPENAI_API_KEY="" python3 -mevals --base-url http://localhost:8000/v1 \
5758
--model $MODEL \
58-
--eval gpqa \
59+
--eval aime25 \
5960
--reasoning-effort medium \
6061
--n-threads $(expr $(nproc) / 2)
6162

6263
# High
6364
OPENAI_API_KEY="" python3 -mevals --base-url http://localhost:8000/v1 \
6465
--model $MODEL \
65-
--eval gpqa \
66+
--eval \
6667
--reasoning-effort high \
6768
--n-threads $(expr $(nproc) / 2)
6869
popd
6970
else
7071
# Low
7172
OPENAI_API_KEY="" python3 -m gpt_oss.evals --base-url http://localhost:8000/v1 \
7273
--model $MODEL \
73-
--eval gpqa \
74+
--eval aime25 \
7475
--reasoning-effort low \
7576
--n-threads $(expr $(nproc) / 2)
7677

7778
# Mid
7879
OPENAI_API_KEY="" python3 -m gpt_oss.evals --base-url http://localhost:8000/v1 \
7980
--model $MODEL \
80-
--eval gpqa \
81+
--eval aime25 \
8182
--reasoning-effort medium \
8283
--n-threads $(expr $(nproc) / 2)
8384

8485
# High
8586
OPENAI_API_KEY="" python3 -m gpt_oss.evals --base-url http://localhost:8000/v1 \
8687
--model $MODEL \
87-
--eval gpqa \
88+
--eval aime25 \
8889
--reasoning-effort high \
8990
--n-threads $(expr $(nproc) / 2)
9091
fi
9192

9293
mv /tmp/gpqa_openai .
94+
mv /tmp/aime25_openai .
9395
popd
9496

9597
kill -9 $server_pid

.github/scripts/gpt-oss/run_benchmarks.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,5 +25,6 @@ if [[ "${DEVICE_NAME}" != "rocm" ]]; then
2525
fi
2626

2727
pip freeze
28-
bash .buildkite/nightly-benchmarks/scripts/run-performance-benchmarks.sh
28+
# Just run accuracy tests for now
29+
# bash .buildkite/nightly-benchmarks/scripts/run-performance-benchmarks.sh
2930
popd

.github/workflows/gpt-oss-benchmark.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,7 @@ jobs:
192192
docker exec -t "${container_name}" bash .github/scripts/gpt-oss/run_benchmarks.sh
193193
194194
# Run accuracy check (turning this on later if needed)
195-
# docker exec -t "${container_name}" bash .github/scripts/gpt-oss/run_accuracy_checks.sh
195+
docker exec -t "${container_name}" bash .github/scripts/gpt-oss/run_accuracy_checks.sh
196196
197197
- name: Authenticate with AWS
198198
# AWS CUDA runners already have access to the bucket via its runner IAM role

0 commit comments

Comments
 (0)