fix(iluvatar): failed to load ixml

Signed-off-by: thxCode <thxcode0824@gmail.com>

Author: thxCode

gpustack_runtime/detector/iluvatar.py

@@ -62,7 +62,7 @@ def is_supported() -> bool:
             pyixml.nvmlShutdown()
             supported = True
         except pyixml.NVMLError:
-            debug_log_exception(logger, "Failed to initialize IXML library")
+            debug_log_exception(logger, "Failed to initialize IXML")
 
         return supported
 
@@ -102,7 +102,7 @@ def detect(self) -> Devices | None:
 
             sys_runtime_ver_original = None
             sys_runtime_ver = None
-            with contextlib.suppress(pyixml.NVMLError):
+            with contextlib.suppress(Exception):
                 sys_runtime_ver_original = pyixml.nvmlSystemGetCudaDriverVersion()
                 sys_runtime_ver_original = ".".join(
                     map(

gpustack_runtime/detector/pyixml/__init__.py

@@ -28,6 +28,7 @@
 # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
 # THE POSSIBILITY OF SUCH DAMAGE.
 #####
+import os
 
 ##
 # Python bindings for the NVML library
@@ -2162,8 +2163,20 @@ def _LoadNvmlLibrary():
                 # Linux path
                 locs = [
                     "libixml.so",
-                    str(Path(__file__).resolve().parent / "libixml.so"),
                 ]
+                for default_path in [
+                    "/usr/local/corex",
+                ]:
+                    iluvatar_path = Path(
+                        os.getenv("COREX_HOME", default_path),
+                    )
+                    if iluvatar_path.exists():
+                        locs.extend(
+                            [
+                                str(iluvatar_path / "lib64/libixml.so"),
+                                str(iluvatar_path / "lib/libixml.so"),
+                            ]
+                        )
                 for loc in locs:
                     try:
                         nvmlLib = CDLL(loc)

gpustack_runtime/detector/thead.py

@@ -64,7 +64,7 @@ def is_supported() -> bool:
             pyhgml.hgmlShutdown()
             supported = True
         except pyhgml.HGMLError:
-            debug_log_exception(logger, "Failed to initialize HGML library")
+            debug_log_exception(logger, "Failed to initialize HGML")
 
         return supported