RUN |5 ENABLE_MITMPROXY=0 VERSION_LIST=CURAND_VERSION CUBLAS_VERSION CUFILE_VERSION LIBFABRIC_INSTALL_PATH=/opt/amazon/efa RDMA_CORE_PREFIX=/usr TARGETARCH=amd64 /bin/sh -c set -exo pipefail if [ "${ENABLE_MITMPROXY}" = "1" ]; then /tmp/manage_cert.sh install; fi mkdir -p /usr/local/nixl cp -a /nixl/usr/local/nixl/. /usr/local/nixl/ ls -lart /usr/local/nixl # Copy etcd library # apt-get update # apt-get install -y libssl-dev libprotobuf-dev libgrpc-dev libgrpc++-dev protobuf-compiler-grpc # rm -rf /var/lib/apt/lists/* cp /nixl/usr/local/lib/libetcd-cpp-api-core.so /usr/local/lib/libetcd-cpp-api-core.so # Copy aws-sdk-cpp headers mkdir -p /usr/local/include/aws cp -a /nixl/usr/local/include/aws/. /usr/local/include/aws/ cp -a /nixl/usr/local/lib/libaws-* /usr/local/lib/ ls -lart /usr/local/include/aws # Copy gusli headers cp -a /nixl/usr/include/gus* /usr/include/ cp -a /nixl/usr/lib/libg* /usr/lib/ # Copy nixlbench mkdir -p /usr/local/nixlbench/bin /usr/local/nixlbench/include /usr/local/nixlbench/lib cp -a /nixl/usr/local/nixlbench/. /usr/local/nixlbench/ ls -lart /usr/local/nixlbench # Install nixlbench runtime dependencies apt update apt install -y libgflags-dev libgrpc-dev libgrpc++-dev libaio1t64 liburing2 rm -rf /var/lib/apt/lists/* # Copy s2n-tls headers and libs cp -a /nixl/usr/local/s2n-tls-install/include/. /usr/local/include/ cp -a /nixl/usr/local/s2n-tls-install/lib/. /usr/local/lib/ ls -lart /usr/local/lib/libs2n* # aws-cli needed by nixlbench with OBJ backend export ARCH=x86_64 if [ "$TARGETARCH" = "arm64" ]; then ARCH=aarch64 fi curl "https://awscli.amazonaws.com/awscli-exe-linux-${ARCH}.zip" -o "awscliv2.zip" unzip awscliv2.zip &&
  ./aws/install &&
  rm -rf awscliv2.zip aws aws --version ldconfig if [ "${ENABLE_MITMPROXY}" = "1" ]; then /tmp/manage_cert.sh uninstall; fi # buildkit

TARGETARCH=amd64

./common.sh ./common.sh

./test_nixlbench.sh ./test_nixlbench.sh

LD_LIBRARY_PATH=/usr/local/nixlbench/lib:/usr/local/lib:/opt/amazon/efa/lib:/usr/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/nixl/lib/x86_64-linux-gnu:/usr/local/nixl/lib/aarch64-linux-gnu

PATH=/usr/local/nixlbench/bin:/usr/local/nixl/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/mpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/amazon/efa/bin

ENABLE_MITMPROXY=0

LD_LIBRARY_PATH=/usr/local/lib:/opt/amazon/efa/lib:/usr/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/nixl/lib/x86_64-linux-gnu:/usr/local/nixl/lib/aarch64-linux-gnu

RDMA_CORE_PREFIX=/usr

LIBFABRIC_INSTALL_PATH=/opt/amazon/efa