From 95fa78df38ea4f4178fbffafba473c4684b23c45 Mon Sep 17 00:00:00 2001 From: Ryan Hill Date: Tue, 20 Apr 2021 22:40:38 -0700 Subject: [PATCH] Fix pybind --- onnxruntime/python/onnxruntime_pybind_state.cc | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/onnxruntime/python/onnxruntime_pybind_state.cc b/onnxruntime/python/onnxruntime_pybind_state.cc index 2219dd2282..4b961c8cc6 100644 --- a/onnxruntime/python/onnxruntime_pybind_state.cc +++ b/onnxruntime/python/onnxruntime_pybind_state.cc @@ -138,9 +138,7 @@ struct OrtStatus { #if defined(USE_CUDA) || defined(USE_ROCM) #ifdef USE_CUDA -#include "core/providers/cuda/shared_inc/cuda_call.h" -#include "core/providers/cuda/cuda_execution_provider.h" -#include "core/providers/cuda/cuda_allocator.h" +#include "core/providers/cuda/cuda_execution_provider_info.h" // TODO remove deprecated global config OrtCudnnConvAlgoSearch cudnn_conv_algo_search = OrtCudnnConvAlgoSearch::EXHAUSTIVE; // TODO remove deprecated global config