From 07dedda93954646d9ed8bf3d468a89634f88f112 Mon Sep 17 00:00:00 2001 From: jimfly01 Date: Tue, 9 Oct 2018 12:29:41 +0100 Subject: IVGCVSW-1929: Optimize function signature change * Optimize now returns reasons for failure !armnn:152481 Change-Id: I1cd7f5796b3c469e9baecbed80e392ac74e40ef4 --- ArmnnDriverImpl.cpp | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/ArmnnDriverImpl.cpp b/ArmnnDriverImpl.cpp index 5d213659..267574c1 100644 --- a/ArmnnDriverImpl.cpp +++ b/ArmnnDriverImpl.cpp @@ -150,12 +150,14 @@ Return ArmnnDriverImpl::prepareModel( armnn::OptimizerOptions OptOptions; OptOptions.m_ReduceFp32ToFp16 = float32ToFloat16; + std::vector errMessages; try { optNet = armnn::Optimize(*modelConverter.GetINetwork(), {options.GetComputeDevice()}, runtime->GetDeviceSpec(), - OptOptions); + OptOptions, + errMessages); } catch (armnn::Exception &e) { @@ -168,8 +170,13 @@ Return ArmnnDriverImpl::prepareModel( // Check that the optimized network is valid. if (!optNet) { + stringstream message; + message << "ArmnnDriverImpl::prepareModel: Invalid optimized network"; + for (const string& msg : errMessages) { + message << "\n" << msg; + } FailPrepareModel(ErrorStatus::GENERAL_FAILURE, - "ArmnnDriverImpl::prepareModel: Invalid optimized network", cb); + message.str(), cb); return ErrorStatus::NONE; } -- cgit v1.2.1