Skip to content

Commit 90bbe0a

Browse files
authored
[Log] Warning Once for Cutlass MLA (#23137)
Signed-off-by: yewentao256 <[email protected]>
1 parent e75f342 commit 90bbe0a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

vllm/v1/attention/backends/mla/cutlass_mla.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -115,16 +115,16 @@ def __init__(
115115
self._use_old_cutlass_mla = False
116116
force_old_cutlass = os.environ.get("FORCE_OLD_CUTLASS_MLA", None)
117117
if force_old_cutlass:
118-
logger.warning("Forcing old cutlass mla kernel")
118+
logger.warning_once("Forcing old cutlass mla kernel")
119119
self._use_old_cutlass_mla = True
120120

121121
# TODO: Currently, num_kv_splits is limited to 16 to avoid hanging
122122
# issues. In case the code hangs, use:
123123
# FORCE_NUM_KV_SPLITS=1
124124
force_num_kv_splits = os.environ.get("FORCE_NUM_KV_SPLITS", None)
125125
if force_num_kv_splits:
126-
logger.warning("Forcing num_kv_splits to %d",
127-
int(force_num_kv_splits))
126+
logger.warning_once("Forcing num_kv_splits to %d",
127+
int(force_num_kv_splits))
128128
self._num_kv_splits = int(force_num_kv_splits)
129129
else:
130130
self._num_kv_splits = -1 # => Auto-detect

0 commit comments

Comments
 (0)