xref: /aosp_15_r20/external/android-nn-driver/CacheDataHandler.cpp (revision 3e777be0405cee09af5d5785ff37f7cfb5bee59a)
1 *3e777be0SXin Li //
2 *3e777be0SXin Li // Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
3 *3e777be0SXin Li // SPDX-License-Identifier: MIT
4 *3e777be0SXin Li //
5 *3e777be0SXin Li 
6 *3e777be0SXin Li #include "CacheDataHandler.hpp"
7 *3e777be0SXin Li 
8 *3e777be0SXin Li #include <log/log.h>
9 *3e777be0SXin Li 
10 *3e777be0SXin Li namespace armnn_driver
11 *3e777be0SXin Li {
12 *3e777be0SXin Li 
CacheDataHandlerInstance()13 *3e777be0SXin Li CacheDataHandler& CacheDataHandlerInstance()
14 *3e777be0SXin Li {
15 *3e777be0SXin Li     static CacheDataHandler instance;
16 *3e777be0SXin Li     return instance;
17 *3e777be0SXin Li }
18 *3e777be0SXin Li 
Register(const HidlToken token,const size_t hashValue,const size_t cacheSize)19 *3e777be0SXin Li void CacheDataHandler::Register(const HidlToken token, const size_t hashValue, const size_t cacheSize)
20 *3e777be0SXin Li {
21 *3e777be0SXin Li     if (m_CacheDataMap.find(hashValue) != m_CacheDataMap.end()
22 *3e777be0SXin Li                         && m_CacheDataMap.at(hashValue).GetToken() == token
23 *3e777be0SXin Li                         && m_CacheDataMap.at(hashValue).GetCacheSize() == cacheSize)
24 *3e777be0SXin Li     {
25 *3e777be0SXin Li         ALOGV("CacheHandler::Register() Hash value has already registered.");
26 *3e777be0SXin Li         return;
27 *3e777be0SXin Li     }
28 *3e777be0SXin Li     CacheHandle cacheHandle(token, cacheSize);
29 *3e777be0SXin Li     m_CacheDataMap.insert({hashValue, cacheHandle});
30 *3e777be0SXin Li }
31 *3e777be0SXin Li 
Validate(const HidlToken token,const size_t hashValue,const size_t cacheSize) const32 *3e777be0SXin Li bool CacheDataHandler::Validate(const HidlToken token, const size_t hashValue, const size_t cacheSize) const
33 *3e777be0SXin Li {
34 *3e777be0SXin Li     return (m_CacheDataMap.find(hashValue) != m_CacheDataMap.end()
35 *3e777be0SXin Li                              && m_CacheDataMap.at(hashValue).GetToken() == token
36 *3e777be0SXin Li                              && m_CacheDataMap.at(hashValue).GetCacheSize() == cacheSize);
37 *3e777be0SXin Li }
38 *3e777be0SXin Li 
Hash(std::vector<uint8_t> & cacheData)39 *3e777be0SXin Li size_t CacheDataHandler::Hash(std::vector<uint8_t>& cacheData)
40 *3e777be0SXin Li {
41 *3e777be0SXin Li     std::size_t hash = cacheData.size();
42 *3e777be0SXin Li     for (auto& i : cacheData)
43 *3e777be0SXin Li     {
44 *3e777be0SXin Li         hash = ((hash << 5) - hash) + i;
45 *3e777be0SXin Li     }
46 *3e777be0SXin Li     return hash;
47 *3e777be0SXin Li }
48 *3e777be0SXin Li 
GetCacheSize(HidlToken token)49 *3e777be0SXin Li size_t CacheDataHandler::GetCacheSize(HidlToken token)
50 *3e777be0SXin Li {
51 *3e777be0SXin Li     for (auto i = m_CacheDataMap.begin(); i != m_CacheDataMap.end(); ++i)
52 *3e777be0SXin Li     {
53 *3e777be0SXin Li         if (i->second.GetToken() == token)
54 *3e777be0SXin Li         {
55 *3e777be0SXin Li             return i->second.GetCacheSize();
56 *3e777be0SXin Li         }
57 *3e777be0SXin Li     }
58 *3e777be0SXin Li     return 0;
59 *3e777be0SXin Li }
60 *3e777be0SXin Li 
Clear()61 *3e777be0SXin Li void CacheDataHandler::Clear()
62 *3e777be0SXin Li {
63 *3e777be0SXin Li     m_CacheDataMap.clear();
64 *3e777be0SXin Li }
65 *3e777be0SXin Li 
66 *3e777be0SXin Li } // armnn_driver
67