summaryrefslogtreecommitdiff
path: root/nn/runtime/test/TestExtensions.cpp
diff options
context:
space:
mode:
authorSlava Shklyaev <slavash@google.com>2019-04-01 14:04:32 +0000
committerAndroid (Google) Code Review <android-gerrit@google.com>2019-04-01 14:04:32 +0000
commit8e0cb533bacca9f173529e2720c8b9eff8e658c7 (patch)
tree6cbcbdf23c3c49dc69ba1660b5ddc7ce7540d49c /nn/runtime/test/TestExtensions.cpp
parent8c976d1961a9654d1f95a8d67ee53ee8ce4df0c5 (diff)
parentd828d15d5c07e3939b2f419466e1648a24af9709 (diff)
downloadml-8e0cb533bacca9f173529e2720c8b9eff8e658c7.tar.gz
Merge changes I1cd258e3,Ibe0fc535
* changes: Document NNAPI extensions Add a test Fibonacci extension
Diffstat (limited to 'nn/runtime/test/TestExtensions.cpp')
-rw-r--r--nn/runtime/test/TestExtensions.cpp7
1 files changed, 6 insertions, 1 deletions
diff --git a/nn/runtime/test/TestExtensions.cpp b/nn/runtime/test/TestExtensions.cpp
index bd182311e..10b305e2c 100644
--- a/nn/runtime/test/TestExtensions.cpp
+++ b/nn/runtime/test/TestExtensions.cpp
@@ -69,11 +69,16 @@ class ExtensionsTest : public ::testing::Test {
}
DeviceManager::get()->forTest_registerDevice(kTestDriverName, new TestDriver());
+ // Discover extensions provided by registered devices.
+ TypeManager::get()->forTest_reset();
mDevice = getDeviceByName(kTestDriverName);
ASSERT_NE(mDevice, nullptr);
}
- virtual void TearDown() { DeviceManager::get()->forTest_reInitializeDeviceList(); }
+ virtual void TearDown() {
+ DeviceManager::get()->forTest_reInitializeDeviceList();
+ TypeManager::get()->forTest_reset();
+ }
ANeuralNetworksDevice* getDeviceByName(const std::string& name) {
ANeuralNetworksDevice* result = nullptr;