diff options
author | Slava Shklyaev <slavash@google.com> | 2019-04-01 14:04:32 +0000 |
---|---|---|
committer | Android (Google) Code Review <android-gerrit@google.com> | 2019-04-01 14:04:32 +0000 |
commit | 8e0cb533bacca9f173529e2720c8b9eff8e658c7 (patch) | |
tree | 6cbcbdf23c3c49dc69ba1660b5ddc7ce7540d49c /nn/runtime/test/TestExtensions.cpp | |
parent | 8c976d1961a9654d1f95a8d67ee53ee8ce4df0c5 (diff) | |
parent | d828d15d5c07e3939b2f419466e1648a24af9709 (diff) | |
download | ml-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.cpp | 7 |
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; |