From ba0a3a6e21b0a7f54bd24c51419a8e85b9217c3f Mon Sep 17 00:00:00 2001 From: John Pollock Date: Thu, 9 Oct 2025 13:01:20 -0500 Subject: [PATCH] improve documentation on supporting ROCm cuda-enumerated GPUs. --- device_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/device_utils.py b/device_utils.py index 8d93f5b..18d55e1 100644 --- a/device_utils.py +++ b/device_utils.py @@ -17,7 +17,7 @@ def get_device_list(): Returns a comprehensive list of all available devices across all types: - CPU (always available) - - CUDA devices (NVIDIA GPUs) + - CUDA devices (NVIDIA GPUs + AMD w/ ROCm GPUs) - XPU devices (Intel GPUs) - NPU devices (Ascend NPUs from Huawei) - MLU devices (Cambricon MLUs)