[Model] Use the same fused_moe configs for all H200 devices (#23642)
Signed-off-by: Roger Meier <r.meier@siemens.com>
This commit is contained in:
@ -818,6 +818,9 @@ def get_config_file_name(
|
||||
E: int, N: int, dtype: str | None, block_shape: list[int] | None = None
|
||||
) -> 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"
|
||||
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}"
|
||||
|
||||
Reference in New Issue
Block a user