File tree Expand file tree Collapse file tree 1 file changed +0
-2
lines changed
Expand file tree Collapse file tree 1 file changed +0
-2
lines changed Original file line number Diff line number Diff line change @@ -192,7 +192,6 @@ set(PT_LIBS
192192if (${TRITON_PYTORCH_NVSHMEM} )
193193 set (PT_LIBS
194194 ${PT_LIBS}
195- "libtorch_nvshmem.so"
196195 )
197196endif () # TRITON_PYTORCH_NVSHMEM
198197
@@ -280,7 +279,6 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
280279 COMMAND docker cp pytorch_backend_ptlib:${PY_INSTALL_PATH} /torch/lib/libbackend_with_compiler.so libbackend_with_compiler.so
281280 COMMAND docker cp pytorch_backend_ptlib:${PY_INSTALL_PATH} /torch/lib/libaoti_custom_ops.so libaoti_custom_ops.so
282281 COMMAND docker cp pytorch_backend_ptlib:${PY_INSTALL_PATH} /torch/lib/libtorch_python.so libtorch_python.so
283- COMMAND /bin/sh -c "if [ ${TRITON_PYTORCH_NVSHMEM} = 'ON' ]; then docker cp pytorch_backend_ptlib:${PY_INSTALL_PATH} /torch/lib/libtorch_nvshmem.so libtorch_nvshmem.so; fi"
284282 COMMAND /bin/sh -c "if [ ${TRITON_PYTORCH_ENABLE_TORCHVISION} = 'ON' ]; then if [ ${RHEL_BUILD} = 'ON' ]; then docker cp -a -L pytorch_backend_ptlib:/usr/local/lib64/libtorchvision.so libtorchvision.so; else docker cp -a -L pytorch_backend_ptlib:/usr/local/${LIB_DIR} /libtorchvision.so.1 libtorchvision.so.1; fi; fi"
285283 COMMAND /bin/sh -c "if [ ${TRITON_PYTORCH_ENABLE_TORCHVISION} = 'ON' ]; then docker cp pytorch_backend_ptlib:/opt/pytorch/vision/torchvision/csrc include/torchvision/torchvision; fi"
286284 COMMAND /bin/sh -c "if [ ${TRITON_PYTORCH_ENABLE_TORCHTRT} = 'ON' ]; then docker cp pytorch_backend_ptlib:${PY_INSTALL_PATH} /torch_tensorrt/lib/libtorchtrt.so libtorchtrt.so; fi"
You can’t perform that action at this time.
0 commit comments