Add ROCm 6.1.2 support. (ZLUDA)

vladmandic · Jul 10, 2024 · a696775 · a696775
1 parent f3d4000
commit a696775
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/modules/zluda_installer.py b/modules/zluda_installer.py
@@ -13,7 +13,7 @@
     '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)