diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2022-08-02 09:17:23 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2022-08-05 15:51:44 +0100 |
commit | 361679da2a86b6794827edf48a3f585738f8e08a (patch) | |
tree | 39fe40e4918ccdf96241307a228d8010ba62cd8a /shim/sl/canonical/CacheDataHandler.hpp | |
parent | 8950f8a59dbfea302dfaa2a5df320bee5d877745 (diff) | |
download | armnn-361679da2a86b6794827edf48a3f585738f8e08a.tar.gz |
IVGCVSW-7063 'Support Library NNAPI Caching'
* Fixed caching issue.
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Ic7b3e0bd4438b2fd1b3dbfa86b6c89d625bbf9dd
Diffstat (limited to 'shim/sl/canonical/CacheDataHandler.hpp')
-rw-r--r-- | shim/sl/canonical/CacheDataHandler.hpp | 64 |
1 files changed, 0 insertions, 64 deletions
diff --git a/shim/sl/canonical/CacheDataHandler.hpp b/shim/sl/canonical/CacheDataHandler.hpp deleted file mode 100644 index 95464a9809..0000000000 --- a/shim/sl/canonical/CacheDataHandler.hpp +++ /dev/null @@ -1,64 +0,0 @@ -// -// Copyright © 2022 Arm Ltd and Contributors. All rights reserved. -// SPDX-License-Identifier: MIT -// - -#pragma once - -#include <vector> -#include <unordered_map> - -#include <nnapi/Types.h> - -namespace armnn_driver -{ - -class CacheHandle -{ -public: - CacheHandle(const android::nn::CacheToken token, const size_t cacheSize) - : m_CacheToken(token), m_CacheSize(cacheSize) {} - - ~CacheHandle() {}; - - android::nn::CacheToken GetToken() const - { - return m_CacheToken; - } - - size_t GetCacheSize() const - { - return m_CacheSize; - } - -private: - const android::nn::CacheToken m_CacheToken; - const size_t m_CacheSize; -}; - -class CacheDataHandler -{ -public: - CacheDataHandler() {} - ~CacheDataHandler() {} - - void Register(const android::nn::CacheToken token, const size_t hashValue, const size_t cacheSize); - - bool Validate(const android::nn::CacheToken token, const size_t hashValue, const size_t cacheSize) const; - - size_t Hash(std::vector<uint8_t>& cacheData); - - size_t GetCacheSize(android::nn::CacheToken token); - - void Clear(); - -private: - CacheDataHandler(const CacheDataHandler&) = delete; - CacheDataHandler& operator=(const CacheDataHandler&) = delete; - - std::unordered_map<size_t, CacheHandle> m_CacheDataMap; -}; - -CacheDataHandler& CacheDataHandlerInstance(); - -} // armnn_driver |