From 5f3dc253c26e1c839be70672832323158aa4cee0 Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Wed, 1 Sep 2021 15:36:23 +0100 Subject: MLCE-401 Added helpers to query backend type Signed-off-by: Pablo Tello Change-Id: I6c3aa179eeaefa298e5c3a08f4d35708661615f0 --- include/armnn/BackendId.hpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'include/armnn/BackendId.hpp') diff --git a/include/armnn/BackendId.hpp b/include/armnn/BackendId.hpp index c94cd66661..5849dea8fb 100644 --- a/include/armnn/BackendId.hpp +++ b/include/armnn/BackendId.hpp @@ -132,6 +132,8 @@ public: } bool IsCpuRef() const { return m_Id == GetComputeDeviceAsCString(Compute::CpuRef); } + bool IsCpuAcc() const { return m_Id == GetComputeDeviceAsCString(Compute::CpuAcc); } + bool IsGpuAcc() const { return m_Id == GetComputeDeviceAsCString(Compute::GpuAcc); } const std::string& Get() const { return m_Id; } -- cgit v1.2.1