diff options
Diffstat (limited to 'delegate/test')
-rw-r--r-- | delegate/test/BatchMatMulTest.cpp | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/delegate/test/BatchMatMulTest.cpp b/delegate/test/BatchMatMulTest.cpp index c6d7bc5f32..5cd1a70141 100644 --- a/delegate/test/BatchMatMulTest.cpp +++ b/delegate/test/BatchMatMulTest.cpp @@ -663,12 +663,22 @@ namespace armnnDelegate BatchMatMul3DFp32SimpleTest (backends); BatchMatMul4DFp32SimpleTest (backends); BatchMatMul3DFp32BatchTest (backends); - BatchMatMul3DFp32BroadcastTest (backends); - BatchMatMul3D2DFp32BroadcastTest (backends); BatchMatMul2DFp32TinyTest (backends); BatchMatMulNonSquareFp32Test (backends); BatchMatMul2DFp32SimpleAdjointTest(backends); } + + TEST_CASE("BATCH_MATMUL_Int8_CpuAccTests") + { + std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc}; + BatchMatMul2DInt8SimpleTest (backends); + BatchMatMul3DInt8SimpleTest (backends); + BatchMatMul4DInt8SimpleTest (backends); + BatchMatMul3DInt8BatchTest (backends); + BatchMatMul2DInt8TinyTest (backends); + BatchMatMulNonSquareInt8Test (backends); + BatchMatMul2DInt8SimpleAdjointTest(backends); + } } TEST_SUITE("BATCH_MATMUL_GpuAccTests") { |