Skip to content

GitLab

Unverified Commit 0e00d40e authored by Roger Wang's avatar Roger Wang Committed by GitHub
Browse files

[V1][Bugfix] Fix typo in MoE TPU checking (#15927) 


Signed-off-by: default avatarRoger Wang <ywang@roblox.com>
parent c920e012
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
vllm/model_executor/layers/fused_moe/layer.py
View file @ 0e00d40e
...@@ -309,7 +309,7 @@ class UnquantizedFusedMoEMethod(FusedMoEMethodBase, CustomOp): ...@@ -309,7 +309,7 @@ class UnquantizedFusedMoEMethod(FusedMoEMethodBase, CustomOp):
expert_map=expert_map, expert_map=expert_map,
renormalize=renormalize) renormalize=renormalize)
forward_native = forward_tpu if current_platform.is_tpu else forward_cuda forward_native = forward_tpu if current_platform.is_tpu() else forward_cuda
def determine_expert_map( def determine_expert_map(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment