Skip to content

Commit 9244981

Browse files
authored
restrict set of copied libtensorflow files (#24007)
1 parent a1c12eb commit 9244981

File tree

1 file changed

+9
-10
lines changed

1 file changed

+9
-10
lines changed

utils/build-script-impl

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2444,12 +2444,11 @@ for host in "${ALL_HOSTS[@]}"; do
24442444
# problematic for overwriting/stripping symbols. Thus, write
24452445
# permission is added here.
24462446
for lib_name in tensorflow tensorflow_framework; do
2447-
lib="lib${lib_name}.so*"
2448-
dylib="lib${lib_name}*.dylib"
2449-
rm -f "${TF_LIB_DIR}/${lib}"
2450-
rm -f "${TF_LIB_DIR}/${dylib}"
2451-
find "${TENSORFLOW_HOST_LIB_DIR}" \( -name "${lib}" -o -name "${dylib}" \) -exec chmod +w {} +
2452-
find "${TENSORFLOW_HOST_LIB_DIR}" \( -name "${lib}" -o -name "${dylib}" \) -exec cp -p {} "${TF_LIB_DIR}" \;
2447+
lib=".*lib${lib_name}.so[0-9.]*"
2448+
dylib=".*lib${lib_name}[0-9.]*.dylib"
2449+
find "${TF_LIB_DIR}" \( -regex "${lib}" -o -regex "${dylib}" \) -exec rm -f {} \;
2450+
find "${TENSORFLOW_HOST_LIB_DIR}" \( -regex "${lib}" -o -regex "${dylib}" \) -exec chmod +w {} +
2451+
find "${TENSORFLOW_HOST_LIB_DIR}" \( -regex "${lib}" -o -regex "${dylib}" \) -exec cp -p {} "${TF_LIB_DIR}" \;
24532452
done
24542453

24552454
if [[ ! "${TENSORFLOW_TARGET_LIB_DIR}" ]] ; then
@@ -3830,10 +3829,10 @@ for host in "${ALL_HOSTS[@]}"; do
38303829
mkdir -p "${TF_DEST_DIR}"
38313830
for lib_name in tensorflow tensorflow_framework
38323831
do
3833-
lib="lib${lib_name}.so*"
3834-
dylib="lib${lib_name}*.dylib"
3835-
echo "${TF_LIBDIR}/${lib} => ${TF_DEST_DIR}"
3836-
find "${TF_LIBDIR}" \( -name "${lib}" -o -name "${dylib}" \) -exec cp -a {} "${TF_DEST_DIR}" \;
3832+
lib=".*lib${lib_name}.so[0-9.]*"
3833+
dylib=".*lib${lib_name}[0-9.]*.dylib"
3834+
find "${TF_LIBDIR}" \( -regex "${lib}" -o -regex "${dylib}" \) -exec echo "{} => ${TF_DEST_DIR}" \;
3835+
find "${TF_LIBDIR}" \( -regex "${lib}" -o -regex "${dylib}" \) -exec cp -a {} "${TF_DEST_DIR}" \;
38373836
done
38383837
continue
38393838
;;

0 commit comments

Comments
 (0)