From 1b4e5d35eb5c51a4b4b2838b1098847ba3c717dc Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Mon, 4 Jan 2021 17:17:02 +0000 Subject: adding BOOST_TEST to EnqueueWorkload in FuseActivation Unit Test Signed-off-by: Teresa Charlin Change-Id: I6a00045967fa48ec0913c8708ffc146a72ed2b87 --- src/armnn/test/optimizations/FuseActivationTests.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'src/armnn/test/optimizations/FuseActivationTests.cpp') diff --git a/src/armnn/test/optimizations/FuseActivationTests.cpp b/src/armnn/test/optimizations/FuseActivationTests.cpp index d5525811d2..c8adea2132 100644 --- a/src/armnn/test/optimizations/FuseActivationTests.cpp +++ b/src/armnn/test/optimizations/FuseActivationTests.cpp @@ -374,7 +374,7 @@ void FuseActivationIntoPreviousLayerTest(ActivationDescriptor activationDescript {0, Tensor(run->GetOutputTensorInfo(networkIdentifier, 0), outputDataFused.data())}}; // Execute network - run->EnqueueWorkload(networkIdentifier, inputTensorsFused, outputTensorsFused); + BOOST_TEST(run->EnqueueWorkload(networkIdentifier, inputTensorsFused, outputTensorsFused) == Status::Success); // SECOND NETWORK: NotFused // Construct ArmNN network @@ -413,7 +413,8 @@ void FuseActivationIntoPreviousLayerTest(ActivationDescriptor activationDescript {1, Tensor(runNotFused->GetOutputTensorInfo(networkIdentifierNotFused, 1), outputData2NotFused.data())}}; // Execute network - runNotFused->EnqueueWorkload(networkIdentifierNotFused, inputTensorsNotFused, outputTensorsNotFused); + BOOST_TEST(runNotFused->EnqueueWorkload(networkIdentifierNotFused, inputTensorsNotFused, outputTensorsNotFused) + == Status::Success); // Check the output of the fused-activation matches with the output of the activation in the "NotFused" network for (unsigned int n = 0; n < outputDataFused.size(); ++n) -- cgit v1.2.1