diff options
author | Miao Wang <miaowang@google.com> | 2020-04-28 15:52:01 -0700 |
---|---|---|
committer | Miao Wang <miaowang@google.com> | 2020-04-28 15:55:08 -0700 |
commit | d62e2398c3497375fdaf2d554aa479d18d30f7b6 (patch) | |
tree | 7cdd327fe26560908c5d7e0aa13649a5028120fe /nn/runtime/test | |
parent | 5d35af6b9f37fcb97d493bcfbfb0a99f5ed44357 (diff) | |
download | ml-d62e2398c3497375fdaf2d554aa479d18d30f7b6.tar.gz |
Skip TestRemoveDefaultArguments tests for useCpuOnly = 1
- The tests require mTestDevice being valid to perform the tests, while
useCpuOnly will only allow CPU fallback path.
Fixes: 155229031
Test: mm
Test: NeuralNetworksTest_static
Change-Id: Ic866d14de39d0c8f52648ad319bac14bc8fdd666
Diffstat (limited to 'nn/runtime/test')
-rw-r--r-- | nn/runtime/test/TestRemoveDefaultArguments.cpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/nn/runtime/test/TestRemoveDefaultArguments.cpp b/nn/runtime/test/TestRemoveDefaultArguments.cpp index 6142aa610..ac4ad7d14 100644 --- a/nn/runtime/test/TestRemoveDefaultArguments.cpp +++ b/nn/runtime/test/TestRemoveDefaultArguments.cpp @@ -137,6 +137,10 @@ class TestDriver : public SampleDriverPartial { class TestRemoveDefaultArguments : public ::testing::Test { virtual void SetUp() { + // skip the tests if useCpuOnly = 1 + if (DeviceManager::get()->getUseCpuOnly()) { + GTEST_SKIP(); + } mTestDriver = new TestDriver(); DeviceManager::get()->forTest_registerDevice(kTestDriverName, mTestDriver); mTestDevice = getDeviceByName(kTestDriverName); |