From 73f48ce559e230fd0d738c52cb2e99bd0dd08754 Mon Sep 17 00:00:00 2001 From: Yanan Cao Date: Wed, 1 Apr 2026 21:30:31 -0700 Subject: [PATCH] [Kernel] [Helion] Use warning_once in get_gpu_name to prevent log spam (#38743) Signed-off-by: Yanan Cao Co-authored-by: Claude Sonnet 4 --- vllm/kernels/helion/utils.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/vllm/kernels/helion/utils.py b/vllm/kernels/helion/utils.py index 5ff8046c7..130d79093 100644 --- a/vllm/kernels/helion/utils.py +++ b/vllm/kernels/helion/utils.py @@ -2,11 +2,10 @@ # SPDX-FileCopyrightText: Copyright contributors to the vLLM project """Utility functions for Helion kernel management.""" -import logging - +from vllm.logger import init_logger from vllm.platforms import current_platform -logger = logging.getLogger(__name__) +logger = init_logger(__name__) # Maps known variant GPU names (after lowercase/underscore normalization) # to their canonical form. @@ -49,7 +48,7 @@ _GPU_NAME_ALIASES: dict[str, str] = { def get_gpu_name(device_id: int | None = None) -> str: if device_id is None: - logger.warning( + logger.warning_once( "get_gpu_name() called without device_id, defaulting to 0. " "This may return the wrong device name in multi-node setups." )