diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-01-04 17:17:02 +0000 |
|---|---|---|
| committer | Nikhil Raj Arm <nikhil.raj@arm.com> | 2021-01-05 22:16:33 +0000 |
| commit | 1b4e5d35eb5c51a4b4b2838b1098847ba3c717dc (patch) | |
| tree | 860db6e921d5604cbb79e30477ca17724493a591 | |
| parent | baa159acb48aa85a577c278e619f2a273b16f29c (diff) | |
| download | armnn-1b4e5d35eb5c51a4b4b2838b1098847ba3c717dc.tar.gz | |
adding BOOST_TEST to EnqueueWorkload in FuseActivation Unit Test
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I6a00045967fa48ec0913c8708ffc146a72ed2b87
| -rw-r--r-- | src/armnn/test/optimizations/FuseActivationTests.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
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) |