Files
lammps-gran-kokkos/examples/tad

Run this example as:

mpirun -np 3 lmp_g++ -partition 3x1 -in in.tad

You should be able to use any number of replicas >= 3.