Method InitEagerBlobObject

oneflow/core/framework/tensor_impl.cpp:120–142

}

Maybe<void> EagerLocalTensorImpl::InitEagerBlobObject(
    const Symbol<one::LocalTensorMeta>& local_tensor_meta,
    const std::shared_ptr<const one::MutLocalTensorMeta>& mut_local_tensor_meta,
    const intrusive::shared_ptr<LocalDepObject>& dep_object) {
  CHECK_OR_RETURN(static_cast<bool>(local_tensor_meta->device()));  // NOLINT
  const auto& mem_case = local_tensor_meta->device()->mem_case();

  if (tensor_storage_) {
    auto tensor_storage = tensor_storage_->storage();
    eager_blob_object_ = std::make_shared<vm::EagerBlobObject>(
        mem_case, local_tensor_meta, mut_local_tensor_meta, local_tensor_meta->dtype(),
        local_tensor_meta->memory_format(), tensor_storage, dep_object);
  } else {
    auto device = local_tensor_meta->device();
    auto storage = device->rematable() ? std::make_shared<vm::RematableTensorStorage>(device)
                                       : std::make_shared<vm::TensorStorage>(true, device);
    const auto& eager_blob_object = std::make_shared<vm::EagerBlobObject>(
        mem_case, local_tensor_meta, mut_local_tensor_meta, local_tensor_meta->dtype(),
        local_tensor_meta->memory_format(), storage, dep_object);
    JUST(set_eager_blob_object(eager_blob_object));
  }
  return Maybe<void>::Ok();
}

Maybe<bool> EagerLocalTensorImpl::is_pinned() const {
  if (this->device() == JUST(Device::New("meta"))) { return false; }