aboutsummaryrefslogtreecommitdiff
path: root/tests/datasets/LargeMatMulDataset.h
diff options
context:
space:
mode:
authorRamy Elgammal <ramy.elgammal@arm.com>2023-03-20 10:19:10 +0000
committerRamy Elgammal <ramy.elgammal@arm.com>2023-03-24 17:12:57 +0000
commitb531b7549abdd5c10b14b00107ea647591baa430 (patch)
treecf4dad7560fc15df11974de5defc1be1ea106dda /tests/datasets/LargeMatMulDataset.h
parentf9a547c448297cc310cad32e2cce2ce1949b4481 (diff)
downloadComputeLibrary-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/LargeMatMulDataset.h')
-rw-r--r--tests/datasets/LargeMatMulDataset.h12
1 files changed, 11 insertions, 1 deletions
diff --git a/tests/datasets/LargeMatMulDataset.h b/tests/datasets/LargeMatMulDataset.h
index b5181bc30b..8f6c000d37 100644
--- a/tests/datasets/LargeMatMulDataset.h
+++ b/tests/datasets/LargeMatMulDataset.h
@@ -65,7 +65,17 @@ public:
add_config(TensorShape(45U, 38U, 3U, 2U, 3U), TensorShape(20U, 45U, 3U, 2U, 3U), TensorShape(20U, 38U, 3U, 2U, 3U));
}
};
-
+class LargeMatMulDatasetRhsExportToCLImageRhsT final : public MatMulDataset
+{
+public:
+ // For shape choices, please refer to the explanations given in SmallMatMulDatasetRhsExportToCLImageRhsT
+ LargeMatMulDatasetRhsExportToCLImageRhsT()
+ {
+ add_config(TensorShape(28U, 13U, 3U, 2U), TensorShape(32U, 28U, 3U, 2U), TensorShape(32U, 13U, 3U, 2U));
+ add_config(TensorShape(40U, 12U, 1U, 5U, 2U), TensorShape(20U, 40U, 1U, 5U, 2U), TensorShape(20U, 12U, 1U, 5U, 2U));
+ add_config(TensorShape(44U, 38U, 3U, 2U, 3U), TensorShape(20U, 44U, 3U, 2U, 3U), TensorShape(20U, 38U, 3U, 2U, 3U));
+ }
+};
} // namespace datasets
} // namespace test
} // namespace arm_compute