Skip to content
This repository was archived by the owner on Apr 28, 2023. It is now read-only.

Commit 0072edf

Browse files
author
Nicolas Vasilache
committed
Fix python options post refactoring: tuner_gpus -> tuner_devices
1 parent 0901ad7 commit 0072edf

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

benchmarks_python/caffe2_benchmark.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def GetArgumentParser():
5151
parser.add_argument("--tuner_cache_file", type=str,
5252
default="tuner_cache",
5353
help="File to store tuned mapping options")
54-
parser.add_argument("--tuner_gpus", type=str,
54+
parser.add_argument("--tuner_devices", type=str,
5555
default="0",
5656
help="String representation of gpus to use for tuning (e.g. \"0,1\")")
5757
parser.add_argument("--tuner_threads", type=int, default=10,
@@ -70,7 +70,7 @@ def main():
7070
core.GlobalInit([
7171
'tc_bench',
7272
'--caffe2_logging_operator_dyno_sampling_rate=0',
73-
'--tuner_gpus=' + args.tuner_gpus,
73+
'--tuner_devices=' + args.tuner_devices,
7474
'--caffe2_simple_net_benchmark_run_whole_net=0',
7575
] + extra_args)
7676
mapping_options = tune(args)

0 commit comments

Comments
 (0)