xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-07-31 21:41:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

[V1][Minor] Do not print attn backend twice (#13985)

Browse Source 
Signed-off-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
This commit is contained in:
Woosuk Kwon 2025-02-28 23:09:14 -08:00 committed by GitHub
parent fdcc405346
commit 3b5567a209
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
vllm/platforms/cuda.py
View File

@ -178,7 +178,8 @@ class CudaPlatformBase(Platform):
block_size) block_size)
else: else:
if use_v1: if use_v1:
logger.info("Using FlashMLA backend on V1 engine.") logger.info_once(
"Using FlashMLA backend on V1 engine.")
return ("vllm.v1.attention.backends.mla." return ("vllm.v1.attention.backends.mla."
"flashmla.FlashMLABackend") "flashmla.FlashMLABackend")
else: else:
@ -187,14 +188,14 @@ class CudaPlatformBase(Platform):
"flashmla.FlashMLABackend") "flashmla.FlashMLABackend")
if use_v1: if use_v1:
logger.info("Using Triton MLA backend on V1 engine.") logger.info_once("Using Triton MLA backend on V1 engine.")
return ("vllm.v1.attention.backends.mla." return ("vllm.v1.attention.backends.mla."
"triton_mla.TritonMLABackend") "triton_mla.TritonMLABackend")
else: else:
logger.info("Using Triton MLA backend.") logger.info("Using Triton MLA backend.")
return "vllm.attention.backends.triton_mla.TritonMLABackend" return "vllm.attention.backends.triton_mla.TritonMLABackend"
if use_v1: if use_v1:
logger.info("Using Flash Attention backend on V1 engine.") logger.info_once("Using Flash Attention backend on V1 engine.")
return ("vllm.v1.attention.backends.flash_attn." return ("vllm.v1.attention.backends.flash_attn."
"FlashAttentionBackend") "FlashAttentionBackend")
if selected_backend == _Backend.FLASHINFER: if selected_backend == _Backend.FLASHINFER: