29 lines
1.1 KiB
Diff
29 lines
1.1 KiB
Diff
diff --git a/third_party/amd/backend/driver.py b/third_party/amd/backend/driver.py
|
|
index ca712f904..0961d2dda 100644
|
|
--- a/third_party/amd/backend/driver.py
|
|
+++ b/third_party/amd/backend/driver.py
|
|
@@ -79,6 +79,9 @@ def _get_path_to_hip_runtime_dylib():
|
|
return mmapped_path
|
|
raise RuntimeError(f"memory mapped '{mmapped_path}' in process does not point to a valid {lib_name}")
|
|
|
|
+ if os.path.isdir("@libhipDir@"):
|
|
+ return ["@libhipDir@"]
|
|
+
|
|
paths = []
|
|
|
|
import site
|
|
diff --git a/third_party/nvidia/backend/driver.py b/third_party/nvidia/backend/driver.py
|
|
index d088ec092..625de2db8 100644
|
|
--- a/third_party/nvidia/backend/driver.py
|
|
+++ b/third_party/nvidia/backend/driver.py
|
|
@@ -23,6 +23,9 @@ def libcuda_dirs():
|
|
if env_libcuda_path:
|
|
return [env_libcuda_path]
|
|
|
|
+ if os.path.exists("@libcudaStubsDir@"):
|
|
+ return ["@libcudaStubsDir@"]
|
|
+
|
|
libs = subprocess.check_output(["/sbin/ldconfig", "-p"]).decode()
|
|
# each line looks like the following:
|
|
# libcuda.so.1 (libc6,x86-64) => /lib/x86_64-linux-gnu/libcuda.so.1
|