diff options
author | Pablo Tello <pablo.tello@arm.com> | 2021-09-01 15:30:10 +0100 |
---|---|---|
committer | Pablo Marquez Tello <pablo.tello@arm.com> | 2021-09-03 10:46:14 +0000 |
commit | 6bf3f43c6f1d88c676f277f4c3c766cb3ea37ed0 (patch) | |
tree | abda8bb6e49654c8f31cf7eae862642db0310056 /1.2/ArmnnDriver.hpp | |
parent | 69e38617cc66c008e092d533f3f3bede8043223c (diff) | |
download | android-nn-driver-6bf3f43c6f1d88c676f277f4c3c766cb3ea37ed0.tar.gz |
android-nn-driver getType returns the right device
* ArmnnDriver queries the options and returns CPU or GPU
depending on which is the first backend listed in the options
* Resolves MLCE-401
Change-Id: If4e63e144507e817449f37926711fa325861b57d
Signed-off-by: Pablo Tello <pablo.tello@arm.com>
Diffstat (limited to '1.2/ArmnnDriver.hpp')
-rw-r--r-- | 1.2/ArmnnDriver.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/1.2/ArmnnDriver.hpp b/1.2/ArmnnDriver.hpp index 5227272f..a350d3f4 100644 --- a/1.2/ArmnnDriver.hpp +++ b/1.2/ArmnnDriver.hpp @@ -129,8 +129,8 @@ public: Return<void> getType(getType_cb cb) { ALOGV("hal_1_2::ArmnnDriver::getType()"); - - cb(V1_0::ErrorStatus::NONE, V1_2::DeviceType::CPU); + const auto device_type = hal_1_2::HalPolicy::GetDeviceTypeFromOptions(this->m_Options); + cb(V1_0::ErrorStatus::NONE, device_type); return Void(); } |