From cccd31b898ae9919f3fbdf53b39360f27149a8f2 Mon Sep 17 00:00:00 2001 From: chengduozh Date: Mon, 25 Mar 2019 13:50:09 +0800 Subject: [PATCH] Add doc for CPUPlace CUDAPlace CUDAPinPlace test=develop --- paddle/fluid/pybind/pybind.cc | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) diff --git a/paddle/fluid/pybind/pybind.cc b/paddle/fluid/pybind/pybind.cc index a57083a1444a16..28d87be4e5c1ec 100644 --- a/paddle/fluid/pybind/pybind.cc +++ b/paddle/fluid/pybind/pybind.cc @@ -763,7 +763,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 @@ -783,7 +787,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("_equals", &IsSamePlace) .def("_equals", &IsSamePlace) @@ -792,7 +799,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