Method CreateStream

oneflow/core/vm/virtual_machine.cpp:384–389

}

Maybe<vm::Stream*> VirtualMachine::CreateStream(Symbol<Stream> stream) {
  std::unique_lock<std::recursive_mutex> lock(stream_and_thread_ctx_mutex_);
  vm::ThreadCtx* thread_ctx =
      JUST(FindOrCreateThreadCtx(stream->device(), stream->stream_type(), stream->thread_uid()));
  return JUST(CreateStream(thread_ctx, stream));
}

Maybe<vm::Stream*> VirtualMachine::GetVmStream(Symbol<Stream> stream) {
  if (stream->unique_stream_id() >= unique_stream_id2vm_stream_.size()) {