mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-15 02:05:37 +00:00
18 lines
355 B
Bash
18 lines
355 B
Bash
|
#!/bin/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} $*
|
||
|
|
|
||
|
|
|
||
|
|
|