1*89c4ff92SAndroid Build Coastguard Worker // 2*89c4ff92SAndroid Build Coastguard Worker // Copyright © 2017 Arm Ltd. All rights reserved. 3*89c4ff92SAndroid Build Coastguard Worker // SPDX-License-Identifier: MIT 4*89c4ff92SAndroid Build Coastguard Worker // 5*89c4ff92SAndroid Build Coastguard Worker 6*89c4ff92SAndroid Build Coastguard Worker #include <armnn/BackendId.hpp> 7*89c4ff92SAndroid Build Coastguard Worker #include <armnn/Types.hpp> 8*89c4ff92SAndroid Build Coastguard Worker 9*89c4ff92SAndroid Build Coastguard Worker #include <doctest/doctest.h> 10*89c4ff92SAndroid Build Coastguard Worker 11*89c4ff92SAndroid Build Coastguard Worker using namespace armnn; 12*89c4ff92SAndroid Build Coastguard Worker 13*89c4ff92SAndroid Build Coastguard Worker TEST_SUITE("BackendIdTests") 14*89c4ff92SAndroid Build Coastguard Worker { 15*89c4ff92SAndroid Build Coastguard Worker TEST_CASE("CreateBackendIdFromCompute") 16*89c4ff92SAndroid Build Coastguard Worker { 17*89c4ff92SAndroid Build Coastguard Worker BackendId fromCompute{Compute::GpuAcc}; 18*89c4ff92SAndroid Build Coastguard Worker CHECK(fromCompute.Get() == GetComputeDeviceAsCString(Compute::GpuAcc)); 19*89c4ff92SAndroid Build Coastguard Worker } 20*89c4ff92SAndroid Build Coastguard Worker 21*89c4ff92SAndroid Build Coastguard Worker TEST_CASE("CreateBackendIdVectorFromCompute") 22*89c4ff92SAndroid Build Coastguard Worker { 23*89c4ff92SAndroid Build Coastguard Worker std::vector<BackendId> fromComputes = {Compute::GpuAcc, Compute::CpuRef}; 24*89c4ff92SAndroid Build Coastguard Worker CHECK(fromComputes[0].Get() == GetComputeDeviceAsCString(Compute::GpuAcc)); 25*89c4ff92SAndroid Build Coastguard Worker CHECK(fromComputes[1].Get() == GetComputeDeviceAsCString(Compute::CpuRef)); 26*89c4ff92SAndroid Build Coastguard Worker } 27*89c4ff92SAndroid Build Coastguard Worker 28*89c4ff92SAndroid Build Coastguard Worker } 29