From ae94e9c9e8e30723dbefb5993c3157ccfb4e0797 Mon Sep 17 00:00:00 2001
From: simon-veitner-redhat <sveitner@redhat.com>
Date: Wed, 14 Oct 2026 20:58:35 +0200
Subject: [PATCH] [Bugfix] Don't crash DeepGEMM MegaMoE warmup when
 get_bf16_mega_gate_config is missing (#61657)

Signed-off-by: Simon Veitner <sveitner@redhat.com>
Co-authored-by: Claude <noreply@anthropic.com>
---
 vllm/model_executor/warmup/deep_gemm_warmup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vllm/model_executor/warmup/deep_gemm_warmup.py b/vllm/model_executor/warmup/deep_gemm_warmup.py
index 309c66f07e..9f7d7c7e4e 100644
--- a/vllm/model_executor/warmup/deep_gemm_warmup.py
+++ b/vllm/model_executor/warmup/deep_gemm_warmup.py
@@ -366,7 +366,7 @@ def _get_fp8_fp4_mega_moe_m_values(
             dg.get_bf16_mega_gate_config(
                 num_tokens, gate.input_size, gate.output_size, experts.top_k
             )
-            if uses_mega_gate
+            if uses_mega_gate and hasattr(dg, "get_bf16_mega_gate_config")
             else {}
         )
         block_m = dg.get_block_m_for_mega_moe(