|
54 | 54 | version_command="bash -c \"nvidia-smi -h | head -1 | egrep -o 'v[0-9.]+'\"", |
55 | 55 | command="nvidia-smi -q -x", |
56 | 56 | precheck_command="lshw -C display -json | jq -r '.[].vendor'", |
57 | | - precheck_regex="nvidia" |
| 57 | + precheck_regex="nvidia", |
58 | 58 | ) |
59 | 59 |
|
60 | 60 | virtualization = DockerTask( |
|
123 | 123 | priority=4, |
124 | 124 | image="ghcr.io/sparecores/benchmark:main", |
125 | 125 | 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 | + }, |
128 | 133 | mem_limit=int(mem_bytes * 0.85), |
129 | 134 | memswap_limit=int(mem_bytes * 0.85), |
130 | 135 | mem_swappiness=0, |
|
140 | 145 | minimum_memory=1, |
141 | 146 | timeout=timedelta(hours=1), |
142 | 147 | 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", |
144 | 149 | ) |
145 | 150 |
|
146 | 151 | bw_mem = DockerTask( |
|
175 | 180 | version_command="bash -c \"nvbandwidth --help | head -1 | egrep -o 'v[0-9.]+'\"", |
176 | 181 | command="nvbandwidth -j", |
177 | 182 | precheck_command="lshw -C display -json | jq -r '.[].vendor'", |
178 | | - precheck_regex="nvidia" |
| 183 | + precheck_regex="nvidia", |
179 | 184 | ) |
180 | 185 |
|
181 | 186 | passmark = DockerTask( |
|
193 | 198 | timeout=timedelta(hours=1), |
194 | 199 | minimum_memory=1, |
195 | 200 | priority=11, |
196 | | - rollout=0.05, # ~100 servers |
| 201 | + rollout=0.2, # ~400 servers |
197 | 202 | image="ghcr.io/sparecores/benchmark-llm:main", |
198 | 203 | command=None, |
199 | 204 | ) |
0 commit comments