Skip to content

GitLab

Unverified Commit b9018a3f authored by wangxiyuan's avatar wangxiyuan Committed by GitHub
Browse files

[BugFix] Fix import error for fused_moe (#18642) 


Signed-off-by: default avatarwangxiyuan <wangxiyuan1007@gmail.com>
parent 4ceafb62
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
vllm/model_executor/layers/fused_moe/layer.py
View file @ b9018a3f
......@@ -41,6 +41,7 @@ if current_platform.is_cuda_alike():
from .pplx_prepare_finalize import PplxPrepareAndFinalize
else:
fused_experts = None # type: ignore
FusedMoEPermuteExpertsUnpermute = None # type: ignore
FusedMoEPrepareAndFinalize = None # type: ignore
if is_rocm_aiter_moe_enabled():
from vllm.model_executor.layers.fused_moe.rocm_aiter_fused_moe import ( # noqa: E501
......
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