Method ShrinkAllMem

oneflow/core/vm/virtual_machine.cpp:191–206

}

Maybe<void> VirtualMachine::ShrinkAllMem() {
  auto try_shrink_men = [](vm::VirtualMachineEngine* engine) -> bool {
    if (engine->mut_active_stream_list()->size()) { return false; }
    INTRUSIVE_FOR_EACH_PTR(thread_ctx, engine->mut_thread_ctx_list()) {
      INTRUSIVE_FOR_EACH_PTR(stream, thread_ctx->mut_stream_list()) {
        vm::Allocator* allocator = stream->mut_stream_policy()->mut_allocator();
        if (allocator) {
          auto* cache = dynamic_cast<vm::CachingAllocator*>(allocator);
          if (cache != nullptr) { cache->Shrink(); }
        }
      }
    }
    return true;
  };
  return BlockingRunProbeFunc(try_shrink_men);
}

VirtualMachine::~VirtualMachine() {
  if (!threads_closed_) { CHECK_JUST(CloseVMThreads()); }