Only for mvapich2-gdr:

#!/bin/bash
# Usage: mpirun -n <num_proc> MV2_USE_AFFINITY=0 MV2_ENABLE_AFFINITY=0 ./launch 
./bin

export CUDA_VISIBLE_DEVICES=$MV2_COMM_WORLD_LOCAL_RANK
case $MV2_COMM_WORLD_LOCAL_RANK in
        [0]) cpus=0-3 ;;
        [1]) cpus=64-67 ;;
        [2]) cpus=72-75 ;;
esac

numactl --physcpubind=$cpus $@

Reply via email to