[ROCm] Make amdsmi import optional for other platforms (#13460)

This commit is contained in:
Cyrus Leung
2025-02-18 19:15:56 +08:00
committed by GitHub
parent 29fc5772c4
commit 435b502a6e

View File

@@ -5,8 +5,6 @@ from functools import lru_cache, wraps
from typing import TYPE_CHECKING, Dict, List, Optional
import torch
from amdsmi import (amdsmi_get_gpu_asic_info, amdsmi_get_processor_handles,
amdsmi_init, amdsmi_shut_down)
import vllm.envs as envs
from vllm.logger import init_logger
@@ -20,6 +18,12 @@ else:
logger = init_logger(__name__)
try:
from amdsmi import (amdsmi_get_gpu_asic_info, amdsmi_get_processor_handles,
amdsmi_init, amdsmi_shut_down)
except ImportError as e:
logger.warning("Failed to import from amdsmi with %r", e)
try:
import vllm._C # noqa: F401
except ImportError as e: