diff options
author | Ellen Norris-Thompson <ellen.norris-thompson@arm.com> | 2019-08-21 15:21:14 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-08-22 08:45:13 +0100 |
commit | 976ad3e257c4a68e809f5a7fe3b7a99bb3f1fba4 (patch) | |
tree | 683a03f831a8d4706ab2a34d155b4b9c3bd0fc44 /test/DriverTestHelpers.cpp | |
parent | 1ba9989c4d4fe152d2cb02c83c2da1c18b6aaf37 (diff) | |
download | android-nn-driver-976ad3e257c4a68e809f5a7fe3b7a99bb3f1fba4.tar.gz |
IVGCVSW-3613 Refactoring Android NN driver for QuantisedLstm test
* Templated AddPoolAndGetData and AddPoolAndSetData to accept different types
* Moved scale and offset to arguments of Add*Operand functions
Signed-off-by: Ellen Norris-Thompson <ellen.norris-thompson@arm.com>
Change-Id: Iaf4c2d5f4183ea54799bedae20950b23b914a727
Diffstat (limited to 'test/DriverTestHelpers.cpp')
-rw-r--r-- | test/DriverTestHelpers.cpp | 27 |
1 files changed, 0 insertions, 27 deletions
diff --git a/test/DriverTestHelpers.cpp b/test/DriverTestHelpers.cpp index 4c26174b..3a3c98ff 100644 --- a/test/DriverTestHelpers.cpp +++ b/test/DriverTestHelpers.cpp @@ -102,33 +102,6 @@ hidl_memory allocateSharedMemory(int64_t size) return memory; } -android::sp<IMemory> AddPoolAndGetData(uint32_t size, Request& request) -{ - hidl_memory pool; - - android::sp<IAllocator> allocator = IAllocator::getService("ashmem"); - allocator->allocate(sizeof(float) * size, [&](bool success, const hidl_memory& mem) { - BOOST_TEST(success); - pool = mem; - }); - - request.pools.resize(request.pools.size() + 1); - request.pools[request.pools.size() - 1] = pool; - - android::sp<IMemory> mapped = mapMemory(pool); - mapped->update(); - return mapped; -} - -void AddPoolAndSetData(uint32_t size, Request& request, const float* data) -{ - android::sp<IMemory> memory = AddPoolAndGetData(size, request); - - float* dst = static_cast<float*>(static_cast<void*>(memory->getPointer())); - - memcpy(dst, data, size * sizeof(float)); -} - android::sp<V1_0::IPreparedModel> PrepareModelWithStatus(const V1_0::Model& model, armnn_driver::ArmnnDriver& driver, ErrorStatus& prepareStatus, |