diff options
Diffstat (limited to 'src/backends/cl/test/ClEndToEndTests.cpp')
-rw-r--r-- | src/backends/cl/test/ClEndToEndTests.cpp | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/src/backends/cl/test/ClEndToEndTests.cpp b/src/backends/cl/test/ClEndToEndTests.cpp index 4ff2b79d08..a6ddd97ecf 100644 --- a/src/backends/cl/test/ClEndToEndTests.cpp +++ b/src/backends/cl/test/ClEndToEndTests.cpp @@ -8,6 +8,7 @@ #include <backendsCommon/test/ActivationEndToEndTestImpl.hpp> #include <backendsCommon/test/AdditionEndToEndTestImpl.hpp> #include <backendsCommon/test/ArgMinMaxEndToEndTestImpl.hpp> +#include <backendsCommon/test/BatchMatMulEndToEndTestImpl.hpp> #include <backendsCommon/test/ComparisonEndToEndTestImpl.hpp> #include <backendsCommon/test/ConcatEndToEndTestImpl.hpp> #include <backendsCommon/test/DepthToSpaceEndToEndTestImpl.hpp> @@ -56,6 +57,17 @@ TEST_CASE("ClAdditionEndToEndUint8Test") AdditionEndToEnd<armnn::DataType::QAsymmU8>(clDefaultBackends); } +// Batch Mat Mul +TEST_CASE("ClBatchMatMulEndToEndFloat32Test") +{ + BatchMatMulEndToEnd<armnn::DataType::Float32>(clDefaultBackends); +} + +TEST_CASE("ClBatchMatMulEndToEndInt8Test") +{ + BatchMatMulEndToEnd<armnn::DataType::QAsymmS8>(clDefaultBackends); +} + // Constant TEST_CASE("ConstantUsage_Cl_Float32") { |