"ssh:/git@developer.sourcefind.cn:2222/OpenDAS/vllm_cscc.git" did not exist on "53243e5c42b006f56e239c4f946b7ca33b7db5cc"
Unverified Commit 73f48ce5 authored by Yanan Cao's avatar Yanan Cao Committed by GitHub
Browse files

[Kernel] [Helion] Use warning_once in get_gpu_name to prevent log spam (#38743)


Signed-off-by: default avatarYanan Cao <gmagogsfm@gmail.com>
Co-authored-by: default avatarClaude Sonnet 4 <noreply@anthropic.com>
parent 3aab680e
...@@ -2,11 +2,10 @@ ...@@ -2,11 +2,10 @@
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
"""Utility functions for Helion kernel management.""" """Utility functions for Helion kernel management."""
import logging from vllm.logger import init_logger
from vllm.platforms import current_platform from vllm.platforms import current_platform
logger = logging.getLogger(__name__) logger = init_logger(__name__)
# Maps known variant GPU names (after lowercase/underscore normalization) # Maps known variant GPU names (after lowercase/underscore normalization)
# to their canonical form. # to their canonical form.
...@@ -49,7 +48,7 @@ _GPU_NAME_ALIASES: dict[str, str] = { ...@@ -49,7 +48,7 @@ _GPU_NAME_ALIASES: dict[str, str] = {
def get_gpu_name(device_id: int | None = None) -> str: def get_gpu_name(device_id: int | None = None) -> str:
if device_id is None: if device_id is None:
logger.warning( logger.warning_once(
"get_gpu_name() called without device_id, defaulting to 0. " "get_gpu_name() called without device_id, defaulting to 0. "
"This may return the wrong device name in multi-node setups." "This may return the wrong device name in multi-node setups."
) )
......
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