From 69e2d4fb66e8dd9df7e9472df44ae29afc1320d1 Mon Sep 17 00:00:00 2001 From: HAI Date: Mon, 2 Dec 2024 19:05:58 -0800 Subject: [PATCH] Relax to include more AMD GPUs (#2319) --- python/sglang/srt/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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,