File tree 1 file changed +5
-3
lines changed
devops/scripts/benchmarks
1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -446,13 +446,13 @@ def validate_and_parse_env_args(env_args):
446
446
default = options .current_run_name ,
447
447
)
448
448
parser .add_argument (
449
- "--cudnn_directory " ,
449
+ "--cudnn-directory " ,
450
450
type = str ,
451
451
help = "Directory for cudnn library" ,
452
452
default = None ,
453
453
)
454
454
parser .add_argument (
455
- "--cublas_directory " ,
455
+ "--cublas-directory " ,
456
456
type = str ,
457
457
help = "Directory for cublas library" ,
458
458
default = None ,
@@ -468,7 +468,9 @@ def validate_and_parse_env_args(env_args):
468
468
type = int ,
469
469
help = "Number of build jobs to run simultaneously" ,
470
470
default = options .build_jobs ,
471
- "--hip_arch" ,
471
+ )
472
+ parser .add_argument (
473
+ "--hip-arch" ,
472
474
type = str ,
473
475
help = "HIP device architecture" ,
474
476
default = None ,
You can’t perform that action at this time.
0 commit comments