#! /bin/bash

#SBATCH -J INJECT
#SBATCH -N 1
#SBATCH --ntasks-per-node=36
#SBATCH -p debug
#SBATCH --mem=240G

module load mpi/openmpi/4.0.2/gcc-7.3.1
date

srun hostname -s | sort -n | awk -F"-" '{print $2}' | uniq > pnodes
mpirun -mca pml ucx -x UCX_NET_DEVICES=mlx5_0:1 -machinefile pnodes -np 36 --map-by node python injection_pipeline.py