From 545567d3bac7f63a173c1dd8c43221e417649275 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Sat, 17 Oct 2020 00:14:56 +0800
Subject: [PATCH] fix(atlas): add NO_MEM_RECLAME flag for the output of atlas
 runtime opr

GitOrigin-RevId: 34df15f1a9f45437268f79f5385285c9a01aa08c
---
 src/opr/impl/atlas_runtime_op.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/opr/impl/atlas_runtime_op.cpp b/src/opr/impl/atlas_runtime_op.cpp
index e2fb4ff3..7853545a 100644
--- a/src/opr/impl/atlas_runtime_op.cpp
+++ b/src/opr/impl/atlas_runtime_op.cpp
@@ -254,7 +254,9 @@ AtlasRuntimeOpr::AtlasRuntimeOpr(SharedBuffer buf,
          * set the flag to SYS_MEM_ALLOC.
          */
         for (size_t i = 0; i < nr_outputs; ++i) {
-            output(i)->add_flag(F::NO_SYS_MEM_ALLOC);
+            output(i)
+                    ->add_flag(F::NO_SYS_MEM_ALLOC)
+                    .add_flag(F::NO_MEM_RECLAIM);
         }
     }
     add_equivalence_component<mgb::ScalarHash<const void*>>(m_buffer.data());
-- 
GitLab