Function NeedTransformPlace

paddle/phi/api/lib/data_transform.h:167–210

    const std::vector<phi::DenseTensor>& tensor);

inline bool NeedTransformPlace(const phi::Place& src_place,
                               const Backend& target,
                               const TransformFlag& transform_flag) {
  // NOTE(dev): The default value of TransformFlag is True, if it is set with
  // False
  // somewhere such as ops.yaml or backward.yaml that means we should skip data
  // transform. Because "stop_transform_" has highest priority.
  if (!transform_flag.need_trans_backend()) {
    return false;
  }

#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
  bool ret = src_place.GetType() == AllocationType::GPUPINNED ||
             (target != Backend::ALL_BACKEND &&
              phi::TransToPhiBackend(src_place) !=
                  (target != Backend::GPUDNN ? target : Backend::GPU));
#elif defined(PADDLE_WITH_XPU)
  bool ret = src_place.GetType() == AllocationType::XPUPINNED ||
             (target != Backend::ALL_BACKEND &&
              phi::TransToPhiBackend(src_place) != target);
#elif defined(PADDLE_WITH_IPU)
  bool ret = target != Backend::ALL_BACKEND &&
             phi::TransToPhiBackend(src_place) != target;
#elif defined(PADDLE_WITH_CUSTOM_DEVICE)
  bool ret = target != Backend::ALL_BACKEND;
  if (target == Backend::CUSTOM) {
    ret = ret && !is_custom_place(src_place);
  } else {
    ret =
        ret && phi::TransToPhiBackend(src_place) !=
                   (target != Backend::GPUDNN ? target
                                              : Backend::DEFAULT_CUSTOM_DEVICE);
  }
#else
  bool ret = false;
#endif

#ifdef PADDLE_WITH_DNNL
  if (target == Backend::ONEDNN) {
    ret = src_place.GetType() != AllocationType::CPU;
  }
#endif
  return ret;
}

/* ------------------ for auto parallel ----------------------- */