Skip to content

Commit ba0a3a6

Browse files
committed
improve documentation on supporting ROCm cuda-enumerated GPUs.
1 parent b04ae1e commit ba0a3a6

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

device_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def get_device_list():
1717
1818
Returns a comprehensive list of all available devices across all types:
1919
- CPU (always available)
20-
- CUDA devices (NVIDIA GPUs)
20+
- CUDA devices (NVIDIA GPUs + AMD w/ ROCm GPUs)
2121
- XPU devices (Intel GPUs)
2222
- NPU devices (Ascend NPUs from Huawei)
2323
- MLU devices (Cambricon MLUs)

0 commit comments

Comments
 (0)