()
| 157 | |
| 158 | |
| 159 | def cuda_device_count() -> int: |
| 160 | if not paddle.device.is_compiled_with_cuda(): |
| 161 | return 0 |
| 162 | |
| 163 | device_count = paddle.device.cuda.device_count() |
| 164 | return device_count |
| 165 | |
| 166 | |
| 167 | def xpu_device_count() -> int: |
no outgoing calls