From afe9eb408ee1191cd57a68d46b6ce2860b1b41e1 Mon Sep 17 00:00:00 2001
From: elvischenv <219235043+elvischenv@users.noreply.github.com>
Date: Thu, 4 Dec 2025 02:50:53 +0800
Subject: [PATCH] [Bugfix] Fix flashinfer ar+norm kernel not available issue
 (#29960)

Signed-off-by: elvischenv <219235043+elvischenv@users.noreply.github.com>
---
 vllm/compilation/fix_functionalization.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/vllm/compilation/fix_functionalization.py b/vllm/compilation/fix_functionalization.py
index 76068f86ebfb3..2625562aadd36 100644
--- a/vllm/compilation/fix_functionalization.py
+++ b/vllm/compilation/fix_functionalization.py
@@ -104,7 +104,8 @@ class FixFunctionalizationPass(VllmInductorPass):
                 mutated_args = {1: "result"}
                 self.defunctionalize(graph, node, mutated_args)
             elif (
-                at_target
+                hasattr(torch.ops.vllm, "flashinfer_trtllm_fused_allreduce_norm")
+                and at_target
                 == torch.ops.vllm.flashinfer_trtllm_fused_allreduce_norm.default
             ):
                 mutated_args = {