diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-11-26 10:38:11 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-11-26 10:38:11 +0000 |
commit | dea8fb6b96663de5d3df2f9fceb9bd09432fd7aa (patch) | |
tree | d041bf8e9d406c80ce089fc5b8d84b44381332e1 /src/backends/cl/ClBackendModelContext.cpp | |
parent | f4f150c30d3c34e9f26757ca43e4a2694b882bce (diff) | |
download | armnn-dea8fb6b96663de5d3df2f9fceb9bd09432fd7aa.tar.gz |
IVGCVSW-5481 'Add ClCompileContext to ClWorkloadFactory'
* Introduced CLCompileContext to ClWorkloadFactory
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Ied38f4336210502e5f518b9955ae6a5ba3d242b3
Diffstat (limited to 'src/backends/cl/ClBackendModelContext.cpp')
-rw-r--r-- | src/backends/cl/ClBackendModelContext.cpp | 29 |
1 files changed, 28 insertions, 1 deletions
diff --git a/src/backends/cl/ClBackendModelContext.cpp b/src/backends/cl/ClBackendModelContext.cpp index 0ef26b64d2..b685bc296c 100644 --- a/src/backends/cl/ClBackendModelContext.cpp +++ b/src/backends/cl/ClBackendModelContext.cpp @@ -17,13 +17,22 @@ bool ParseBool(const armnn::BackendOptions::Var& value, bool defaultValue) return defaultValue; } +std::string ParseFile(const armnn::BackendOptions::Var& value, std::string defaultValue) +{ + if (value.IsString()) + { + return value.AsString(); + } + return defaultValue; +} + } // namespace anonymous namespace armnn { ClBackendModelContext::ClBackendModelContext(const ModelOptions& modelOptions) - : m_IsFastMathEnabled(false) + : m_CachedNetworkFilePath(""), m_IsFastMathEnabled(false), m_SaveCachedNetwork(false) { if (!modelOptions.empty()) { @@ -33,13 +42,31 @@ ClBackendModelContext::ClBackendModelContext(const ModelOptions& modelOptions) { m_IsFastMathEnabled |= ParseBool(value, false); } + if (name == "SaveCachedNetwork") + { + m_SaveCachedNetwork |= ParseBool(value, false); + } + if (name == "CachedNetworkFilePath") + { + m_CachedNetworkFilePath = ParseFile(value, ""); + } }); } } +std::string ClBackendModelContext::GetCachedNetworkFilePath() const +{ + return m_CachedNetworkFilePath; +} + bool ClBackendModelContext::IsFastMathEnabled() const { return m_IsFastMathEnabled; } +bool ClBackendModelContext::SaveCachedNetwork() const +{ + return m_SaveCachedNetwork; +} + } // namespace armnn
\ No newline at end of file |