diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-03-27 13:55:59 +0000 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-04-02 10:24:21 +0000 |
commit | 108a95e046dde880075b6c278b44033d13f55be3 (patch) | |
tree | 7337f266bc3ef4379dfa0e60ccf2a420c31f41b6 /utils | |
parent | 875a751cb0159329b44f545d993aa56b277773c7 (diff) | |
download | ComputeLibrary-108a95e046dde880075b6c278b44033d13f55be3.tar.gz |
COMPMID-2067: Add batch size option in validate_cl_gemm example
-Add batch option
-Switches to the command line parsing interface
Change-Id: I7c6df752797a37440f980bce6b29e9b5c5244fcb
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/917
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'utils')
-rw-r--r-- | utils/Utils.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/utils/Utils.h b/utils/Utils.h index 788ae4eeb7..afd90a11a3 100644 --- a/utils/Utils.h +++ b/utils/Utils.h @@ -774,7 +774,7 @@ void fill_random_tensor(T &tensor, float lower_bound, float upper_bound) template <typename T> void init_sgemm_output(T &dst, T &src0, T &src1, arm_compute::DataType dt) { - dst.allocator()->init(TensorInfo(TensorShape(src1.info()->dimension(0), src0.info()->dimension(1)), 1, dt)); + dst.allocator()->init(TensorInfo(TensorShape(src1.info()->dimension(0), src0.info()->dimension(1), src0.info()->dimension(2)), 1, dt)); } /** This function returns the amount of memory free reading from /proc/meminfo * |