Skip to content

Commit 7974706

Browse files
committed
cmake: Remogin old lig references.
1 parent 055f7c7 commit 7974706

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

CMakeLists.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff 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})

0 commit comments

Comments
 (0)