@@ -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