#!/bin/bash
#SBATCH --job-name=pingpong_mpi4py
#SBATCH --partition=unite
#SBATCH --nodes=2
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=1
#SBATCH --time=00:01:00
#SBATCH --output=pingpong_%j.out

module purge
#Load necessary modules
module add unite/python/3.14/mpi4py

#Start the example
mpirun -np $SLURM_NTASKS python3 ./pingpong_mpi4py.py