@@ -2444,10 +2444,12 @@ for host in "${ALL_HOSTS[@]}"; do
2444
2444
# problematic for overwriting/stripping symbols. Thus, write
2445
2445
# permission is added here.
2446
2446
for lib_name in tensorflow tensorflow_framework; do
2447
- lib=" lib${lib_name} .so"
2447
+ lib=" lib${lib_name} .so*"
2448
+ dylib=" lib${lib_name} *.dylib"
2448
2449
rm -f " ${TF_LIB_DIR} /${lib} "
2449
- chmod +w " ${TENSORFLOW_HOST_LIB_DIR} /${lib} "
2450
- cp -p " ${TENSORFLOW_HOST_LIB_DIR} /${lib} " " ${TF_LIB_DIR} "
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} " \;
2451
2453
done
2452
2454
2453
2455
if [[ ! " ${TENSORFLOW_TARGET_LIB_DIR} " ]] ; then
@@ -3828,10 +3830,10 @@ for host in "${ALL_HOSTS[@]}"; do
3828
3830
mkdir -p " ${TF_DEST_DIR} "
3829
3831
for lib_name in tensorflow tensorflow_framework
3830
3832
do
3831
- lib=" lib${lib_name} .so"
3833
+ lib=" lib${lib_name} .so*"
3834
+ dylib=" lib${lib_name} *.dylib"
3832
3835
echo " ${TF_LIBDIR} /${lib} => ${TF_DEST_DIR} "
3833
- rm -f " ${TF_DEST_DIR} /${lib} "
3834
- cp -a " ${TF_LIBDIR} /${lib} " " ${TF_DEST_DIR} "
3836
+ find " ${TF_LIBDIR} " \( -name " ${lib} " -o -name " ${dylib} " \) -exec cp -a {} " ${TF_DEST_DIR} " \;
3835
3837
done
3836
3838
continue
3837
3839
;;
0 commit comments