We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 053e304 commit 151fdfcCopy full SHA for 151fdfc
vllm/platforms/rocm.py
@@ -5,8 +5,6 @@
5
from typing import TYPE_CHECKING, Dict, List, Optional
6
7
import torch
8
-from amdsmi import (amdsmi_get_gpu_asic_info, amdsmi_get_processor_handles,
9
- amdsmi_init, amdsmi_shut_down)
10
11
import vllm.envs as envs
12
from vllm.logger import init_logger
@@ -20,6 +18,12 @@
20
18
21
19
logger = init_logger(__name__)
22
+try:
+ from amdsmi import (amdsmi_get_gpu_asic_info, amdsmi_get_processor_handles,
23
+ amdsmi_init, amdsmi_shut_down)
24
+except ImportError as e:
25
+ logger.warning("Failed to import from amdsmi with %r", e)
26
+
27
try:
28
import vllm._C # noqa: F401
29
except ImportError as e:
0 commit comments