15 lines
360 B
Bash
15 lines
360 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=mlx5_${lrank}:1
|
||
|
BINDING="--interleave=$numa1,$numa2"
|
||
|
|
||
|
echo "`hostname` - $lrank device=$CUDA_VISIBLE_DEVICES binding=$BINDING"
|
||
|
|
||
|
numactl ${BINDING} $*
|