diff --git a/onnxruntime/core/providers/mkldnn/mkldnn_execution_provider.cc b/onnxruntime/core/providers/mkldnn/mkldnn_execution_provider.cc index 501b4efab3..d58b507a96 100644 --- a/onnxruntime/core/providers/mkldnn/mkldnn_execution_provider.cc +++ b/onnxruntime/core/providers/mkldnn/mkldnn_execution_provider.cc @@ -111,8 +111,7 @@ std::shared_ptr MKLDNNExecutionProvider::GetKernelRegistry() con return kernel_registry; } -bool MKLDNNExecutionProvider::UseSubgraph(const onnxruntime::GraphViewer& graph_viewer, - const std::vector& kernel_registries) const { +bool MKLDNNExecutionProvider::UseSubgraph(const onnxruntime::GraphViewer& graph_viewer) const { // switch between mkldnn-vanilla and mkldnn-subgraph implementation using // MKLDNN_SUBGRAPH environment variable bool use_subgraph = true; @@ -224,7 +223,7 @@ std::vector> MKLDNNExecutionProvider::GetCapa // temporary switch to toggle between mkldnn-vanilla and mkldnn-subgraph implementation using // ORT_MKLDNN_SUBGRAPH environment variable - if (UseSubgraph(graph_viewer, kernel_registries) == false) { + if (UseSubgraph(graph_viewer) == false) { return IExecutionProvider::GetCapability(graph_viewer, kernel_registries); } diff --git a/onnxruntime/core/providers/mkldnn/mkldnn_execution_provider.h b/onnxruntime/core/providers/mkldnn/mkldnn_execution_provider.h index 53dcdf45c6..2869698568 100644 --- a/onnxruntime/core/providers/mkldnn/mkldnn_execution_provider.h +++ b/onnxruntime/core/providers/mkldnn/mkldnn_execution_provider.h @@ -98,8 +98,7 @@ class MKLDNNExecutionProvider : public IExecutionProvider { return graph_name; } - bool UseSubgraph(const onnxruntime::GraphViewer& graph_viewer, - const std::vector& kernel_registries) const; + bool UseSubgraph(const onnxruntime::GraphViewer& graph_viewer) const; // Some dimensions are not supported by MKL-DNN // example: Pool with NumDimensions <= 3 is not supported