Skip to content

Commit 1f6a658

Browse files
committed
extend LLM benchmarks to another ~300 servers
1 parent 042b1e8 commit 1f6a658

File tree

1 file changed

+11
-6
lines changed

1 file changed

+11
-6
lines changed

inspector/tasks.py

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@
5454
version_command="bash -c \"nvidia-smi -h | head -1 | egrep -o 'v[0-9.]+'\"",
5555
command="nvidia-smi -q -x",
5656
precheck_command="lshw -C display -json | jq -r '.[].vendor'",
57-
precheck_regex="nvidia"
57+
precheck_regex="nvidia",
5858
)
5959

6060
virtualization = DockerTask(
@@ -123,8 +123,13 @@
123123
priority=4,
124124
image="ghcr.io/sparecores/benchmark:main",
125125
version_command="bash -c \"/usr/local/geekbench-$(uname -m)/geekbench6 --version | awk '{print $2}'\"",
126-
docker_opts=DOCKER_OPTS | dict(
127-
environment={"BENCHMARK_SECRETS_PASSPHRASE": os.environ.get("BENCHMARK_SECRETS_PASSPHRASE")},
126+
docker_opts=DOCKER_OPTS
127+
| dict(
128+
environment={
129+
"BENCHMARK_SECRETS_PASSPHRASE": os.environ.get(
130+
"BENCHMARK_SECRETS_PASSPHRASE"
131+
)
132+
},
128133
mem_limit=int(mem_bytes * 0.85),
129134
memswap_limit=int(mem_bytes * 0.85),
130135
mem_swappiness=0,
@@ -140,7 +145,7 @@
140145
minimum_memory=1,
141146
timeout=timedelta(hours=1),
142147
image="ghcr.io/sparecores/benchmark:main",
143-
command="nice -n -20 python /usr/local/bin/compress.py"
148+
command="nice -n -20 python /usr/local/bin/compress.py",
144149
)
145150

146151
bw_mem = DockerTask(
@@ -175,7 +180,7 @@
175180
version_command="bash -c \"nvbandwidth --help | head -1 | egrep -o 'v[0-9.]+'\"",
176181
command="nvbandwidth -j",
177182
precheck_command="lshw -C display -json | jq -r '.[].vendor'",
178-
precheck_regex="nvidia"
183+
precheck_regex="nvidia",
179184
)
180185

181186
passmark = DockerTask(
@@ -193,7 +198,7 @@
193198
timeout=timedelta(hours=1),
194199
minimum_memory=1,
195200
priority=11,
196-
rollout=0.05, # ~100 servers
201+
rollout=0.2, # ~400 servers
197202
image="ghcr.io/sparecores/benchmark-llm:main",
198203
command=None,
199204
)

0 commit comments

Comments
 (0)