@@ -3489,10 +3489,16 @@ def test_compare_to_io_binding(self, model_arch):
3489
3489
3490
3490
model_id = MODEL_NAMES [model_arch ]
3491
3491
onnx_model = ORTModelForCTC .from_pretrained (
3492
- self .onnx_model_dirs [model_arch ], use_io_binding = False , provider = "CUDAExecutionProvider"
3492
+ self .onnx_model_dirs [model_arch ],
3493
+ use_io_binding = False ,
3494
+ provider = "CUDAExecutionProvider" ,
3495
+ provider_options = {"cudnn_conv_algo_search" : "DEFAULT" },
3493
3496
)
3494
3497
io_model = ORTModelForCTC .from_pretrained (
3495
- self .onnx_model_dirs [model_arch ], use_io_binding = True , provider = "CUDAExecutionProvider"
3498
+ self .onnx_model_dirs [model_arch ],
3499
+ use_io_binding = True ,
3500
+ provider = "CUDAExecutionProvider" ,
3501
+ provider_options = {"cudnn_conv_algo_search" : "DEFAULT" },
3496
3502
)
3497
3503
3498
3504
self .assertFalse (onnx_model .use_io_binding )
@@ -4713,10 +4719,20 @@ def test_compare_to_io_binding(self, test_name: str, model_arch: str, use_cache:
4713
4719
4714
4720
model_id = MODEL_NAMES [model_arch ]
4715
4721
onnx_model = ORTModelForSpeechSeq2Seq .from_pretrained (
4716
- self .onnx_model_dirs [test_name ], use_io_binding = False , provider = "CUDAExecutionProvider"
4722
+ self .onnx_model_dirs [test_name ],
4723
+ use_io_binding = False ,
4724
+ provider = "CUDAExecutionProvider" ,
4725
+ provider_options = {
4726
+ "cudnn_conv_algo_search" : "DEFAULT" ,
4727
+ },
4717
4728
)
4718
4729
io_model = ORTModelForSpeechSeq2Seq .from_pretrained (
4719
- self .onnx_model_dirs [test_name ], use_io_binding = True , provider = "CUDAExecutionProvider"
4730
+ self .onnx_model_dirs [test_name ],
4731
+ use_io_binding = True ,
4732
+ provider = "CUDAExecutionProvider" ,
4733
+ provider_options = {
4734
+ "cudnn_conv_algo_search" : "DEFAULT" ,
4735
+ },
4720
4736
)
4721
4737
4722
4738
self .assertFalse (onnx_model .use_io_binding )
0 commit comments