mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-14 01:35:36 +00:00
18 lines
355 B
Bash
Executable File
18 lines
355 B
Bash
Executable File
#!/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} $*
|
|
|
|
|
|
|