diff options
Diffstat (limited to 'src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp')
-rw-r--r-- | src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp | 35 |
1 files changed, 13 insertions, 22 deletions
diff --git a/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp b/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp index dda4d1ffaf..34af1909e6 100644 --- a/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp +++ b/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp @@ -79,34 +79,25 @@ TEST_CASE("IsLayerSupportedGpuFsaConv2dUnsupported") REQUIRE(reasonIfNotSupported.find("NCHW not supported by this kernel") != std::string::npos); } -TEST_CASE("IsLayerSupportedGpuFsaElementWiseBinaryAdd") +TEST_CASE("IsLayerSupportedGpuFsaElementWiseBinary") { 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); -} - -TEST_CASE("IsLayerSupportedGpuFsaElementWiseBinarySub") -{ - TensorInfo input0Info({ 2, 2 }, DataType::Float32); - TensorInfo input1Info({ 2, 2 }, DataType::Float32); - TensorInfo outputInfo({ 2, 2 }, DataType::Float32); - - ElementwiseBinaryDescriptor desc; - desc.m_Operation = BinaryOperation::Sub; + SUBCASE("Add") + { + desc.m_Operation = BinaryOperation::Add; + } + SUBCASE("Mul") + { + desc.m_Operation = BinaryOperation::Mul; + } + SUBCASE("Sub") + { + desc.m_Operation = BinaryOperation::Sub; + } GpuFsaLayerSupport supportChecker; std::string reasonIfNotSupported; |