Unverified Commit 26990d25 authored by Robert Shaw's avatar Robert Shaw Committed by GitHub
Browse files

[Bugfix] Update device name for H200 detection (#28349)


Signed-off-by: default avatarRobert Shaw <114415538+robertgshaw2-redhat@users.noreply.github.com>
parent d9ab1ad9
......@@ -819,8 +819,8 @@ def get_config_file_name(
) -> str:
device_name = current_platform.get_device_name().replace(" ", "_")
# Set device_name to H200 if a device from the H200 family is detected
if "H200" in device_name:
device_name = "H200"
if "H200" in device_name.split("_"):
device_name = "NVIDIA_H200"
dtype_selector = "" if not dtype else f",dtype={dtype}"
block_shape_selector = (
"" if not block_shape or not all(block_shape) else f",block_shape={block_shape}"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment