Function Trans2Contiguous

paddle/phi/api/lib/data_transform.cc:272–317

}

phi::DenseTensor Trans2Contiguous(const phi::DenseTensor& tensor) {
  auto& pool = phi::DeviceContextPool::Instance();

  VLOG(3) << "Trans2Contiguous...";

  if (tensor.place().GetType() == phi::AllocationType::CPU) {
    auto* dev_ctx = static_cast<phi::CPUContext*>(pool.Get(tensor.place()));
    return TensorContiguous<phi::CPUContext>(*dev_ctx, tensor);
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
  } else if (tensor.place().GetType() == phi::AllocationType::GPU) {
    auto* dev_ctx = static_cast<phi::GPUContext*>(pool.Get(tensor.place()));
    return TensorContiguous<phi::GPUContext>(*dev_ctx, tensor);
#endif
#ifdef PADDLE_WITH_XPU
  } else if (tensor.place().GetType() == phi::AllocationType::XPU) {
    auto* dev_ctx = static_cast<phi::XPUContext*>(pool.Get(tensor.place()));
    return TensorContiguous<phi::XPUContext>(*dev_ctx, tensor);
#endif
#ifdef PADDLE_WITH_CUSTOM_DEVICE
  } else if (tensor.place().GetType() == phi::AllocationType::CUSTOM) {
    auto* dev_ctx = static_cast<phi::CustomContext*>(pool.Get(tensor.place()));
    phi::DenseTensor dense_out;
    phi::MetaTensor meta_input(tensor);
    phi::MetaTensor meta_out(&dense_out);
    UnchangedInferMeta(meta_input, &meta_out);
    const phi::KernelKey& kernel_key = {phi::TransToPhiBackend(tensor.place()),
                                        phi::DataLayout::ALL_LAYOUT,
                                        tensor.dtype()};
    using kernel_signature = void (*)(
        const phi::DeviceContext&, const phi::DenseTensor&, phi::DenseTensor*);
    PD_VISIT_KERNEL("contiguous",
                    kernel_key,
                    kernel_signature,
                    false,
                    *dev_ctx,
                    tensor,
                    &dense_out);
    return dense_out;
#endif
  } else {
    PADDLE_THROW(common::errors::Unimplemented(
        "Place type is not supported when casting data type."));
  }

  return tensor;
}

void CheckAndTrans2Contiguous(phi::DenseTensor* tensor) {
  if (!tensor->meta().is_contiguous()) {