Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-10 23:08:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add ROCm 6.1.2 support. (ZLUDA)

Browse Source
This commit is contained in:
Seunghoon Lee
2024-07-11 02:53:12 +09:00
parent f3d4000e83
commit a696775cf0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/zluda_installer.py
+2 -2
View File
@@ -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)