diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2021-08-04 15:22:28 +0100 |
|---|---|---|
| committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2021-08-13 12:34:17 +0000 |
| commit | 37a4611dcbefc9ad19af803216d9e2cd5ec64697 (patch) | |
| tree | b479ebcd25e1326afa6e756c70ded4e87c0fa651 /tests/datasets/SmallGEMMDataset.h | |
| parent | 7c2965760501ac96836b4f131d361266c99b91f2 (diff) | |
| download | ComputeLibrary-37a4611dcbefc9ad19af803216d9e2cd5ec64697.tar.gz | |
Add test for validating batched-GEMM on CPU
- Add new dataset for batched-GEMM
- Add test for running batched-GEMM without bias. Currently bias is not
supported in batched-GEMM
- Fix reference implementation to slide correctly the RHS tensor
Resolves COMPMID-4588
Change-Id: I20fcb5d9160f44292b7cc34570add911b1d732f6
Signed-off-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6040
Reviewed-by: SiCong Li <sicong.li@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/datasets/SmallGEMMDataset.h')
| -rw-r--r-- | tests/datasets/SmallGEMMDataset.h | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/tests/datasets/SmallGEMMDataset.h b/tests/datasets/SmallGEMMDataset.h index 7d2b42a0d6..23b46a6547 100644 --- a/tests/datasets/SmallGEMMDataset.h +++ b/tests/datasets/SmallGEMMDataset.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2019 Arm Limited. + * Copyright (c) 2017-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -50,6 +50,16 @@ public: add_config(TensorShape(32U, 1U), TensorShape(17U, 32U), TensorShape(17U, 1U), TensorShape(17U, 1U), 0.4f, 0.7f); } }; + +class SmallBatchedGEMMDataset final : public GEMMDataset +{ +public: + SmallBatchedGEMMDataset() + { + add_config(TensorShape(2U, 4U, 1U, 3U), TensorShape(5U, 2U, 3U), TensorShape(5U), TensorShape(5U, 4U, 1U, 3U), 1.0f, 0.0f); + } +}; + class SmallGEMMOutput3DDataset final : public GEMMDataset { public: |