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: