Whether this whl package can be used to run the model on AMD or Hygon GPU(ROCm). Returns: Bool: `True` if ROCm is currently available, otherwise `False`. Examples: .. code-block:: pycon >>> import paddle >>> support_gpu = paddle.device.is_compi
()
| 1012 | |
| 1013 | |
| 1014 | def is_compiled_with_rocm() -> bool: |
| 1015 | """ |
| 1016 | Whether this whl package can be used to run the model on AMD or Hygon GPU(ROCm). |
| 1017 | |
| 1018 | Returns: |
| 1019 | Bool: `True` if ROCm is currently available, otherwise `False`. |
| 1020 | |
| 1021 | Examples: |
| 1022 | .. code-block:: pycon |
| 1023 | |
| 1024 | >>> import paddle |
| 1025 | >>> support_gpu = paddle.device.is_compiled_with_rocm() |
| 1026 | """ |
| 1027 | return core.is_compiled_with_rocm() |
| 1028 | |
| 1029 | |
| 1030 | def cuda_places( |
no outgoing calls
no test coverage detected