Relax to include more AMD GPUs (#2319)

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,