[XPU] Route to fused_qk_rmsnorm_rope_gate triton kernel (#56096)
Signed-off-by: Marceli Fylcek <marceli.fylcek@intel.com> Co-authored-by: Kunshang Ji <kunshang.ji@intel.com>
Marceli Fylcek committed
Sep 16, 2026 at 05:05 UTC
ccbaed5b91e09f528180db363af7ebb9d4fafe32
1 file changed
+1
-1
vllm/model_executor/models/qwen3_next.py
+1
-1
@@ -378,7 +378,7 @@ class Qwen3NextAttention(nn.Module):
378
self.use_fused_qk_norm_rope_gate = (
379
self.attn_output_gate
380
and getattr(self.rotary_emb, "is_neox_style", False)
381
- and current_platform.is_cuda()
381
+ and (current_platform.is_cuda() or current_platform.is_xpu())
382
and supports_dtype
383
and (text_only or supports_mrope)
384
)