ClBackendModelContext.cpp

Go to the documentation of this file.

//
// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
 
#include "ClBackendModelContext.hpp"
 
namespace
{
 
bool ParseBool(const armnn::BackendOptions::Var& value, bool defaultValue)
{
    if (value.IsBool())
    {
        return value.AsBool();
    }
    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_CachedNetworkFilePath(""), m_IsFastMathEnabled(false), m_SaveCachedNetwork(false), m_CachedFileDescriptor(-1)
{
   if (!modelOptions.empty())
   {
       ParseOptions(modelOptions, "GpuAcc", [&](std::string name, const BackendOptions::Var& value)
       {
           if (name == "FastMathEnabled")
           {
               m_IsFastMathEnabled |= ParseBool(value, false);
           }
           if (name == "SaveCachedNetwork")
           {
               m_SaveCachedNetwork |= ParseBool(value, false);
           }
           if (name == "CachedNetworkFilePath")
           {
               m_CachedNetworkFilePath = ParseFile(value, "");
           }
           if (name == "CachedFileDescriptor")
           {
               m_CachedFileDescriptor = armnn::ParseIntBackendOption(value, -1);
           }
       });
   }
}
 
std::string ClBackendModelContext::GetCachedNetworkFilePath() const
{
    return m_CachedNetworkFilePath;
}
 
bool ClBackendModelContext::IsFastMathEnabled() const
{
    return m_IsFastMathEnabled;
}
 
bool ClBackendModelContext::SaveCachedNetwork() const
{
    return m_SaveCachedNetwork;
}
 
int ClBackendModelContext::GetCachedFileDescriptor() const
{
    return m_CachedFileDescriptor;
}
 
} // namespace armnn