IVGCVSW-3987 Add armnn::Exception warning and std::exception error
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: I4963fb4d44501287e161f7366f38557962d06cac
diff --git a/ArmnnPreparedModel.cpp b/ArmnnPreparedModel.cpp
index 2f1abef..0899430 100644
--- a/ArmnnPreparedModel.cpp
+++ b/ArmnnPreparedModel.cpp
@@ -209,9 +209,15 @@
pOutputTensors->emplace_back(i, outputTensor);
}
}
+ catch (armnn::Exception& e)
+ {
+ ALOGW("armnn::Exception caught while preparing for EnqueueWorkload: %s", e.what());
+ NotifyCallbackAndCheck(callback, ErrorStatus::GENERAL_FAILURE, "ArmnnPreparedModel::execute");
+ return ErrorStatus::GENERAL_FAILURE;
+ }
catch (std::exception& e)
{
- ALOGW("Exception caught while preparing for EnqueueWorkload: %s", e.what());
+ ALOGE("std::exception caught while preparing for EnqueueWorkload: %s", e.what());
NotifyCallbackAndCheck(callback, ErrorStatus::GENERAL_FAILURE, "ArmnnPreparedModel::execute");
return ErrorStatus::GENERAL_FAILURE;
}
@@ -253,9 +259,15 @@
return;
}
}
+ catch (armnn::Exception& e)
+ {
+ ALOGW("armnn::Exception caught from EnqueueWorkload: %s", e.what());
+ cb.callback(ErrorStatus::GENERAL_FAILURE, "ArmnnPreparedModel::ExecuteGraph");
+ return;
+ }
catch (std::exception& e)
{
- ALOGW("Exception caught from EnqueueWorkload: %s", e.what());
+ ALOGE("std::exception caught from EnqueueWorkload: %s", e.what());
cb.callback(ErrorStatus::GENERAL_FAILURE, "ArmnnPreparedModel::ExecuteGraph");
return;
}
@@ -306,9 +318,14 @@
return false;
}
}
+ catch (armnn::Exception& e)
+ {
+ ALOGW("ExecuteWithDummyInputs: armnn::Exception caught from EnqueueWorkload: %s", e.what());
+ return false;
+ }
catch (std::exception& e)
{
- ALOGW("ExecuteWithDummyInputs: Exception caught from EnqueueWorkload: %s", e.what());
+ ALOGE("ExecuteWithDummyInputs: std::exception caught from EnqueueWorkload: %s", e.what());
return false;
}
return true;