#!/bin/sh

if [ $SLURM_LOCALID = "0" ]
then
    NUMA=1
    MLX=mlx5_0:1
    GPU=1
fi
if [ $SLURM_LOCALID = "1" ]
then
    NUMA=3
    MLX=mlx5_1:1
    GPU=0
fi
if [ $SLURM_LOCALID = "2" ]
then
    NUMA=5
    MLX=mlx5_2:1
    GPU=3
fi
if [ $SLURM_LOCALID = "3" ]
then
    NUMA=7
    MLX=mlx5_3:1
    GPU=2
fi

CMD="env CUDA_VISIBLE_DEVICES=$GPU UCX_NET_DEVICES=$MLX numactl -m $NUMA ./Benchmark_ITT --mpi 1.1.1.4"
echo $CMD
$CMD