Unverified Commit 0ab23c2b authored by jiahanc's avatar jiahanc Committed by GitHub
Browse files

[fix] fix SM check for Flashinfer TRTLLM MOE (#30314)


Signed-off-by: default avatarjiahanc <173873397+jiahanc@users.noreply.github.com>
parent 48661d27
......@@ -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 "
......
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