From a5f57f43c260536cbb3d2ae4806327d2f1d2f05e Mon Sep 17 00:00:00 2001 From: jywu-msft <43355415+jywu-msft@users.noreply.github.com> Date: Mon, 22 Jul 2019 12:22:49 -0700 Subject: [PATCH] fix mkldnn linux build break. (#1458) --- .../core/providers/mkldnn/mkldnn_execution_provider.cc | 5 ++--- .../core/providers/mkldnn/mkldnn_execution_provider.h | 3 +-- 2 files changed, 3 insertions(+), 5 deletions(-) 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