Skip to content

Commit 6365b79

Browse files
committed
Change hip-arch option to match others
1 parent 8166909 commit 6365b79

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

devops/scripts/benchmarks/main.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -446,13 +446,13 @@ def validate_and_parse_env_args(env_args):
446446
default=options.current_run_name,
447447
)
448448
parser.add_argument(
449-
"--cudnn_directory",
449+
"--cudnn-directory",
450450
type=str,
451451
help="Directory for cudnn library",
452452
default=None,
453453
)
454454
parser.add_argument(
455-
"--cublas_directory",
455+
"--cublas-directory",
456456
type=str,
457457
help="Directory for cublas library",
458458
default=None,
@@ -468,7 +468,9 @@ def validate_and_parse_env_args(env_args):
468468
type=int,
469469
help="Number of build jobs to run simultaneously",
470470
default=options.build_jobs,
471-
"--hip_arch",
471+
)
472+
parser.add_argument(
473+
"--hip-arch",
472474
type=str,
473475
help="HIP device architecture",
474476
default=None,

0 commit comments

Comments
 (0)