From a696775cf0d4bcf60df3219c2dbe0e01fe57d830 Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Thu, 11 Jul 2024 02:53:12 +0900 Subject: [PATCH] Add ROCm 6.1.2 support. (ZLUDA) --- modules/zluda_installer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/zluda_installer.py b/modules/zluda_installer.py index d1e12f011..d9d91c0ac 100644 --- a/modules/zluda_installer.py +++ b/modules/zluda_installer.py @@ -13,7 +13,7 @@ DLL_MAPPING = { 'cusparse.dll': 'cusparse64_11.dll', 'nvrtc.dll': 'nvrtc64_112_0.dll', } -HIP_TARGETS = ['rocblas.dll', 'rocsolver.dll', 'hiprtc0507.dll',] +HIP_TARGETS = ['rocblas.dll', 'rocsolver.dll', 'hiprtc0601.dll',] ZLUDA_TARGETS = ('nvcuda.dll', 'nvml.dll',) @@ -23,7 +23,7 @@ def get_path() -> str: def find_hip_sdk() -> Union[str, None]: program_files = os.environ.get('ProgramFiles', r'C:\Program Files') - hip_path_default = rf'{program_files}\AMD\ROCm\5.7' + hip_path_default = rf'{program_files}\AMD\ROCm\6.1' if not os.path.exists(hip_path_default): hip_path_default = None return os.environ.get('HIP_PATH', hip_path_default)