From 72297d859ba34e019da4d7e5de58abbbd59961d4 Mon Sep 17 00:00:00 2001 From: Yintong Lu Date: Wed, 29 Jul 2026 18:59:58 +0800 Subject: [PATCH] [XPU] Route weightless RMSNorm to _C dispatch (#47121) Signed-off-by: Yintong Lu Co-authored-by: Yongqi Wang Co-authored-by: Kunshang Ji --- vllm/kernels/xpu_ops.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/vllm/kernels/xpu_ops.py b/vllm/kernels/xpu_ops.py index df82962d80..6838883238 100644 --- a/vllm/kernels/xpu_ops.py +++ b/vllm/kernels/xpu_ops.py @@ -30,9 +30,6 @@ def rms_norm( x: Tensor, weight: Tensor | None, epsilon: float, variance_size: int | None = None ) -> Tensor: assert variance_size is None - if weight is None: - # Kernel requires weight tensor, pass ones - weight = torch.ones(x.shape[-1], device=x.device, dtype=x.dtype) output = torch.empty(x.shape, device=x.device, dtype=x.dtype) torch.ops._C.rms_norm(output, x, weight, epsilon) return output @@ -58,8 +55,5 @@ def fused_add_rms_norm( variance_size: int | None = None, ) -> tuple[Tensor, Tensor]: assert variance_size is None - if weight is None: - # Kernel requires weight tensor, pass ones - weight = torch.ones(x.shape[-1], device=x.device, dtype=x.dtype) torch.ops._C.fused_add_rms_norm(x, x_residual, weight, epsilon) return x, x_residual