执行摘要

• 一句话：回退 ROCm MXFP4 MoE 的 TRITON_UNFUSED 回落
• 推荐动作：值得重点阅读 select_mxfp4_moe_backend 函数的设计变更，尤其是后端优先级与平台特定回调的取舍。对于 ROCm 量化路径的维护者，该 PR 是必要的基础修正。

功能与动机

PR #45896 为支持 amd/MiniMax-M3-MXFP4 模型，在 ROCm 路径下添加了 Mxfp4MoeBackend.TRITON_UNFUSED 回落。但该后端不遵守 requested_activation_key（激活量化 key），而模型需要激活量化。在 AITER 不可用时，正确的回落应该是 EMULATION 后端，它能正确应用激活量化。将 TRITON_UNFUSED 作为默认回落可能导致量化配置被忽略，影响模型精度或功能。

实现拆解

1. 移除 ROCm 专用回落：在 mxfp4.py 的 select_mxfp4_moe_backend 函数中，删除了 if current_platform.is_rocm(): backend = Mxfp4MoeBackend.TRITON_UNFUSED ... 代码块。原先该代码块在无其他后端支持时，直接返回 TRITON_UNFUSED 而不检查激活 key。

2. 添加候选后端失败原因收集：在遍历候选后端时，新增 unsupported_reasons 列表，记录每个后端的拒绝原因。遍历结束后，如果没有任何后端支持，则拼接所有原因并抛出 NotImplementedError，包含候选后端列表和具体原因，便于调试。

3. 新增验证测试：在 tests/kernels/moe/test_ocp_mx_moe.py 中增加 test_select_mxfp4_moe_backend_raises_with_unsupported_reasons，通过 monkeypatch 模拟所有后端拒绝配置，验证异常消息中包含 "Unsupported reasons"。

4. 删除过时测试：在 tests/quantization/test_gfx950_moe.py 中删除 test_w4a4_falls_back_to_triton_unfused_without_aiter，因为 TRITON_UNFUSED 不再是默认回落，该测试已不适用。

关键文件：

• vllm/model_executor/layers/fused_moe/oracle/mxfp4.py（模块 后端选择；类别 source；类型 data-contract；符号 select_mxfp4_moe_backend, select_deepseek_v4_mxfp4_moe_backend）: 核心文件，修改后端选择逻辑，移除 ROCm 专用 TRITON_UNFUSED 回落，添加失败原因收集
• tests/kernels/moe/test_ocp_mx_moe.py（模块 MXFP4 测试；类别 test；类型 test-coverage；符号 test_select_mxfp4_moe_backend_raises_with_unsupported_reasons, UnsupportedExperts, is_supported_config）: 新增测试验证后端不可用时错误信息包含拒绝原因
• tests/quantization/test_gfx950_moe.py（模块 GFX950 测试；类别 test；类型 test-coverage；符号 test_w4a4_falls_back_to_triton_unfused_without_aiter）: 删除过时的回落测试，因为 TRITON_UNFUSED 不再作为默认回落

关键符号：select_mxfp4_moe_backend, test_select_mxfp4_moe_backend_raises_with_unsupported_reasons, test_w4a4_falls_back_to_triton_unfused_without_aiter

关键源码片段

vllm/model_executor/layers/fused_moe/oracle/mxfp4.py

核心文件，修改后端选择逻辑，移除 ROCm 专用 TRITON_UNFUSED 回落，添加失败原因收集

    # Select kernels in order of backend.
    AVAILABLE_BACKENDS = _filter_by_activation(
        _get_priority_backends_for_gpt_oss(), requested_activation_key
    )
​
    unsupported_reasons = [] # 收集每个候选后端的拒绝原因
    for backend in AVAILABLE_BACKENDS:
        # 确定激活 key：优先使用请求的 key，否则使用后端默认
        act_key = (
            requested_activation_key
            if requested_activation_key is not None
            else _backend_activation_key(backend)
        )
        for k_cls in backend_to_kernel_cls(backend):
            supported, reason = k_cls.is_supported_config(
                k_cls, config, kMxfp4Static, act_key, activation_format
            )
            if supported:
                logger.info_once(_make_log_backend(backend))
                return backend, k_cls
            else:
                logger.debug_once(_make_log_unsupported(backend, reason))
                unsupported_reasons.append((backend, reason)) # 记录失败原因
​
    # 平台特定后端的回落（仅 XPU/CPU 保留特殊路径）
    if current_platform.is_xpu():
        backend = Mxfp4MoeBackend.XPU
        logger.info_once(_make_log_backend(backend))
        return _return_or_raise(backend, config, kMxfp4Static, None, activation_format)
​
    if current_platform.is_cpu():
        backend = Mxfp4MoeBackend.CPU
        logger.info_once(_make_log_backend(backend))
        return _return_or_raise(backend, config, kMxfp4Static, None, activation_format)
​
    # 所有候选后端都不支持，抛出包含详细原因的异常
    unsupported_log = "; ".join(
        f"backend: {backend.value}, reason: {reason}"
        for backend, reason in unsupported_reasons
    )
    raise NotImplementedError(
        "No MXFP4 MoE backend supports the deployment configuration. "
        f"weight_key=kMxfp4Static, activation_key={activation_key}. "
        f"Candidate backends were: "
        f"{[backend.value for backend in AVAILABLE_BACKENDS]}. "
        f"Unsupported reasons: {unsupported_log}."
    )

评论区精华

1. TRITON_UNFUSED 回落合理性：PR 作者指出 TRITON_UNFUSED 不遵守 requested_activation_key，且已被从全局优先级列表移除（#40860），不应作为 ROCm 独占回落。其他后端如 Marlin/Humming 也忽略 activation key，但它们是针对 Nvidia GPU 的性能优化。结论：应 revert 该回落，待 TRITON_UNFUSED 重新加入全局列表后再考虑作为通用回落。

2. 后续跟踪：AndreasKaratzas 询问是否有 issue 或人员跟踪将来将 TRITON_UNFUSED 添加回全局优先级列表。目前没有 issue，但此 PR 是临时修复，未来需全局解决。

3. 测试移除原因：作者解释 test_w4a4_falls_back_to_triton_unfused_without_aiter 因 PR #46629 合并后不再相关，因为回落行为统一经过 oracle；同时 test_minimax_m3_mxfp4_moe_backend 也被移除。

• TRITON_UNFUSED 不应作为 ROCm 默认回落 (design): 确认应 revert 该回落，待 TRITON_UNFUSED 重新加入全局列表后再考虑作为通用回落。
• 后续跟踪 issue (question): 目前没有 issue，但 PR 作者表示此 PR 是临时修复，未来需全局解决。
• 测试移除原因 (testing): 测试删除正确，CI 中该测试已由其他测试覆盖。

风险与影响

• 风险：

  1. 默认行为变更：用户可能依赖隐式 TRITON_UNFUSED 回落，升级后需要显式指定 --moe-backend triton_unfused 或设置 VLLM_ROCM_USE_AITER=1 才能使用同一后端。但此行为本身是错误的，因此风险可控。
  2. 错误信息依赖：新错误消息的完整性依赖各候选后端提供的拒绝原因字符串。如果某后端不提供原因，unsupported_reasons 条目将包含 None 或空字符串，但现有后端均提供。
  3. 测试覆盖：新增测试验证了全部拒绝的场景，但未覆盖部分后端支持的部分拒绝场景。不过现有测试已足够。
     - 影响：影响范围：仅影响 ROCm 平台且使用 MXFP4 MoE 的用户，尤其是运行 MiniMax-M3-MXFP4 等需要使用激活量化的模型。
     影响程度：中低。绝大多数用户已通过 AITER 或显式后端设置获得正确行为；此修复确保默认行为合法，避免了隐式忽略激活量化可能导致的问题。
     团队影响：明确了后端选择策略，简化了 oracle 逻辑，便于未来维护。

• 风险标记：默认行为变更, 后端选择逻辑改动

关联脉络

• PR #45896 Add support for loading amd/MiniMax-M3-MXFP4: 被 revert 的源 PR，引入了错误的 TRITON_UNFUSED 回落
• PR #40860 Remove TRITON_UNFUSED from GPT-OSS priority backends: 此前移除了 TRITON_UNFUSED 从全局优先级列表，本 PR 延续该决定
• PR #46629 Merge related changes for MXFP4 MoE: 合并后影响了回落测试，导致其中一个测试被删除