add topyobject for place

PaddlePaddle · Nov 19, 2021 · aa67cca · aa67cca
1 parent fbde0b1
commit aa67cca
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/paddle/fluid/pybind/eager_properties.cc b/paddle/fluid/pybind/eager_properties.cc
@@ -100,10 +100,7 @@ PyObject* eager_tensor_properties_get_shape(EagerTensorObject* self,
 
 PyObject* eager_tensor_properties_get_place(EagerTensorObject* self,
                                             void* closure) {
-  auto place = self->eagertensor.place();
-  auto obj = ::pybind11::cast(place);
-  obj.inc_ref();
-  return obj.ptr();
+  return ToPyObject(self->eagertensor.place());
 }
 
 PyObject* eager_tensor_properties_get_place_str(EagerTensorObject* self,

diff --git a/paddle/fluid/pybind/eager_utils.cc b/paddle/fluid/pybind/eager_utils.cc
@@ -329,5 +329,11 @@ PyObject* ToPyObject(const std::vector<egr::EagerTensor>& value) {
   return result;
 }
 
+PyObject* ToPyObject(const platform::Place& value) {
+  auto obj = ::pybind11::cast(value);
+  obj.inc_ref();
+  return obj.ptr();
+}
+
 }  // namespace pybind
 }  // namespace paddle
diff --git a/paddle/fluid/pybind/eager_utils.h b/paddle/fluid/pybind/eager_utils.h
@@ -48,6 +48,7 @@ PyObject* ToPyObject(const std::vector<int64_t>& value);
 PyObject* ToPyObject(const std::vector<float>& value);
 PyObject* ToPyObject(const std::vector<double>& value);
 PyObject* ToPyObject(const std::vector<egr::EagerTensor>& value);
+PyObject* ToPyObject(const platform::Place& value);
 
 }  // namespace pybind
 }  // namespace paddle