diff --git a/projects/rocprofiler-systems/cmake/Templates/console-script.in b/projects/rocprofiler-systems/cmake/Templates/console-script.in index 76a5fafbd8..5d5f088c8d 100755 --- a/projects/rocprofiler-systems/cmake/Templates/console-script.in +++ b/projects/rocprofiler-systems/cmake/Templates/console-script.in @@ -14,4 +14,25 @@ run-script() eval $@ } +add_so_paths_to_ld_library_path() +{ + local python_name_version=$(basename ${PYTHON_EXECUTABLE}) + local libdir=$(dirname $(dirname $PYTHON_EXECUTABLE))/lib/${python_name_version}/site-packages + + if [ -d "$libdir" ]; then + local so_dirs=$(find "$libdir" -name "*.so" -type f -exec dirname {} \; 2>/dev/null | sort -u) + + for dir in $so_dirs; do + if [ -d "$dir" ] && [[ ":$LD_LIBRARY_PATH:" != *":$dir:"* ]]; then + export LD_LIBRARY_PATH="${LD_LIBRARY_PATH:+$LD_LIBRARY_PATH:}${dir}" + fi + done + fi +} + +OLD_LD_LIBRARY_PATH=${LD_LIBRARY_PATH:-} +add_so_paths_to_ld_library_path + run-script ${PYTHON_EXECUTABLE} -m @SCRIPT_SUBMODULE@ "$(printf ' %q' "$@")" + +export LD_LIBRARY_PATH=${OLD_LD_LIBRARY_PATH}