diff options
Diffstat (limited to 'src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp')
-rw-r--r-- | src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp b/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp index 79dd9d357d..9972b24637 100644 --- a/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp +++ b/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp @@ -6,7 +6,9 @@ #include "backendsCommon/test/EndToEndTestImpl.hpp" #include "backendsCommon/test/Convolution2dEndToEndTestImpl.hpp" + #include "backendsCommon/test/DepthwiseConvolution2dEndToEndTests.hpp" +#include "backendsCommon/test/ElementwiseBinaryEndToEndTestImpl.hpp" #include <doctest/doctest.h> @@ -32,4 +34,15 @@ TEST_CASE("GpuFsaDepthwiseConvolution2dEndtoEndTestFloat32") armnn::DataLayout::NHWC); } +// ElementwiseBinary Add +TEST_CASE("GpuFsaElementwiseBinaryAddTestFloat32") +{ + ElementwiseBinarySimple3DEndToEnd<armnn::DataType::Float32>(gpuFsaDefaultBackends, BinaryOperation::Add); +} + +TEST_CASE("GpuFsaElementwiseBinaryAddTestFloat16") +{ + ElementwiseBinarySimple3DEndToEnd<armnn::DataType::Float16>(gpuFsaDefaultBackends, BinaryOperation::Add); +} + } |