Skip to content

Commit

Permalink
Add support for Amazon EFA
Browse files Browse the repository at this point in the history
Issue addressed #167
    
    The upstream JAX container contains only MOFED NIC support. The MOFED package
    from Mellanox that we use installs *ibverbs* libraries which do not contain
    libefa*.so which are required for AWS.
    
    A temporary solution is to provide a script as a part of base container
    (/usr/local/bin/install-efa.sh) that AWS user can run inside the container
    to handle this issue.
    The script does the following:
    1. Remove all *ibverbs* and RDMA related libraries
    2. Download Amazon EFA installer
    3. Install EFA
  • Loading branch information
DwarKapex authored Aug 31, 2023
1 parent 524f5d0 commit d0def62
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion .github/container/install-efa.sh
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ pushd ${WORKDIR}
AMAZON_EFA_LINK="https://efa-installer.amazonaws.com/aws-efa-installer-${EFA_INSTALLER_VERSION}.tar.gz"
curl -O "$AMAZON_EFA_LINK"
tar -xf aws-efa-installer-${EFA_INSTALLER_VERSION}.tar.gz && cd aws-efa-installer
./efa_installer.sh -y --skip-kmod
./efa_installer.sh -y -g -d --skip-kmod --skip-limit-conf --no-verify

popd

Expand Down

0 comments on commit d0def62

Please sign in to comment.