arm_compute v18.05

[platform/upstream/armcl.git] / utils / Utils.cpp

diff --git a/utils/Utils.cpp b/utils/Utils.cpp
index 32d5e3a..a5c6a95 100644 (file)
--- 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
  *
@@ -67,16 +67,16 @@ void discard_comments_and_spaces(std::ifstream &fs)
 } // namespace
 
 #ifndef BENCHMARK_EXAMPLES
-int run_example(int argc, char **argv, Example &example)
+int run_example(int argc, char **argv, std::unique_ptr<Example> example)
 {
     std::cout << "\n"
               << argv[0] << "\n\n";
 
     try
     {
-        example.do_setup(argc, argv);
-        example.do_run();
-        example.do_teardown();
+        example->do_setup(argc, argv);
+        example->do_run();
+        example->do_teardown();
 
         std::cout << "\nTest passed\n";
         return 0;
@@ -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