Skip to content

Commit 31315c5

Browse files
committed
Update copy file configuration
1 parent 3a7d08d commit 31315c5

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

CMakeLists.txt

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,7 @@ set(OPENCV_LIBS
223223
# in some cases leads to segmentation faults.
224224
if (${TRITON_PYTORCH_DOCKER_BUILD})
225225
string(REPLACE ";" " " LIBTORCH_LIBS_STR "${LIBTORCH_LIBS}")
226+
string(RANDOM 8 "abcdefghijklmnopqrstuvwxyz" random_id)
226227

227228
add_custom_command(
228229
OUTPUT
@@ -263,8 +264,8 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
263264
COMMAND docker cp -L pytorch_backend_ptlib:/usr/local/${LIB_DIR}/libopencv_calib3d.so libopencv_calib3d.so
264265
COMMAND docker cp -L pytorch_backend_ptlib:/usr/local/${LIB_DIR}/libopencv_features2d.so libopencv_features2d.so
265266
COMMAND docker cp -L pytorch_backend_ptlib:/usr/local/${LIB_DIR}/libopencv_flann.so libopencv_flann.so
266-
COMMAND /bin/sh -c "docker cp pytorch_backend_ptlib:$<IF:$<BOOL:${RHEL_BUILD}>,/usr/lib64/libpng16.so.16.34.0,/usr/local/lib/python3.12/dist-packages/torchvision.libs/libpng16-748299c7.so.16.34.0> libpng16.so"
267-
COMMAND /bin/sh -c "docker cp pytorch_backend_ptlib:$<IF:$<BOOL:${RHEL_BUILD}>,/usr/lib64/libjpeg.so.62.2.0,/usr/local/lib/python3.12/dist-packages/torchvision.libs/libjpeg-da649728.so.62.2.0> libjpeg.so"
267+
COMMAND /bin/sh -c "docker run --name libjpeg-${random_id} -it ${TRITON_PYTORCH_DOCKER_IMAGE} find /usr/lib64 /usr/local/lib/python3.12/dist-packages/torchvision.libs/ -name libjpeg* -exec cp -v {} /tmp/libjpeg.so \\; && docker cp libjpeg-${random_id}:/tmp/libjpeg.so libjpeg.so"
268+
COMMAND /bin/sh -c "docker run --name libpng-${random_id} -it ${TRITON_PYTORCH_DOCKER_IMAGE} find /usr/lib64 /usr/local/lib/python3.12/dist-packages/torchvision.libs/ -name libpng* -exec cp -v {} /tmp/libpng16.so \\; && docker cp libpng-${random_id}:/tmp/libpng16.so libpng16.so"
268269
COMMAND /bin/sh -c "if [ -f libmkl_def.so.1 ]; then patchelf --add-needed libmkl_gnu_thread.so.1 libmkl_def.so.1; fi"
269270
COMMAND /bin/sh -c "if [ -f libmkl_def.so.1 ]; then patchelf --add-needed libmkl_core.so.1 libmkl_def.so.1; fi"
270271
COMMAND /bin/sh -c "if [ -f libmkl_avx2.so.1 ]; then patchelf --add-needed libmkl_gnu_thread.so.1 libmkl_avx2.so.1; fi"

0 commit comments

Comments
 (0)