14 lines
315 B
Bash
Executable File
14 lines
315 B
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
lrank=$OMPI_COMM_WORLD_LOCAL_RANK
|
|
numa1=$((lrank))
|
|
netdev=mlx5_${lrank}:1
|
|
|
|
export CUDA_VISIBLE_DEVICES=$OMPI_COMM_WORLD_LOCAL_RANK
|
|
export UCX_NET_DEVICES=${netdev}
|
|
BINDING="--interleave=$numa1"
|
|
|
|
echo "$(hostname) - $lrank device=$CUDA_VISIBLE_DEVICES binding=$BINDING"
|
|
|
|
numactl ${BINDING} "$@"
|