From 216ebcaa50fb132630126a4071f5ba4cbee0e47b Mon Sep 17 00:00:00 2001 From: Maxime Desroches Date: Wed, 11 Dec 2024 13:27:39 -0800 Subject: [PATCH] Fix model runtime on PC (#34210) exhaustive --- selfdrive/modeld/runners/onnxmodel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/selfdrive/modeld/runners/onnxmodel.py b/selfdrive/modeld/runners/onnxmodel.py index 2b17422740..2a870392d5 100644 --- a/selfdrive/modeld/runners/onnxmodel.py +++ b/selfdrive/modeld/runners/onnxmodel.py @@ -49,7 +49,7 @@ def create_ort_session(path, fp16_to_fp32): provider = 'OpenVINOExecutionProvider' elif 'CUDAExecutionProvider' in ort.get_available_providers() and 'ONNXCPU' not in os.environ: options.intra_op_num_threads = 2 - provider = ('CUDAExecutionProvider', {'cudnn_conv_algo_search': 'DEFAULT'}) + provider = ('CUDAExecutionProvider', {'cudnn_conv_algo_search': 'EXHAUSTIVE'}) else: options.intra_op_num_threads = 2 options.execution_mode = ort.ExecutionMode.ORT_SEQUENTIAL