mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-06-05 23:02:17 +08:00
[Bug] Fix has_flashinfer_moe Import Error when it is not installed (#21634)
Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
parent
1cd6eaba54
commit
c215f5c877
@ -82,7 +82,8 @@ autotune = _lazy_import_wrapper(
|
|||||||
@functools.cache
|
@functools.cache
|
||||||
def has_flashinfer_moe() -> bool:
|
def has_flashinfer_moe() -> bool:
|
||||||
"""Return ``True`` if FlashInfer MoE module is available."""
|
"""Return ``True`` if FlashInfer MoE module is available."""
|
||||||
return importlib.util.find_spec("flashinfer.fused_moe") is not None
|
return has_flashinfer() and importlib.util.find_spec(
|
||||||
|
"flashinfer.fused_moe") is not None
|
||||||
|
|
||||||
|
|
||||||
@functools.cache
|
@functools.cache
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user