run_mps.sh

#!/bin/bash
export CUDA_VISIBLE_DEVICES=0
lrank=$OMPI_COMM_WORLD_LOCAL_RANK
case ${lrank} in
[0])
 export CUDA_MPS_PIPE_DIRECTORY=/home/aymen/mps_0; ./vector_add
 ;;
[1])
 export CUDA_MPS_PIPE_DIRECTORY=/home/aymen/mps_1; ./vector_add
 ;;
[2])
 export CUDA_MPS_PIPE_DIRECTORY=/home/aymen/mps_0; ./vector_add
 ;;
[3])
 export CUDA_MPS_PIPE_DIRECTORY=/home/aymen/mps_1; ./vector_add
 ;;
esac