mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-29 10:14:01 +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 == "latency"
|
||||
and not current_platform.is_device_capability(100)
|
||||
and not current_platform.has_device_capability(100)
|
||||
):
|
||||
logger.info_once(
|
||||
"Flashinfer TRTLLM MOE backend is only supported on "
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user