changed docstring for AMDGPUs to only support 6.0

ml-energy · May 2, 2024 · c217baf · c217baf
1 parent a6da5cf
commit c217baf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/zeus/device/gpu.py b/zeus/device/gpu.py
@@ -805,7 +805,7 @@ class AMDGPUs(GPUs):
     ```
 
     Note: This class instantiates (grabs the handle, by calling `amdsmi.amdsmi_get_processor_handles()`) all GPUs that are visible to the system, as determined by the `HIP_VISIBLE_DEVICES` environment variable if set.
-    Supports ROCM 5.7 and ROCM 6.0.
+    !!! Only supports ROCM 6.0 !!!
     """
 
     def __init__(self, ensure_homogeneous: bool = False) -> None: