Skip to content

Commit

Permalink
Use shfmt to format A3 series prolog/epilog scripts
Browse files Browse the repository at this point in the history
  • Loading branch information
tpdownes committed Sep 30, 2024
1 parent 6e4500a commit b400646
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions tools/prologs-epilogs/receive-data-path-manager-mega
Original file line number Diff line number Diff line change
Expand Up @@ -36,8 +36,8 @@ RXDM_CONTAINER=receive-datapath-manager-"${SLURM_JOB_ID}"
if [[ ${SLURM_SCRIPT_CONTEXT} == "prolog_slurmd" ]]; then
docker container list --filter "name=receive-datapath-manager-*" --quiet | xargs --no-run-if-empty docker container stop

export PATH=${PATH}:/usr/local/lib/google-cloud-sdk/bin/
gcloud auth configure-docker --quiet us-docker.pkg.dev 2>&1 &>/dev/null
export PATH=${PATH}:/usr/local/lib/google-cloud-sdk/bin/
gcloud auth configure-docker --quiet us-docker.pkg.dev 2>&1 &>/dev/null

# Install the nccl, nccl-net lib into /var/lib/tcpxo/lib64/.
docker run --rm --name nccl-installer \
Expand All @@ -50,12 +50,12 @@ if [[ ${SLURM_SCRIPT_CONTEXT} == "prolog_slurmd" ]]; then
# /var/lib/tcpxo/lib64/nccl-env-profile.sh is written by the nccl-installer container
# above, and assumes interface names of eth{0..8}.
if (grep -q "ID=debian" /etc/os-release && lsb_release -rs | grep -q "12"); then
cat >> /var/lib/tcpxo/lib64/nccl-env-profile.sh <<- EOF
export NCCL_FASTRAK_CTRL_DEV=enp0s12
export NCCL_FASTRAK_IFNAME=enp6s0,enp7s0,enp13s0,enp14s0,enp134s0,enp135s0,enp141s0,enp142s0
export NCCL_SOCKET_IFNAME=enp0s12
export NCCL_FASTRAK_LLCM_DEVICE_DIRECTORY=/dev/aperture_devices
EOF
cat >>/var/lib/tcpxo/lib64/nccl-env-profile.sh <<-EOF
export NCCL_FASTRAK_CTRL_DEV=enp0s12
export NCCL_FASTRAK_IFNAME=enp6s0,enp7s0,enp13s0,enp14s0,enp134s0,enp135s0,enp141s0,enp142s0
export NCCL_SOCKET_IFNAME=enp0s12
export NCCL_FASTRAK_LLCM_DEVICE_DIRECTORY=/dev/aperture_devices
EOF
fi

# Start FasTrak receive-datapath-manager
Expand Down

0 comments on commit b400646

Please sign in to comment.