We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b04ae1e commit ba0a3a6Copy full SHA for ba0a3a6
1 file changed
device_utils.py
@@ -17,7 +17,7 @@ def get_device_list():
17
18
Returns a comprehensive list of all available devices across all types:
19
- CPU (always available)
20
- - CUDA devices (NVIDIA GPUs)
+ - CUDA devices (NVIDIA GPUs + AMD w/ ROCm GPUs)
21
- XPU devices (Intel GPUs)
22
- NPU devices (Ascend NPUs from Huawei)
23
- MLU devices (Cambricon MLUs)
0 commit comments