Skip to content

Commit 8a1f142

Browse files
authored
add shared memory size option (#646)
* add shm size to inline params * fix bug
1 parent 5c3b174 commit 8a1f142

File tree

2 files changed

+19
-1
lines changed

2 files changed

+19
-1
lines changed

cli/medperf/config.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -235,10 +235,21 @@
235235
"evaluate_timeout",
236236
"platform",
237237
"gpus",
238+
"shm_size",
238239
"cleanup",
239240
"container_loglevel",
240241
]
241-
configurable_parameters = inline_parameters + [
242+
configurable_parameters = [
243+
"loglevel",
244+
"prepare_timeout",
245+
"sanity_check_timeout",
246+
"statistics_timeout",
247+
"infer_timeout",
248+
"evaluate_timeout",
249+
"platform",
250+
"gpus",
251+
"cleanup",
252+
"container_loglevel",
242253
"server",
243254
"certificate",
244255
"auth_class",

cli/medperf/decorators.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -219,6 +219,13 @@ def wrapper(
219219
- Form "device=<id1>,<id2>": to expose specific GPUs.
220220
(e.g., --gpus="device=0,2")\n""",
221221
),
222+
shm_size: str = typer.Option(
223+
config.shm_size,
224+
"--shm-size",
225+
help="""
226+
Only for Docker. See --shm-size argument
227+
in docker run: https://docs.docker.com/engine/containers/run/""",
228+
),
222229
cleanup: bool = typer.Option(
223230
config.cleanup,
224231
"--cleanup/--no-cleanup",

0 commit comments

Comments
 (0)