From 108a95e046dde880075b6c278b44033d13f55be3 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Wed, 27 Mar 2019 13:55:59 +0000 Subject: 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 Reviewed-on: https://review.mlplatform.org/c/917 Tested-by: Arm Jenkins Reviewed-by: Michalis Spyrou Comments-Addressed: Arm Jenkins --- utils/Utils.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'utils') 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 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 * -- cgit v1.2.1