Skip to content

Commit 1ef9251

Browse files
committed
Correct bad conflict resolution over cudnn/cublas flags
1 parent 0ff0142 commit 1ef9251

File tree

1 file changed

+15
-1
lines changed

1 file changed

+15
-1
lines changed

devops/scripts/benchmarks/main.py

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -447,13 +447,25 @@ def validate_and_parse_env_args(env_args):
447447
type=str,
448448
help="The name of the results which should be used as a baseline for metrics calculation",
449449
default=options.current_run_name,
450+
)
451+
parser.add_argument(
452+
"--cudnn_directory",
453+
type=str,
454+
help="Directory for cudnn library",
455+
default=None,
456+
)
457+
parser.add_argument(
458+
"--cublas_directory",
459+
type=str,
460+
help="Directory for cublas library",
461+
default=None,
450462
)
451463
parser.add_argument(
452464
"--results-dir",
453465
type=str,
454466
help="Specify a custom results directory",
455467
default=options.custom_results_dir,
456-
)
468+
)
457469
parser.add_argument(
458470
"--build-jobs",
459471
type=int,
@@ -484,6 +496,8 @@ def validate_and_parse_env_args(env_args):
484496
options.iterations_stddev = args.iterations_stddev
485497
options.build_igc = args.build_igc
486498
options.current_run_name = args.relative_perf
499+
options.cudnn_directory = args.cudnn_directory
500+
options.cublas_directory = args.cublas_directory
487501
options.custom_results_dir = args.results_dir
488502
options.build_jobs = args.build_jobs
489503

0 commit comments

Comments
 (0)