[None][feat] Add libfabric to dockerfile

Signed-off-by: Yoray Zack <[email protected]>

Author: zackyoray

docker/Dockerfile.multi

@@ -41,6 +41,7 @@ COPY docker/common/install.sh \
      docker/common/install_mpi4py.sh \
      docker/common/install_pytorch.sh \
      docker/common/install_ucx.sh \
+     docker/common/install_libfabric.sh \
      docker/common/install_nixl.sh \
      docker/common/install_etcd.sh \
      docker/common/install_mooncake.sh \
@@ -80,6 +81,9 @@ RUN bash ./install.sh --opencv && rm install.sh
 # Install UCX first
 RUN GITHUB_MIRROR=${GITHUB_MIRROR} bash ./install_ucx.sh && rm install_ucx.sh
 
+# Install libfabric for NIXL LIBFABRIC plugin support
+RUN GITHUB_MIRROR=${GITHUB_MIRROR} bash ./install_libfabric.sh && rm install_libfabric.sh
+
 # Install NIXL
 RUN GITHUB_MIRROR=${GITHUB_MIRROR} bash ./install_nixl.sh && rm install_nixl.sh
 

docker/common/install_libfabric.sh

@@ -0,0 +1,27 @@
+#!/bin/bash
+set -ex
+
+LIBFABRIC_VERSION="v1.21.0"
+LIBFABRIC_INSTALL_PATH="/opt/libfabric"
+LIBFABRIC_REPO="https://github.com/ofiwg/libfabric.git"
+
+# Install dependencies
+apt-get update
+apt-get install -y rdma-core libibverbs-dev ibverbs-providers librdmacm-dev
+
+# Clone and build libfabric
+rm -rf ${LIBFABRIC_INSTALL_PATH}
+git clone --depth 1 -b ${LIBFABRIC_VERSION} ${LIBFABRIC_REPO}
+cd libfabric
+./autogen.sh
+./configure \
+  --prefix=${LIBFABRIC_INSTALL_PATH} \
+  --enable-verbs \
+  --enable-tcp \
+  --enable-rxm
+make install -j"$(nproc)"
+cd ..
+rm -rf libfabric  # Remove libfabric source to save space
+
+# Add to LD_LIBRARY_PATH
+echo "export LD_LIBRARY_PATH=${LIBFABRIC_INSTALL_PATH}/lib:\$LD_LIBRARY_PATH" >> "${ENV}"