xref: /aosp_15_r20/external/android-nn-driver/1.0/ArmnnDriverImpl.cpp (revision 3e777be0405cee09af5d5785ff37f7cfb5bee59a)

//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//

#include "ArmnnDriverImpl.hpp"
#include "../SystemPropertiesUtils.hpp"

#include <log/log.h>

namespace
{

const char *g_Float32PerformanceExecTimeName      = "ArmNN.float32Performance.execTime";
const char *g_Float32PerformancePowerUsageName    = "ArmNN.float32Performance.powerUsage";
const char *g_Quantized8PerformanceExecTimeName   = "ArmNN.quantized8Performance.execTime";
const char *g_Quantized8PerformancePowerUsageName = "ArmNN.quantized8Performance.powerUsage";

} // anonymous namespace

namespace armnn_driver
{
namespace hal_1_0
{

Return<void> ArmnnDriverImpl::getCapabilities(const armnn::IRuntimePtr& runtime,
                                              V1_0::IDevice::getCapabilities_cb cb)
{
    ALOGV("hal_1_0::ArmnnDriverImpl::getCapabilities()");

    V1_0::Capabilities capabilities;
    if (runtime)
    {
        capabilities.float32Performance.execTime =
            ParseSystemProperty(g_Float32PerformanceExecTimeName, .1f);

        capabilities.float32Performance.powerUsage =
            ParseSystemProperty(g_Float32PerformancePowerUsageName, .1f);

        capabilities.quantized8Performance.execTime =
            ParseSystemProperty(g_Quantized8PerformanceExecTimeName, .1f);

        capabilities.quantized8Performance.powerUsage =
            ParseSystemProperty(g_Quantized8PerformancePowerUsageName, .1f);

        cb(V1_0::ErrorStatus::NONE, capabilities);
    }
    else
    {
        capabilities.float32Performance.execTime      = 0;
        capabilities.float32Performance.powerUsage    = 0;
        capabilities.quantized8Performance.execTime   = 0;
        capabilities.quantized8Performance.powerUsage = 0;

        cb(V1_0::ErrorStatus::DEVICE_UNAVAILABLE, capabilities);
    }

    return Void();
}

} // namespace hal_1_0
} // namespace armnn_driver