File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -2014,13 +2014,19 @@ def backend_build(
20142014
20152015 cmake_script .mkdir (os .path .join (install_dir , "backends" ))
20162016 cmake_script .rmdir (os .path .join (install_dir , "backends" , be ))
2017-
2017+
20182018 # The python library version available for install via 'yum install python3.X-devel' does not
20192019 # match the version of python inside the RHEL base container. This means that python packages
20202020 # installed within the container will not be picked up by the python backend stub process pybind
20212021 # bindings. It must instead must be installed via pyenv. We package it here for better usability.
20222022 if target_platform () == "rhel" and be == "python" :
2023- cmake_script .cp ("/usr/lib64/libpython3*" , os .path .join (repo_install_dir , "backends" , be ))
2023+ major_minor_version = "." .join (
2024+ (TRITON_VERSION_MAP [FLAGS .version ][7 ]).split ("." )[:2 ]
2025+ )
2026+ version_matched_files = "/usr/lib64/libpython" + major_minor_version + "*"
2027+ cmake_script .cp (
2028+ version_matched_files , os .path .join (repo_install_dir , "backends" , be )
2029+ )
20242030
20252031 cmake_script .cpdir (
20262032 os .path .join (repo_install_dir , "backends" , be ),
You can’t perform that action at this time.
0 commit comments