hmz007
36ed224bac
Signed-off-by: hmz007 <hmz007@gmail.com> Change-Id: Ib87fdbe7a0be7dc961af3500fe9ea4589a127f9f
67 lines
1.7 KiB
C++
67 lines
1.7 KiB
C++
//
|
|
// 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()
|
|
&& m_CacheDataMap.at(hashValue).GetToken() == token
|
|
&& m_CacheDataMap.at(hashValue).GetCacheSize() == cacheSize)
|
|
{
|
|
ALOGV("CacheHandler::Register() Hash value has already registered.");
|
|
return;
|
|
}
|
|
CacheHandle cacheHandle(token, cacheSize);
|
|
m_CacheDataMap.insert({hashValue, cacheHandle});
|
|
}
|
|
|
|
bool CacheDataHandler::Validate(const HidlToken token, const size_t hashValue, const size_t cacheSize) const
|
|
{
|
|
return (m_CacheDataMap.find(hashValue) != m_CacheDataMap.end()
|
|
&& m_CacheDataMap.at(hashValue).GetToken() == token
|
|
&& m_CacheDataMap.at(hashValue).GetCacheSize() == cacheSize);
|
|
}
|
|
|
|
size_t CacheDataHandler::Hash(std::vector<uint8_t>& cacheData)
|
|
{
|
|
std::size_t hash = cacheData.size();
|
|
for (auto& i : cacheData)
|
|
{
|
|
hash = ((hash << 5) - hash) + 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
|