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>(m_buffer.data());