[AMD] Fix AMD Multimodal Test - skip nvfp4 tests (#23045)
This commit is contained in:
@@ -361,7 +361,12 @@ if not current_platform.is_hip():
|
|||||||
T2V_sampling_params,
|
T2V_sampling_params,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
ONE_GPU_CASES_C = [
|
# Skip all ModelOpt tests on AMD: FP8 requires torch._scaled_mm (HIPBLAS_STATUS_NOT_SUPPORTED
|
||||||
|
# on ROCm), NVFP4 requires flashinfer or sgl_kernel FP4 kernels (CUDA-only)
|
||||||
|
if current_platform.is_hip():
|
||||||
|
ONE_GPU_CASES_C = []
|
||||||
|
else:
|
||||||
|
ONE_GPU_CASES_C = [
|
||||||
_make_modelopt_ci_case(
|
_make_modelopt_ci_case(
|
||||||
"flux1_modelopt_fp8_t2i",
|
"flux1_modelopt_fp8_t2i",
|
||||||
model_path=DEFAULT_FLUX_1_DEV_MODEL_NAME_FOR_TEST,
|
model_path=DEFAULT_FLUX_1_DEV_MODEL_NAME_FOR_TEST,
|
||||||
@@ -407,7 +412,7 @@ ONE_GPU_CASES_C = [
|
|||||||
extras=["--transformer-path", MODELOPT_WAN22_NVFP4_TRANSFORMER],
|
extras=["--transformer-path", MODELOPT_WAN22_NVFP4_TRANSFORMER],
|
||||||
env_vars=MODELOPT_NVFP4_B200_ENV_VARS,
|
env_vars=MODELOPT_NVFP4_B200_ENV_VARS,
|
||||||
),
|
),
|
||||||
]
|
]
|
||||||
|
|
||||||
TWO_GPU_CASES_A = [
|
TWO_GPU_CASES_A = [
|
||||||
DiffusionTestCase(
|
DiffusionTestCase(
|
||||||
|
|||||||
Reference in New Issue
Block a user