diff --git a/paddle/fluid/pybind/pybind.cc b/paddle/fluid/pybind/pybind.cc index cef95de2ef675e417b5a2c49d01e3c85e23f9718..a37c04b5ffbaaa073fcfb05fb03caed77652d7e6 100644 --- a/paddle/fluid/pybind/pybind.cc +++ b/paddle/fluid/pybind/pybind.cc @@ -769,7 +769,11 @@ All parameter, weight, gradient are variables in Paddle. #if (defined(PADDLE_WITH_CUDA) && !defined(_WIN32)) py::class_(m, "Communicator").def(py::init<>()); #endif - py::class_(m, "CUDAPlace") + py::class_(m, "CUDAPlace", R"DOC( + CUDAPlace is a descriptor of a device. It represents a GPU, and each CUDAPlace + has a dev_id to indicate the number of cards represented by the current CUDAPlace. + The memory of CUDAPlace with different dev_id is not accessible. + )DOC") .def("__init__", [](platform::CUDAPlace &self, int dev_id) { #ifdef PADDLE_WITH_CUDA @@ -790,7 +794,10 @@ All parameter, weight, gradient are variables in Paddle. &IsSamePlace) .def("__str__", string::to_string); - py::class_(m, "CPUPlace") + py::class_(m, "CPUPlace", R"DOC( + CPUPlace is a descriptor of a device. It represents a CPU, and the memory + CPUPlace can be accessed by CPU. + )DOC") .def(py::init<>()) .def("_type", &PlaceIndex) .def("_equals", &IsSamePlace) @@ -800,7 +807,10 @@ All parameter, weight, gradient are variables in Paddle. &IsSamePlace) .def("__str__", string::to_string); - py::class_(m, "CUDAPinnedPlace") + py::class_(m, "CUDAPinnedPlace", R"DOC( + CUDAPinnedPlace is a descriptor of a device. The memory of CUDAPinnedPlace + can be accessed by GPU and CPU. + )DOC") .def("__init__", [](platform::CUDAPinnedPlace &self) { #ifndef PADDLE_WITH_CUDA