[XPU] Fix topk_sigmoid arg mismatch on XPU (#47858)
Signed-off-by: zhenwei-intel <zhenwei.liu@intel.com>
liuzhenwei committed
Jul 8, 2026 at 00:53 UTC
b93cbd7416b073c15e51d34ff7f1b9a2c2337def
1 file changed
+12
vllm/_custom_ops.py
+12
@@ -2246,6 +2246,18 @@ def topk_sigmoid(
2246
e_score_correction_bias: torch.Tensor | None = None,
2247
routed_scaling_factor: float = 1.0,
2248
) -> None:
2249
+ if current_platform.is_xpu():
2250
+ # xpu doesn't support routed_scaling_factor currently, will revert
2251
+ # in next vllm-xpu-kernels bumpup
2252
+ torch.ops._moe_C.topk_sigmoid(
2253
+ topk_weights,
2254
+ topk_ids,
2255
+ token_expert_indices,
2256
+ gating_output,
2257
+ renormalize,
2258
+ e_score_correction_bias,
2259
+ )
2260
+ return
2261
torch.ops._moe_C.topk_sigmoid(
2262
topk_weights,
2263
topk_ids,