From ace132f9aaad6d14c37223aa1ec0c78673d52eb0 Mon Sep 17 00:00:00 2001 From: Changming Sun Date: Thu, 5 Dec 2019 15:03:22 -0800 Subject: [PATCH] Fix android build (#2558) --- .../core/providers/nnapi/nnapi_execution_provider.cc | 6 ++++-- onnxruntime/test/providers/nnapi/nnapi_basic_test.cc | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/onnxruntime/core/providers/nnapi/nnapi_execution_provider.cc b/onnxruntime/core/providers/nnapi/nnapi_execution_provider.cc index a92b9caab0..9d04f55c43 100644 --- a/onnxruntime/core/providers/nnapi/nnapi_execution_provider.cc +++ b/onnxruntime/core/providers/nnapi/nnapi_execution_provider.cc @@ -42,7 +42,8 @@ NnapiExecutionProvider::GetCapability(const onnxruntime::GraphViewer& graph, const std::vector& /*kernel_registries*/) const { // This method is based on that of TRT EP // Construct modelproto from graph - onnxruntime::Model model(graph.Name(), true, ModelMetaData(), IOnnxRuntimeOpSchemaRegistryList(), graph.DomainToVersionMap()); + onnxruntime::Model model(graph.Name(), true, ModelMetaData(), IOnnxRuntimeOpSchemaRegistryList(), graph.DomainToVersionMap(), + std::vector(), *GetLogger()); onnxruntime::Graph& graph_build = model.MainGraph(); const std::vector& node_index = graph.GetNodesInTopologicalOrder(); std::set all_node_inputs; @@ -202,7 +203,8 @@ common::Status NnapiExecutionProvider::Compile(const std::vectorBody(); onnxruntime::Model model(graph_body.Name(), true, ModelMetaData(), - IOnnxRuntimeOpSchemaRegistryList(), graph_body.DomainToVersionMap()); + IOnnxRuntimeOpSchemaRegistryList(), graph_body.DomainToVersionMap(), + std::vector(), *GetLogger()); ONNX_NAMESPACE::ModelProto model_proto = model.ToProto(); *(model_proto.mutable_graph()) = graph_body.ToGraphProto(); model_proto.set_ir_version(ONNX_NAMESPACE::Version::IR_VERSION); diff --git a/onnxruntime/test/providers/nnapi/nnapi_basic_test.cc b/onnxruntime/test/providers/nnapi/nnapi_basic_test.cc index 90fc3c1d52..dc497bad25 100644 --- a/onnxruntime/test/providers/nnapi/nnapi_basic_test.cc +++ b/onnxruntime/test/providers/nnapi/nnapi_basic_test.cc @@ -23,7 +23,7 @@ void VerifyOutputs(const std::vector& fetches, const std::vector inputs; std::vector outputs;