diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-10-06 16:41:44 +0100 |
---|---|---|
committer | mike.kelly <mike.kelly@arm.com> | 2021-11-04 15:16:10 +0000 |
commit | 0a2dfabd76a45c58d0a14567f0503369c4e6fbf3 (patch) | |
tree | 035340e9f663d599f83992846e1772b161640654 /CacheDataHandler.cpp | |
parent | 1b46d132a3330692fcf9a603b21363a28f46ef03 (diff) | |
download | android-nn-driver-0a2dfabd76a45c58d0a14567f0503369c4e6fbf3.tar.gz |
IVGCVSW-5636 'Implement NNAPI caching functions'
* Cached serialized ArmNN model.
!armnn:6384
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: I78120a7f8ea892a28c0ff25f1b54e67a4f912574
Diffstat (limited to 'CacheDataHandler.cpp')
-rw-r--r-- | CacheDataHandler.cpp | 63 |
1 files changed, 63 insertions, 0 deletions
diff --git a/CacheDataHandler.cpp b/CacheDataHandler.cpp new file mode 100644 index 00000000..36881629 --- /dev/null +++ b/CacheDataHandler.cpp @@ -0,0 +1,63 @@ +// +// Copyright © 2021 Arm Ltd and Contributors. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#include "CacheDataHandler.hpp" + +#include <log/log.h> + +namespace armnn_driver +{ + +CacheDataHandler& CacheDataHandlerInstance() +{ + static CacheDataHandler instance; + return instance; +} + +void CacheDataHandler::Register(const HidlToken token, const size_t hashValue, const size_t cacheSize) +{ + if (m_CacheDataMap.find(hashValue) != m_CacheDataMap.end()) + { + ALOGV("CacheHandler::Register() Token has been already registered."); + return; + } + CacheHandle cacheHandle(token, cacheSize); + m_CacheDataMap.insert({hashValue, cacheHandle}); +} + +bool CacheDataHandler::Validate(const HidlToken token, const size_t hashValue) const +{ + return (m_CacheDataMap.find(hashValue) != m_CacheDataMap.end() + && m_CacheDataMap.at(hashValue).GetToken() == token); +} + +size_t CacheDataHandler::Hash(std::vector<uint8_t>& cacheData) +{ + std::size_t hash = cacheData.size(); + for (auto& i : cacheData) + { + hash ^= std::hash<unsigned int>{}(i); + } + return hash; +} + +size_t CacheDataHandler::GetCacheSize(HidlToken token) +{ + for (auto i = m_CacheDataMap.begin(); i != m_CacheDataMap.end(); ++i) + { + if (i->second.GetToken() == token) + { + return i->second.GetCacheSize(); + } + } + return 0; +} + +void CacheDataHandler::Clear() +{ + m_CacheDataMap.clear(); +} + +} // armnn_driver |