mirror of
https://github.com/vladmandic/automatic
synced 2026-09-18 16:54:33 +02:00
Merge pull request #4230 from lbeltrame/fix-hip-lib-path
Fix HIP library detection (ROCm)
This commit is contained in:
+7
-1
@@ -59,7 +59,13 @@ class ROCmEnvironment(Environment):
|
||||
super().__init__(lib)
|
||||
break
|
||||
else:
|
||||
super().__init__(os.path.join(path, "lib", "libamdhip64.so"))
|
||||
# Check 64bit path first, then fall back if it doesn't exist
|
||||
# FIXME: This may be brittle on Debian-based distributions
|
||||
joined_path = os.path.join(path, "lib64", "libamdhip64.so")
|
||||
if not os.path.exists(joined_path):
|
||||
joined_path = os.path.join(path, "lib", "libamdhip64.so")
|
||||
|
||||
super().__init__(joined_path)
|
||||
self.path = path
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user