Skip to content

Commit 0e00d40

Browse files
authored
[V1][Bugfix] Fix typo in MoE TPU checking (#15927)
Signed-off-by: Roger Wang <ywang@roblox.com>
1 parent c920e01 commit 0e00d40

File tree

1 file changed

+1
-1
lines changed
  • vllm/model_executor/layers/fused_moe

1 file changed

+1
-1
lines changed

vllm/model_executor/layers/fused_moe/layer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -309,7 +309,7 @@ def forward_tpu(
309309
expert_map=expert_map,
310310
renormalize=renormalize)
311311

312-
forward_native = forward_tpu if current_platform.is_tpu else forward_cuda
312+
forward_native = forward_tpu if current_platform.is_tpu() else forward_cuda
313313

314314

315315
def determine_expert_map(

0 commit comments

Comments
 (0)