diff options
author | Gian Marco <gianmarco.iodice@arm.com> | 2018-02-08 16:21:54 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:47:18 +0000 |
commit | 5ca7409bc02ea1ac8ea34f0779f18221880fa6ac (patch) | |
tree | 3018aa85c14bb8f89979b13bca76c60d249e452e /utils/Utils.cpp | |
parent | 284cfe2e3a44e5b20978e561c96c94d2193e93a1 (diff) | |
download | ComputeLibrary-5ca7409bc02ea1ac8ea34f0779f18221880fa6ac.tar.gz |
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 <bsgcomp@arm.com>
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'utils/Utils.cpp')
-rw-r--r-- | utils/Utils.cpp | 36 |
1 files changed, 35 insertions, 1 deletions
diff --git a/utils/Utils.cpp b/utils/Utils.cpp index 32d5e3a6c0..8a2d11814e 100644 --- a/utils/Utils.cpp +++ b/utils/Utils.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017, 2018 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -191,5 +191,39 @@ std::tuple<std::vector<unsigned long>, bool, std::string> parse_npy_header(std:: return std::make_tuple(shape, fortran_order, typestr); } + +/** 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() +{ + std::string line_attribute; + std::ifstream file_meminfo("/proc/meminfo"); + + if(file_meminfo.is_open()) + { + while(!(file_meminfo >> line_attribute).fail()) + { + //Test if is the line containing MemFree + if(line_attribute == "MemFree:") + { + uint64_t mem_available; + if(!(file_meminfo >> mem_available).fail()) + { + return mem_available; + } + else + { + return 0; + } + } + // if it's not MemFree ignore rest of the line + file_meminfo.ignore(std::numeric_limits<std::streamsize>::max(), '\n'); + } + } + // Nothing found or an error during opening the file + return 0; +} } // namespace utils } // namespace arm_compute |