diff options
Diffstat (limited to 'src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp')
-rw-r--r-- | src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp b/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp index f162df0b55..49ddadea59 100644 --- a/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp +++ b/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2022-2023 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2022-2024 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -61,4 +61,24 @@ TEST_CASE("IsLayerSupportedGpuFsaConv2dUnsupported") REQUIRE(reasonIfNotSupported.find("NCHW not supported by this kernel") != std::string::npos); } +TEST_CASE("IsLayerSupportedGpuFsaElementWiseBinaryAdd") +{ + TensorInfo input0Info({ 2, 2 }, DataType::Float32); + TensorInfo input1Info({ 2, 2 }, DataType::Float32); + TensorInfo outputInfo({ 2, 2 }, DataType::Float32); + + ElementwiseBinaryDescriptor desc; + desc.m_Operation = BinaryOperation::Add; + + GpuFsaLayerSupport supportChecker; + std::string reasonIfNotSupported; + auto supported = supportChecker.IsLayerSupported(LayerType::ElementwiseBinary, + {input0Info, input1Info, outputInfo}, + desc, + EmptyOptional(), + EmptyOptional(), + reasonIfNotSupported); + CHECK(supported); +} + }
\ No newline at end of file |