diff options
Diffstat (limited to 'src/backends/backendsCommon/test/PreluEndToEndTestImpl.hpp')
| -rw-r--r-- | src/backends/backendsCommon/test/PreluEndToEndTestImpl.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/test/PreluEndToEndTestImpl.hpp b/src/backends/backendsCommon/test/PreluEndToEndTestImpl.hpp index 0dc1e78ced..e11553dd38 100644 --- a/src/backends/backendsCommon/test/PreluEndToEndTestImpl.hpp +++ b/src/backends/backendsCommon/test/PreluEndToEndTestImpl.hpp @@ -10,6 +10,8 @@ #include <backendsCommon/test/CommonTestUtils.hpp> +#include <doctest/doctest.h> + namespace { template<typename armnn::DataType DataType> @@ -56,7 +58,7 @@ void PreluEndToEnd(const std::vector<BackendId>& backends, INetworkPtr net = CreatePreluNetwork<ArmnnType>(inputInfo, alphaInfo, outputInfo); - BOOST_TEST_CHECKPOINT("Create a network"); + CHECK(net); std::map<int, std::vector<T>> inputTensorData = { { 0, inputData }, { 1, alphaData} }; std::map<int, std::vector<T>> expectedOutputTensorData = { { 0, expectedOutputData } }; |