Skip to content

Commit 3a7d08d

Browse files
committed
Update pytorch libraryr destination
1 parent 9b9d34e commit 3a7d08d

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
@@ -263,8 +263,8 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
263263
COMMAND docker cp -L pytorch_backend_ptlib:/usr/local/${LIB_DIR}/libopencv_calib3d.so libopencv_calib3d.so
264264
COMMAND docker cp -L pytorch_backend_ptlib:/usr/local/${LIB_DIR}/libopencv_features2d.so libopencv_features2d.so
265265
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/lib/${LIBS_ARCH}-linux-gnu/libpng16.so.16.43.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/lib/${LIBS_ARCH}-linux-gnu/libjpeg.so.8.2.2> libjpeg.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"
268268
COMMAND /bin/sh -c "if [ -f libmkl_def.so.1 ]; then patchelf --add-needed libmkl_gnu_thread.so.1 libmkl_def.so.1; fi"
269269
COMMAND /bin/sh -c "if [ -f libmkl_def.so.1 ]; then patchelf --add-needed libmkl_core.so.1 libmkl_def.so.1; fi"
270270
COMMAND /bin/sh -c "if [ -f libmkl_avx2.so.1 ]; then patchelf --add-needed libmkl_gnu_thread.so.1 libmkl_avx2.so.1; fi"
@@ -493,6 +493,7 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
493493
COMMAND ln -sf libopencv_flann.so libopencv_flann.so.${OPENCV_VERSION}
494494
COMMAND ln -sf libpng16.so libpng16.so.16
495495
COMMAND ln -sf libjpeg.so libjpeg.so.8
496+
COMMAND ln -sf libjpeg.so libjpeg.so.62
496497
COMMAND ln -sf libcusparseLt.so libcusparseLt.so.0
497498
RESULT_VARIABLE LINK_STATUS
498499
WORKING_DIRECTORY ${CMAKE_INSTALL_PREFIX}/backends/pytorch)

0 commit comments

Comments
 (0)