add custom_device api (#5585)

* add custom_device api * fix bug
PaddlePaddle · Jan 16, 2023 · 3bb20e9 · 3bb20e9
1 parent 2ffdcdf
commit 3bb20e9
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/docs/api/paddle/device/is_compiled_with_custom_device_cn.rst b/docs/api/paddle/device/is_compiled_with_custom_device_cn.rst
@@ -0,0 +1,17 @@
+.. _cn_api_fluid_is_compiled_with_custom_device:
+
+is_compiled_with_custom_device
+-------------------------------
+
+.. py:function:: paddle.device.is_compiled_with_custom_device(device_type)
+
+检查 ``whl`` 包是否可以被用来在指定类型的自定义新硬件上运行模型
+
+返回
+::::::::::::
+bool，支持指定 device_type 则为 True，否则为 False。
+
+代码示例
+::::::::::::
+
+COPY-FROM: paddle.device.is_compiled_with_custom_device