diff --git a/CMakeLists.txt b/CMakeLists.txt index 6adc487..24ebed5 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -133,12 +133,6 @@ endif() # TRITON_ENABLE_GPU configure_file(src/libtriton_tensorflow.ldscript libtriton_tensorflow.ldscript COPYONLY) if (${TRITON_TENSORFLOW_DOCKER_BUILD}) - if (CMAKE_HOST_SYSTEM_PROCESSOR MATCHES "aarch64") - set(LIBS_ARCH "aarch64") - else() - set(LIBS_ARCH "x86_64") - endif() - add_custom_command( OUTPUT ${TRITON_TENSORFLOW_CC_LIBNAME}.${TRITON_TENSORFLOW_VERSION} @@ -155,7 +149,7 @@ if (${TRITON_TENSORFLOW_DOCKER_BUILD}) COMMAND docker stop tensorflow_backend_deps || echo "error ignored..." || true COMMAND docker rm tensorflow_backend_deps || echo "error ignored..." || true COMMAND if [ "${TRITON_TENSORFLOW_INSTALL_EXTRA_DEPS}" = "ON" ] \; then mkdir tf_backend_deps && docker run -it -d --name tensorflow_backend_deps ${TRITON_TENSORFLOW_DOCKER_IMAGE} \; fi \; - COMMAND if [ "${TRITON_TENSORFLOW_INSTALL_EXTRA_DEPS}" = "ON" ] \; then docker exec tensorflow_backend_deps sh -c "tar -cf - $,/usr/local/cuda/targets/x86_64-linux/lib/,/usr/lib/${LIBS_ARCH}-linux-gnu/>libnccl.so*" | tar --strip-components=3 -xf - -C ./tf_backend_deps \; fi + COMMAND if [ "${TRITON_TENSORFLOW_INSTALL_EXTRA_DEPS}" = "ON" ] \; then docker exec tensorflow_backend_deps sh -c "find $,/usr/local/cuda/targets/*-linux/lib/,/usr/lib/*-linux-gnu/>libnccl.so*" | xargs -I {} docker cp tensorflow_backend_deps:{} ./tf_backend_deps \; fi COMMAND if [ "${TRITON_TENSORFLOW_INSTALL_EXTRA_DEPS}" = "ON" ] \; then docker stop tensorflow_backend_deps && docker rm tensorflow_backend_deps \; fi \; COMMENT "Extracting ${TRITON_TENSORFLOW_CC_LIBNAME}.${TRITON_TENSORFLOW_VERSION} and ${TRITON_TENSORFLOW_FW_LIBNAME}.${TRITON_TENSORFLOW_VERSION} from ${TRITON_TENSORFLOW_DOCKER_IMAGE}"