From eab7c0d5bfbe69895bfca6715d0aee216d0ed0a9 Mon Sep 17 00:00:00 2001 From: Boris Fomitchev Date: Thu, 31 Mar 2022 11:05:49 -0700 Subject: [PATCH] Fixing optimizer failure due to missing provider list (#10497) Signed-off-by: Boris Fomitchev --- onnxruntime/python/tools/transformers/optimizer.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/onnxruntime/python/tools/transformers/optimizer.py b/onnxruntime/python/tools/transformers/optimizer.py index 55699ae7b0..d865381e00 100644 --- a/onnxruntime/python/tools/transformers/optimizer.py +++ b/onnxruntime/python/tools/transformers/optimizer.py @@ -93,7 +93,9 @@ def optimize_by_onnxruntime(onnx_model_path: str, providers=['CPUExecutionProvider'], **kwargs) else: - session = onnxruntime.InferenceSession(onnx_model_path, sess_options, **kwargs) + session = onnxruntime.InferenceSession(onnx_model_path, sess_options, + providers=['CUDAExecutionProvider'], + **kwargs) assert 'CUDAExecutionProvider' in session.get_providers() # Make sure there is GPU assert os.path.exists(optimized_model_path) and os.path.isfile(optimized_model_path)