diff --git a/vllm/_custom_ops.py b/vllm/_custom_ops.py
index 1d38bcade0..dcecce71b3 100644
--- a/vllm/_custom_ops.py
+++ b/vllm/_custom_ops.py
@@ -2274,6 +2274,19 @@ def moe_align_block_size(
     expert_map: torch.Tensor | None = None,
     scatter_idx: torch.Tensor | None = None,
 ) -> None:
+    if current_platform.is_xpu():
+        if scatter_idx is not None:
+            raise NotImplementedError("scatter_idx is not supported on XPU")
+        torch.ops._moe_C.moe_align_block_size(
+            topk_ids,
+            num_experts,
+            block_size,
+            sorted_token_ids,
+            experts_ids,
+            num_tokens_post_pad,
+            expert_map,
+        )
+        return
     torch.ops._moe_C.moe_align_block_size(
         topk_ids,
         num_experts,