Browse Source

fix(atlas): add NO_MEM_RECLAME flag for the output of atlas runtime opr

GitOrigin-RevId: 34df15f1a9
release-1.1
Megvii Engine Team 4 years ago
parent
commit
545567d3ba
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      src/opr/impl/atlas_runtime_op.cpp

+ 3
- 1
src/opr/impl/atlas_runtime_op.cpp View File

@@ -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());


Loading…
Cancel
Save