diff options
Diffstat (limited to 'src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp')
-rw-r--r-- | src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp b/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp index b6f7f32ea6..cb1ddd8182 100644 --- a/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp +++ b/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp @@ -181,4 +181,25 @@ TEST_CASE("IsLayerSupportedGpuFsaResize") CHECK(supported); } +TEST_CASE("UNSUPPORTED_IsLayerSupportedGpuFsaSoftmax") +{ + TensorInfo inputInfo({ 2, 2 }, DataType::Float32); + TensorInfo outputInfo({ 2, 2 }, DataType::Float32); + + SoftmaxDescriptor desc; + desc.m_Axis = 1; + desc.m_Beta = 1.0f; + + GpuFsaLayerSupport supportChecker; + std::string reasonIfNotSupported; + auto supported = supportChecker.IsLayerSupported(LayerType::Softmax, + {inputInfo, outputInfo}, + desc, + EmptyOptional(), + EmptyOptional(), + reasonIfNotSupported); + + CHECK(!supported); +} + }
\ No newline at end of file |