Skip to content

Commit 26990d2

Browse files
[Bugfix] Update device name for H200 detection (vllm-project#28349)
Signed-off-by: Robert Shaw <114415538+robertgshaw2-redhat@users.noreply.github.com>
1 parent d9ab1ad commit 26990d2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm/model_executor/layers/fused_moe/fused_moe.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -819,8 +819,8 @@ def get_config_file_name(
819819
) -> str:
820820
device_name = current_platform.get_device_name().replace(" ", "_")
821821
# Set device_name to H200 if a device from the H200 family is detected
822-
if "H200" in device_name:
823-
device_name = "H200"
822+
if "H200" in device_name.split("_"):
823+
device_name = "NVIDIA_H200"
824824
dtype_selector = "" if not dtype else f",dtype={dtype}"
825825
block_shape_selector = (
826826
"" if not block_shape or not all(block_shape) else f",block_shape={block_shape}"

0 commit comments

Comments
 (0)