Skip to content

GitLab

Unverified Commit bd380048 authored by Tim Geypens's avatar Tim Geypens Committed by GitHub
Browse files

fall back after failing ldconfig-based lib loading for cuDNN (#2277) 


Signed-off-by: default avatarTim Geypens <tim.geypens@gmail.com>
parent 05dc1e62
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 9 deletions
transformer_engine/common/__init__.py
View file @ bd380048
...@@ -252,9 +252,7 @@ def _load_cudnn(): ...@@ -252,9 +252,7 @@ def _load_cudnn():
return handle return handle
# Attempt to locate libcudnn via ldconfig # Attempt to locate libcudnn via ldconfig
libs = subprocess.check_output( libs = subprocess.check_output(["ldconfig", "-p"])
f"ldconfig -p | grep 'libcudnn{_get_sys_extension()}'", shell=True
)
libs = libs.decode("utf-8").split("\n") libs = libs.decode("utf-8").split("\n")
sos = [] sos = []
for lib in libs: for lib in libs:
...@@ -284,9 +282,7 @@ def _load_nvrtc(): ...@@ -284,9 +282,7 @@ def _load_nvrtc():
return handle return handle
# Attempt to locate NVRTC via ldconfig # Attempt to locate NVRTC via ldconfig
libs = subprocess.check_output( libs = subprocess.check_output(["ldconfig", "-p"])
f"ldconfig -p | grep 'libnvrtc{_get_sys_extension()}'", shell=True
)
libs = libs.decode("utf-8").split("\n") libs = libs.decode("utf-8").split("\n")
sos = [] sos = []
for lib in libs: for lib in libs:
...@@ -316,9 +312,7 @@ def _load_curand(): ...@@ -316,9 +312,7 @@ def _load_curand():
return handle return handle
# Attempt to locate cuRAND via ldconfig # Attempt to locate cuRAND via ldconfig
libs = subprocess.check_output( libs = subprocess.check_output(["ldconfig", "-p"])
f"ldconfig -p | grep 'libcurand{_get_sys_extension()}'", shell=True
)
libs = libs.decode("utf-8").split("\n") libs = libs.decode("utf-8").split("\n")
sos = [] sos = []
for lib in libs: for lib in libs:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment