Function AllocationTypeStr

paddle/phi/common/place.cc:28–50

namespace phi {

const char *AllocationTypeStr(AllocationType type) {
  switch (type) {
    case AllocationType::UNDEFINED:
      return "undefined";
    case AllocationType::CPU:
      return "cpu";
    case AllocationType::GPU:
      return "gpu";
    case AllocationType::GPUPINNED:
      return "gpu_pinned";
    case AllocationType::XPU:
      return "xpu";
    case AllocationType::XPUPINNED:
      return "xpu_pinned";
    case AllocationType::IPU:
      return "ipu";
    case AllocationType::CUSTOM:
      return "custom_device";
    default:
      PD_THROW("Invalid phi device type.");
      return {};
  }
}

std::ostream &operator<<(std::ostream &os, AllocationType type) {
  os << AllocationTypeStr(type);