diff --git a/frontier/containers_on_frontier_docs/gpu_aware_mpi_example/submit.sbatch b/frontier/containers_on_frontier_docs/gpu_aware_mpi_example/submit.sbatch index 930de7a..535acef 100644 --- a/frontier/containers_on_frontier_docs/gpu_aware_mpi_example/submit.sbatch +++ b/frontier/containers_on_frontier_docs/gpu_aware_mpi_example/submit.sbatch @@ -11,6 +11,7 @@ module load rocm/5.7.1 # The cray-mpich-abi module contains the libmpi.so that the MPI application in # the container (that wasn't compiled with a Cray compiler) will look for. +module load craype-network-ofi module load cray-mpich-abi # This is set so to turn on GPU aware MPI