Skip to content

GitLab

Unverified Commit 79e799eb authored by bnellnm's avatar bnellnm Committed by GitHub
Browse files

[Bugfix] Temporarily disable B200 fp4 MoE layer tests (#40057) 


Signed-off-by: default avatarBill Nell <bnell@redhat.com>
parent c4e601c7
Show whitespace changes
Inline Side-by-side
Showing with 8 additions and 0 deletions
tests/kernels/moe/test_moe_layer.py
View file @ 79e799eb
...@@ -465,6 +465,14 @@ def is_valid_config(config: MoETestConfig) -> tuple[bool, str | None]: ...@@ -465,6 +465,14 @@ def is_valid_config(config: MoETestConfig) -> tuple[bool, str | None]:
if config.enable_eplb and config.ep_size == 1: if config.enable_eplb and config.ep_size == 1:
return False, "EPLB only works with EP+DP" return False, "EPLB only works with EP+DP"
# Disable fp4 tests until flashinfer is updated or the Dockerfile is
# modified to install cublasLt.h. See #39525.
if (
config.quantization == "modelopt_fp4"
and current_platform.is_device_capability_family(100)
):
return False, "Temporarily skip until #39525 is resolved"
return True, None return True, None
......
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