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