mirror of
https://github.com/paboyle/Grid.git
synced 2025-06-18 07:47:06 +01:00
Adding build basics for different systems
This commit is contained in:
17
systems/Tursa/mpiwrapper.sh
Executable file
17
systems/Tursa/mpiwrapper.sh
Executable file
@ -0,0 +1,17 @@
|
||||
#!/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} $*
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user