File tree Expand file tree Collapse file tree 1 file changed +15
-1
lines changed
devops/scripts/benchmarks Expand file tree Collapse file tree 1 file changed +15
-1
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments