hmz007
36ed224bac
Signed-off-by: hmz007 <hmz007@gmail.com> Change-Id: Ib87fdbe7a0be7dc961af3500fe9ea4589a127f9f
56 lines
1.6 KiB
C++
56 lines
1.6 KiB
C++
//
|
|
// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
//
|
|
|
|
#define LOG_TAG "ArmnnDriverTests"
|
|
#include <log/log.h>
|
|
|
|
#ifndef DOCTEST_CONFIG_IMPLEMENT_WITH_MAIN
|
|
#define DOCTEST_CONFIG_IMPLEMENT_WITH_MAIN
|
|
#endif
|
|
|
|
#include "DriverTestHelpers.hpp"
|
|
|
|
using namespace android::hardware;
|
|
using namespace driverTestHelpers;
|
|
using namespace armnn_driver;
|
|
|
|
DOCTEST_TEST_SUITE("DriverTests")
|
|
{
|
|
|
|
DOCTEST_TEST_CASE("Init")
|
|
{
|
|
// Making the driver object on the stack causes a weird libc error, so make it on the heap instead
|
|
auto driver = std::make_unique<ArmnnDriver>(DriverOptions(armnn::Compute::CpuRef));
|
|
|
|
V1_0::DeviceStatus status = driver->getStatus();
|
|
// Note double-parentheses to avoid compile error from doctest trying to printf the DeviceStatus
|
|
DOCTEST_CHECK((status == V1_0::DeviceStatus::AVAILABLE));
|
|
}
|
|
|
|
DOCTEST_TEST_CASE("TestCapabilities")
|
|
{
|
|
// Making the driver object on the stack causes a weird libc error, so make it on the heap instead
|
|
auto driver = std::make_unique<ArmnnDriver>(DriverOptions(armnn::Compute::CpuRef));
|
|
|
|
V1_0::ErrorStatus error;
|
|
V1_0::Capabilities cap;
|
|
|
|
auto cb = [&](V1_0::ErrorStatus status, const V1_0::Capabilities& capabilities)
|
|
{
|
|
error = status;
|
|
cap = capabilities;
|
|
};
|
|
|
|
driver->getCapabilities(cb);
|
|
|
|
DOCTEST_CHECK((int)error == (int)V1_0::ErrorStatus::NONE);
|
|
DOCTEST_CHECK(cap.float32Performance.execTime > 0.f);
|
|
DOCTEST_CHECK(cap.float32Performance.powerUsage > 0.f);
|
|
DOCTEST_CHECK(cap.quantized8Performance.execTime > 0.f);
|
|
DOCTEST_CHECK(cap.quantized8Performance.powerUsage > 0.f);
|
|
}
|
|
|
|
}
|