aboutsummaryrefslogtreecommitdiff
path: root/delegate
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2023-01-10 10:32:51 +0000
committerTeresaARM <teresa.charlinreyes@arm.com>2023-05-08 13:16:25 +0000
commit97a3aefff63ae081ae62aa5bac17d6e9c401937e (patch)
tree4cda3515b8718215be14ae95283a51a49b372e68 /delegate
parent1fe6c8170ae2fe90b53fb71b7570aec9dfe75c45 (diff)
downloadarmnn-97a3aefff63ae081ae62aa5bac17d6e9c401937e.tar.gz
IVGCVSW-7308 Add GpuAcc Batch MatMul workload
* Call dedicated MatMul kernel in ACL * Add int8 tests * Add int8 to documentation * Force tensors to be dynamic (nonConst) as per request of ACL Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com> Change-Id: I7b7ac20deec8637dc46ca990d339d92c4587cbe4
Diffstat (limited to 'delegate')
-rw-r--r--delegate/test/BatchMatMulTest.cpp16
1 files changed, 13 insertions, 3 deletions
diff --git a/delegate/test/BatchMatMulTest.cpp b/delegate/test/BatchMatMulTest.cpp
index 5cd1a70141..fcf1ec2b7c 100644
--- a/delegate/test/BatchMatMulTest.cpp
+++ b/delegate/test/BatchMatMulTest.cpp
@@ -303,7 +303,7 @@ namespace armnnDelegate
{
// Set input data
std::vector<int32_t> LHSInputShape { 2,2,2 };
- std::vector<int32_t> RHSInputShape { 1,2,2 };
+ std::vector<int32_t> RHSInputShape { 2,2 };
std::vector<int32_t> outputShape { 2,2,2 };
std::vector<int8_t> LHSInputValues = { 1, 2,
@@ -680,6 +680,7 @@ namespace armnnDelegate
BatchMatMul2DInt8SimpleAdjointTest(backends);
}
}
+
TEST_SUITE("BATCH_MATMUL_GpuAccTests")
{
TEST_CASE("BATCH_MATMUL_Fp32_GpuAccTests")
@@ -689,11 +690,20 @@ namespace armnnDelegate
BatchMatMul3DFp32SimpleTest (backends);
BatchMatMul4DFp32SimpleTest (backends);
BatchMatMul3DFp32BatchTest (backends);
- BatchMatMul3DFp32BroadcastTest (backends);
- BatchMatMul3D2DFp32BroadcastTest (backends);
BatchMatMul2DFp32TinyTest (backends);
BatchMatMulNonSquareFp32Test (backends);
BatchMatMul2DFp32SimpleAdjointTest(backends);
}
+
+ TEST_CASE("BATCH_MATMUL_Int8_GpuAccTests")
+ {
+ std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
+ BatchMatMul2DInt8SimpleTest (backends);
+ BatchMatMul3DInt8SimpleTest (backends);
+ BatchMatMul3DInt8BatchTest (backends);
+ BatchMatMul2DInt8TinyTest (backends);
+ BatchMatMulNonSquareInt8Test (backends);
+ BatchMatMul2DInt8SimpleAdjointTest(backends);
+ }
}
}