Updates knem location in install script

This commit is contained in:
2024-12-17 13:46:43 +00:00
parent c049a2ad0b
commit 9056e9023c
2 changed files with 6 additions and 7 deletions

View File

@ -1,13 +1,12 @@
#!/usr/bin/env bash
lrank=$OMPI_COMM_WORLD_LOCAL_RANK
numa1=$(( 2 * lrank))
numa2=$(( 2 * lrank + 1 ))
numa1=$((lrank))
netdev=mlx5_${lrank}:1
export CUDA_VISIBLE_DEVICES=$OMPI_COMM_WORLD_LOCAL_RANK
export UCX_NET_DEVICES=${netdev}
BINDING="--interleave=$numa1,$numa2"
BINDING="--interleave=$numa1"
echo "$(hostname) - $lrank device=$CUDA_VISIBLE_DEVICES binding=$BINDING"