diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-01-10 10:32:51 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2023-05-08 13:16:25 +0000 |
commit | 97a3aefff63ae081ae62aa5bac17d6e9c401937e (patch) | |
tree | 4cda3515b8718215be14ae95283a51a49b372e68 /delegate/test | |
parent | 1fe6c8170ae2fe90b53fb71b7570aec9dfe75c45 (diff) | |
download | armnn-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/test')
-rw-r--r-- | delegate/test/BatchMatMulTest.cpp | 16 |
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); + } } } |