From 5ca7409bc02ea1ac8ea34f0779f18221880fa6ac Mon Sep 17 00:00:00 2001 From: Gian Marco Date: Thu, 8 Feb 2018 16:21:54 +0000 Subject: COMPMID-765 - Used GEMM-based convolution in VGG16 In order to use GEMM-based convolution in VGG16, it has been created a function which allocates 1.8 GB. If the function fails, will be used DIRECT convolution instead Change-Id: Ibec8928ee6fe6684d6dc24b7df380beeb671bf27 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/119490 Tested-by: Jenkins Reviewed-by: Michalis Spyrou Reviewed-by: Gian Marco Iodice Reviewed-by: Anthony Barbier --- utils/Utils.h | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'utils/Utils.h') diff --git a/utils/Utils.h b/utils/Utils.h index ff4c4c99fd..8822fe7121 100644 --- a/utils/Utils.h +++ b/utils/Utils.h @@ -895,7 +895,11 @@ 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)); } - +/** This function returns the amount of memory free reading from /proc/meminfo + * + * @return The free memory in kB + */ +uint64_t get_mem_free_from_meminfo(); } // namespace utils } // namespace arm_compute #endif /* __UTILS_UTILS_H__*/ -- cgit v1.2.1