diff --git a/python/sglang/srt/utils.py b/python/sglang/srt/utils.py
index c19d521a0..04372bac1 100644
--- a/python/sglang/srt/utils.py
+++ b/python/sglang/srt/utils.py
@@ -879,7 +879,7 @@ def get_amdgpu_memory_capacity():
         # Run rocm-smi and capture the output
         result = subprocess.run(
             [
-                "rocminfo | grep 'gfx94' -A 100 | grep 'Pool 1' -A 5 | grep 'Size:' | awk '{print $2}'"
+                "rocminfo | grep 'gfx' -A 100 | grep 'Pool 1' -A 5 | grep 'Size:' | awk '{print $2}'"
             ],
             stdout=subprocess.PIPE,
             stderr=subprocess.PIPE,