Unverified Commit 435b502a authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

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

parent 29fc5772
...@@ -5,8 +5,6 @@ from functools import lru_cache, wraps ...@@ -5,8 +5,6 @@ from functools import lru_cache, wraps
from typing import TYPE_CHECKING, Dict, List, Optional from typing import TYPE_CHECKING, Dict, List, Optional
import torch import torch
from amdsmi import (amdsmi_get_gpu_asic_info, amdsmi_get_processor_handles,
amdsmi_init, amdsmi_shut_down)
import vllm.envs as envs import vllm.envs as envs
from vllm.logger import init_logger from vllm.logger import init_logger
...@@ -20,6 +18,12 @@ else: ...@@ -20,6 +18,12 @@ else:
logger = init_logger(__name__) 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: try:
import vllm._C # noqa: F401 import vllm._C # noqa: F401
except ImportError as e: except ImportError as e:
......
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