diff options
| author | Miao Wang <miaowang@google.com> | 2019-01-16 16:00:00 -0800 |
|---|---|---|
| committer | Miao Wang <miaowang@google.com> | 2019-01-18 08:53:04 -0800 |
| commit | ee6a835b1a4875dcd5fb094bad783b2dedd708ec (patch) | |
| tree | 0a7b5ed0de76019f7ac246b0ddecb26e3eeff34f /nn/runtime/Manager.cpp | |
| parent | d2f4303e27c8d22fafe27916e4ed865586f4d3e2 (diff) | |
| download | ml-ee6a835b1a4875dcd5fb094bad783b2dedd708ec.tar.gz | |
Add NDK API to allow developers to query the device type
Bug: 111425781
Bug: 112661050
Test: mm
Test: NeuralNetworksTest_static
Change-Id: Iff756fa8d245c2c18d8ea8682ca648640ab45a3c
Diffstat (limited to 'nn/runtime/Manager.cpp')
| -rw-r--r-- | nn/runtime/Manager.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/nn/runtime/Manager.cpp b/nn/runtime/Manager.cpp index 45df862ca..3d73465fc 100644 --- a/nn/runtime/Manager.cpp +++ b/nn/runtime/Manager.cpp @@ -50,6 +50,7 @@ class DriverDevice : public Device { const char* getVersionString() const override { return mVersionString.c_str(); } VersionedIDevice* getInterface() override { return &mInterface; } int64_t getFeatureLevel() override { return mInterface.getFeatureLevel(); } + int32_t getType() const override { return mInterface.getType(); } void getSupportedOperations(const Model& hidlModel, hidl_vec<bool>* supported) override; PerformanceInfo getFloat32Performance() const override { return mFloat32Performance; } PerformanceInfo getQuantized8Performance() const override { return mQuantized8Performance; } @@ -230,6 +231,7 @@ class CpuDevice : public Device { const char* getVersionString() const override { return kVersionString.c_str(); } VersionedIDevice* getInterface() override { return nullptr; } int64_t getFeatureLevel() override { return kFeatureLevel; } + int32_t getType() const override { return ANEURALNETWORKS_DEVICE_CPU; } void getSupportedOperations(const Model& hidlModel, hidl_vec<bool>* supported) override; PerformanceInfo getFloat32Performance() const override { return kPerformance; } PerformanceInfo getQuantized8Performance() const override { return kPerformance; } |