[ROCM] add is_compiled_with_rocm api, test=develop (#3526)

* [ROCM] add is_compiled_with_rocm api, test=develop * [ROCM] update api doc, test=develop * [ROCM] address review comment, test=develop
PaddlePaddle · May 31, 2021 · 110d5b9 · 110d5b9
1 parent c91f98b
commit 110d5b9
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/docs/api/paddle/is_compiled_with_rocm_cn.rst b/docs/api/paddle/is_compiled_with_rocm_cn.rst
@@ -0,0 +1,22 @@
+.. _cn_api_paddle_is_compiled_with_rocm:
+
+is_compiled_with_rocm
+-------------------------------
+
+.. py:function:: paddle.is_compiled_with_rocm()
+
+
+
+
+检查 ``whl`` 包是否可以被用来在AMD或海光GPU(ROCm)上运行模型
+
+返回：bool，支持GPU(ROCm)则为True，否则为False。
+
+**示例代码**
+
+.. code-block:: python
+
+    import paddle
+    support_gpu = paddle.is_compiled_with_rocm()
+
+