diff --git a/paddle/fluid/pybind/place.cc b/paddle/fluid/pybind/place.cc index 72e672792d5e61b3d2f14d4e15c92ca2a6106d85..b39427f8b3962fdd17fe336f7059950b0a005257 100644 --- a/paddle/fluid/pybind/place.cc +++ b/paddle/fluid/pybind/place.cc @@ -553,8 +553,11 @@ void BindPlace(pybind11::module &m) { // NOLINT Examples: .. code-block:: python + + # required: npu + import paddle - npu_place = paddle.NPUPlace(0) + place = paddle.NPUPlace(0) )DOC"); g_npuplace_pytype = reinterpret_cast(npuplace.ptr());