From 27503620a536b95537eedfc9e94b52bd55fbf032 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 25 Dec 2020 11:57:24 +0800
Subject: [PATCH] fix(fastrun): release free blocks before profiling

GitOrigin-RevId: b0d9887e9a947f1ae1fc5ea809fbee4336879df4
---
 src/opr/impl/search_policy/profiler.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/opr/impl/search_policy/profiler.cpp b/src/opr/impl/search_policy/profiler.cpp
index 3365bf856..1b3c7c838 100644
--- a/src/opr/impl/search_policy/profiler.cpp
+++ b/src/opr/impl/search_policy/profiler.cpp
@@ -234,6 +234,9 @@ typename TimedProfiler<Opr>::TResult TimedProfiler<Opr>::prof_impl(
         using namespace std::literals;
         std::this_thread::sleep_for(1000us);
     }
+    // release all free blocks owned by child process,
+    // in order to avoid main process running out of memory
+    cn.try_coalesce_all_free_memory();
 
     mgb_assert(ev_start->finished());
     return TResult::from_pod(Result{ev_start->elapsed_time_until(*ev_end)});
-- 
GitLab