#!/bin/bash
set -e

nx=560
ny=280
nz=280
rt=60

# 4 nodes, 32 GPUs
./mpirun_rochpcg --nx ${nx} --ny ${ny} --nz ${nz} --rt ${rt} \
    --npx 4 --npy 4 --npz 2 \
    -H node01,node02,node03,node04 \
    --tcp-iface p14p2 \
    --ssh-port 3333

# 2 nodes, 16 GPUs
./mpirun_rochpcg --nx ${nx} --ny ${ny} --nz ${nz} --rt ${rt} \
    --npx 4 --npy 2 --npz 2 \
    -H node01,node02 \
    --tcp-iface p14p2 \
    --ssh-port 3333

# 1 nodes, 8 GPUs
./mpirun_rochpcg --nx ${nx} --ny ${ny} --nz ${nz} --rt ${rt} \
    --npx 2 --npy 2 --npz 2

# 1 nodes, 4 GPUs
./mpirun_rochpcg --nx ${nx} --ny ${ny} --nz ${nz} --rt ${rt} \
    --npx 2 --npy 2 --npz 2

# 1 nodes, 2 GPUs
./mpirun_rochpcg --nx ${nx} --ny ${ny} --nz ${nz} --rt ${rt} \
    --npx 2 --npy 1 --npz 1

# 1 nodes, 1 GPU
./mpirun_rochpcg --nx ${nx} --ny ${ny} --nz ${nz} --rt ${rt} \
    --npx 1 --npy 1 --npz 1