#!/bin/bash
let "RANK_MOD_4=$OMPI_COMM_WORLD_LOCAL_RANK % 4"
export CUDA_VISIBLE_DEVICES=$RANK_MOD_4
#echo $CUDA_VISIBLE_DEVICES
$@
