diff options
| author | Finn Williams <finn.williams@arm.com> | 2022-03-30 12:09:47 +0100 |
|---|---|---|
| committer | Finn Williams <finn.williams@arm.com> | 2022-04-04 16:59:15 +0100 |
| commit | 8dc56289c8f6acddbbd17a7a77b06d333369a7f8 (patch) | |
| tree | 298882d9ae9d230fb7e19ff70058b4443604516e /shim | |
| parent | de1cd79500a3e6b8744d00fbf54abaae741d881f (diff) | |
| download | armnn-8dc56289c8f6acddbbd17a7a77b06d333369a7f8.tar.gz | |
IVGCVSW-6754 Integrate Import Host Memory to arm-armnn-sl
Signed-off-by: Finn Williams <finn.williams@arm.com>
Change-Id: I1d465725e56f61034cb2425157bb594b4ceab9fa
Diffstat (limited to 'shim')
| -rw-r--r-- | shim/sl/canonical/ArmnnPreparedModel.cpp | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/shim/sl/canonical/ArmnnPreparedModel.cpp b/shim/sl/canonical/ArmnnPreparedModel.cpp index 94809497d1..35edfb7d99 100644 --- a/shim/sl/canonical/ArmnnPreparedModel.cpp +++ b/shim/sl/canonical/ArmnnPreparedModel.cpp @@ -385,7 +385,15 @@ ErrorStatus ArmnnPreparedModel::ExecuteGraph( } armnn::Status status; VLOG(DRIVER) << "ArmnnPreparedModel::ExecuteGraph m_AsyncModelExecutionEnabled false"; - status = m_Runtime->EnqueueWorkload(m_NetworkId, inputTensors, outputTensors); + + std::vector<armnn::ImportedInputId> importedInputIds; + importedInputIds = m_Runtime->ImportInputs(m_NetworkId, inputTensors, armnn::MemorySource::Malloc); + + std::vector<armnn::ImportedOutputId> importedOutputIds; + importedOutputIds = m_Runtime->ImportOutputs(m_NetworkId, outputTensors, armnn::MemorySource::Malloc); + + status = m_Runtime->EnqueueWorkload(m_NetworkId, inputTensors, outputTensors, + importedInputIds, importedOutputIds); if (ctx.measureTimings == MeasureTiming::YES) { |