File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed
Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change @@ -302,7 +302,6 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
302302 COMMAND /bin/sh -c "if [ -f libmkl_vml_def.so.1 ]; then patchelf --add-needed libmkl_intel_thread.so.1 libmkl_vml_def.so.1; fi"
303303 COMMAND /bin/sh -c "if [ -f libmkl_vml_def.so.1 ]; then patchelf --add-needed libmkl_core.so.1 libmkl_vml_def.so.1; fi"
304304 COMMAND /bin/sh -c "if [ -f libmkl_intel_thread.so.1 ]; then patchelf --add-needed libmkl_intel_lp64.so.1 libmkl_intel_thread.so.1; fi"
305- COMMAND /bin/sh -c "if [ ${TRITON_PYTORCH_ENABLE_TORCHVISION} = 'ON' ]; then if [ ${RHEL_BUILD} = 'OFF' ]; then ln -s libtorchvision.so.1 libtorchvision.so; fi; fi;"
306305 COMMAND docker rm pytorch_backend_ptlib
307306 COMMENT "Extracting pytorch and torchvision libraries and includes from ${TRITON_PYTORCH_DOCKER_IMAGE} "
308307 VERBATIM
@@ -410,7 +409,7 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
410409 if (${TRITON_PYTORCH_ENABLE_TORCHVISION} )
411410 set (TRITON_PYTORCH_LIBS
412411 ${TRITON_PYTORCH_LIBS}
413- "${CMAKE_CURRENT_BINARY_DIR} /$<IF:$<BOOL: ${RHEL_BUILD} >, libtorchvision.so,libtorchvision.so.1> " )
412+ "${CMAKE_CURRENT_BINARY_DIR} /libtorchvision.so.1 " )
414413 endif () # TRITON_PYTORCH_ENABLE_TORCHVISION
415414
416415 if (${TRITON_PYTORCH_ENABLE_TORCHTRT} )
You can’t perform that action at this time.
0 commit comments