Skip to content

Commit f0b04db

Browse files
committed
More specific version cp
1 parent 5f1c4be commit f0b04db

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

build.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff 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),

0 commit comments

Comments
 (0)