From bdf6501f40d1072f1b99cb2eb17e7e4491fc7f48 Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Sun, 28 Jul 2024 00:34:54 +0900 Subject: [PATCH] fix --- installer.py | 2 +- modules/rocm.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/installer.py b/installer.py index 6c5bd95e6..c52b35ede 100644 --- a/installer.py +++ b/installer.py @@ -475,7 +475,7 @@ def install_rocm_zluda(torch_command): for idx, gpu in enumerate(amd_gpus): gfx_version = gpu.get_gfx_version() if gfx_version is None: - log.debug(f'HSA_OVERRIDE_GFX_VERSION auto config is skipped for {gpu}') + log.debug(f'HSA_OVERRIDE_GFX_VERSION auto config is skipped for {gpu.name}') else: hip_default_device = gpu log.debug(f'ROCm agent used by default: idx={idx} gpu={gpu.name}') diff --git a/modules/rocm.py b/modules/rocm.py index 19d2b79f6..dfd24c133 100644 --- a/modules/rocm.py +++ b/modules/rocm.py @@ -6,6 +6,10 @@ import subprocess from typing import Union, List +HIPBLASLT_TENSILE_LIBPATH = os.environ.get("HIPBLASLT_TENSILE_LIBPATH", None if sys.platform == "win32" # not available + else "/opt/rocm/lib/hipblaslt/library") + + def resolve_link(path_: str) -> str: if not os.path.islink(path_): return path_ @@ -132,8 +136,6 @@ if sys.platform == "win32": is_wsl: bool = False else: - HIPBLASLT_TENSILE_LIBPATH = os.environ.get("HIPBLASLT_TENSILE_LIBPATH", "/opt/rocm/lib/hipblaslt/library") - def find() -> Union[str, None]: rocm_path = shutil.which("hipconfig") if rocm_path is not None: