mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-30 13:17:10 +08:00
[fix] fix SM check for Flashinfer TRTLLM MOE (#30314)
Signed-off-by: jiahanc <173873397+jiahanc@users.noreply.github.com>
This commit is contained in:
parent
48661d275f
commit
0ab23c2b2b
@ -290,7 +290,7 @@ def get_flashinfer_moe_backend() -> FlashinferMoeBackend:
|
|||||||
if flashinfer_moe_backend in backend_map:
|
if flashinfer_moe_backend in backend_map:
|
||||||
if (
|
if (
|
||||||
flashinfer_moe_backend == "latency"
|
flashinfer_moe_backend == "latency"
|
||||||
and not current_platform.is_device_capability(100)
|
and not current_platform.has_device_capability(100)
|
||||||
):
|
):
|
||||||
logger.info_once(
|
logger.info_once(
|
||||||
"Flashinfer TRTLLM MOE backend is only supported on "
|
"Flashinfer TRTLLM MOE backend is only supported on "
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user