diff options
author | Ramy Elgammal <ramy.elgammal@arm.com> | 2023-03-20 10:19:10 +0000 |
---|---|---|
committer | Ramy Elgammal <ramy.elgammal@arm.com> | 2023-03-24 17:12:57 +0000 |
commit | b531b7549abdd5c10b14b00107ea647591baa430 (patch) | |
tree | cf4dad7560fc15df11974de5defc1be1ea106dda /tests/datasets/SmallMatMulDataset.h | |
parent | f9a547c448297cc310cad32e2cce2ce1949b4481 (diff) | |
download | ComputeLibrary-b531b7549abdd5c10b14b00107ea647591baa430.tar.gz |
Add Texture Pipe Support for Matmul Lhs T/NT Rhs T kernels
Resolves: COMPMID-5952, COMPMID-5956
Change-Id: Idbd14538e7660792254072fa9631a6f03966f89b
Signed-off-by: Ramy Elgammal <ramy.elgammal@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9371
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/datasets/SmallMatMulDataset.h')
-rw-r--r-- | tests/datasets/SmallMatMulDataset.h | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tests/datasets/SmallMatMulDataset.h b/tests/datasets/SmallMatMulDataset.h index 93e5f7dc2c..52ef01da7b 100644 --- a/tests/datasets/SmallMatMulDataset.h +++ b/tests/datasets/SmallMatMulDataset.h @@ -57,6 +57,18 @@ public: } }; +class SmallMatMulDatasetRhsExportToCLImageRhsT final : public MatMulDataset +{ +public: + // Some considerations: + // 1) K dimension should be a multiple of 4 + // See (2), (3), and (4) in SmallMatMulDatasetRhsExportToCLImageRhsNT + SmallMatMulDatasetRhsExportToCLImageRhsT() + { + add_config(TensorShape(8U /*K*/, 3U /*M*/, 2U, 1U, 2U), TensorShape(20U /*N*/, 8U /*K*/, 2U, 1U, 2U), TensorShape(20U /*N*/, 3U /*M*/, 2U, 1U, 2U)); + } +}; + class SmallMatMulDatasetRhsExportToCLImageRhsNT final : public MatMulDataset { public: |