diff options
Diffstat (limited to 'test/1.1')
-rw-r--r-- | test/1.1/Mean.cpp | 7 | ||||
-rw-r--r-- | test/1.1/Transpose.cpp | 5 |
2 files changed, 9 insertions, 3 deletions
diff --git a/test/1.1/Mean.cpp b/test/1.1/Mean.cpp index 10ca3aec..b3074909 100644 --- a/test/1.1/Mean.cpp +++ b/test/1.1/Mean.cpp @@ -89,8 +89,11 @@ void MeanTestImpl(const TestTensor& input, android::sp<IMemory> outMemory = AddPoolAndGetData<float>(expectedOutput.GetNumElements(), request); const float* outputData = static_cast<const float*>(static_cast<void*>(outMemory->getPointer())); - V1_0::ErrorStatus execStatus = Execute(preparedModel, request); - BOOST_TEST(execStatus == V1_0::ErrorStatus::NONE); + if (preparedModel.get() != nullptr) + { + V1_0::ErrorStatus execStatus = Execute(preparedModel, request); + BOOST_TEST(execStatus == V1_0::ErrorStatus::NONE); + } const float* expectedOutputData = expectedOutput.GetData(); for (unsigned int i = 0; i < expectedOutput.GetNumElements(); i++) diff --git a/test/1.1/Transpose.cpp b/test/1.1/Transpose.cpp index 5679ca22..7a36130a 100644 --- a/test/1.1/Transpose.cpp +++ b/test/1.1/Transpose.cpp @@ -89,7 +89,10 @@ void TransposeTestImpl(const TestTensor & inputs, int32_t perm[], android::sp<IMemory> outMemory = AddPoolAndGetData<float>(expectedOutputTensor.GetNumElements(), request); float* outdata = static_cast<float*>(static_cast<void*>(outMemory->getPointer())); - auto execStatus = Execute(preparedModel, request); + if (preparedModel.get() != nullptr) + { + auto execStatus = Execute(preparedModel, request); + } const float * expectedOutput = expectedOutputTensor.GetData(); for (unsigned int i = 0; i < expectedOutputTensor.GetNumElements(); ++i) |