diff options
author | Finn Williams <Finn.Williams@arm.com> | 2021-05-31 14:57:15 +0100 |
---|---|---|
committer | Finn Williams <Finn.Williams@arm.com> | 2021-05-31 14:57:15 +0100 |
commit | 8fde84b92c211c6499c6ea588886861215835c17 (patch) | |
tree | 416d2fb6a128f4b3e2cb7873521906c63d22ebeb | |
parent | d8fb540568b29fd1d81a1cca667a1ad3e33ef5a1 (diff) | |
download | android-nn-driver-8fde84b92c211c6499c6ea588886861215835c17.tar.gz |
IVGCVSW-6081 Fix hal 1.0 async execution
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I4037e3bb47ed0f5585663ca56eed2866bc8a9ef7
-rw-r--r-- | ArmnnPreparedModel.cpp | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/ArmnnPreparedModel.cpp b/ArmnnPreparedModel.cpp index 978f3787..259d1b8c 100644 --- a/ArmnnPreparedModel.cpp +++ b/ArmnnPreparedModel.cpp @@ -346,7 +346,17 @@ bool ArmnnPreparedModel<HalVersion>::ExecuteWithDummyInputs() try { - armnn::Status status = m_Runtime->EnqueueWorkload(m_NetworkId, inputTensors, outputTensors); + armnn::Status status; + if (m_AsyncModelExecutionEnabled) + { + ALOGW("ArmnnPreparedModel::ExecuteGraph m_AsyncModelExecutionEnabled true"); + status = m_Runtime->Execute(*m_WorkingMemHandle, inputTensors, outputTensors); + } + else + { + ALOGW("ArmnnPreparedModel::ExecuteGraph m_AsyncModelExecutionEnabled false"); + status = m_Runtime->EnqueueWorkload(m_NetworkId, inputTensors, outputTensors); + } if (status != armnn::Status::Success) { ALOGW("ExecuteWithDummyInputs: EnqueueWorkload failed"); |