[PTEN] Add pten::Place data structure.

paddle/pten/api/lib/utils/CMakeLists.txt

@@ -1,2 +1,2 @@
-cc_library(pten_api_utils SRCS allocator.cc storage.cc tensor_utils.cc place_utils.cc DEPS
-tensor_base convert_utils dense_tensor lod_tensor selected_rows place var_type_traits pten_common)
+cc_library(pten_api_utils SRCS allocator.cc storage.cc tensor_utils.cc DEPS
+tensor_base convert_utils dense_tensor lod_tensor selected_rows place var_type_traits)

paddle/pten/common/CMakeLists.txt

@@ -1 +1 @@
-cc_library(pten_common SRCS device.cc place.cc DEPS enforce)
+cc_library(pten_place SRCS place.cc)

paddle/pten/common/place.cc

@@ -1,4 +1,4 @@
-/* Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
+/* Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
 
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
@@ -13,14 +13,57 @@ See the License for the specific language governing permissions and
 limitations under the License. */
 
 #include "paddle/pten/common/place.h"
-#include "paddle/fluid/platform/enforce.h"
 
-namespace paddle {
-namespace experimental {
+#include <sstream>
+#include <string>
+
+#include "paddle/pten/api/ext/exception.h"
+
+namespace pten {
+
+const char *AllocationTypeStr(AllocationType type) {
+  switch (type) {
+    case AllocationType::UNDEF:
+      return "undef";
+    case AllocationType::CPU:
+      return "cpu";
+    case AllocationType::GPU:
+      return "gpu";
+    case AllocationType::GPUPINNED:
+      return "gpu pinned";
+    case AllocationType::XPU:
+      return "xpu";
+    case AllocationType::NPU:
+      return "npu";
+    case AllocationType::NPUPINNED:
+      return "npu pinned";
+    case AllocationType::IPU:
+      return "ipu";
+    case AllocationType::MLU:
+      return "mlu";
+    default:
+      PD_THROW("Invalid pten device type.");
+      return {};
+  }
+}
 
 std::string Place::DebugString() const {
-  return device_.DebugString() + ", is_pinned: " + std::to_string(is_pinned_);
+  std::ostringstream os;
+  os << "Place(";
+  os << AllocationTypeStr(alloc_type_);
+  if (alloc_type_ == AllocationType::GPUPINNED ||
+      alloc_type_ == AllocationType::NPUPINNED ||
+      alloc_type_ == AllocationType::CPU) {
+    os << ")";
+  } else {
+    os << ":" << std::to_string(device) << ")";
+  }
+  return os.str();
+}
+
+std::ostream &operator<<(std::ostream &os, const Place &p) {
+  os << p.DebugString();
+  return os;
 }
 
-}  // namespace experimental
-}  // namespace paddle
+}  // namespace pten