#!/bin/bash 
#SBATCH --job-name=mpi_parallel
#SBATCH --output=parallel_res_%j.out
#SBATCH --error=parallel_err_%j.err
#SBATCH --partition=batch
#SBATCH --nodes=2               
#SBATCH --ntasks-per-node=4 
#SBATCH --time=00:10:00
#SBATCH --mem=2G

module purge
module load gcc/15.2.0
module load openmpi/5.0.8
source /home/username/mpivenv/bin/activate     
srun --mpi=pmix python3 -u parallel_test.py