Co-array Fortran on Niagara

Versions 12 and higher of the Intel Fortran compiler, and version 5.1 and up of the GNU Fortran compiler, support almost all of Co-array Fortran, and are installed on Niagara.

This page will briefly sketch how to compile and run Co-array Fortran programs using these compilers.

Example

Here is an example of a co-array fortran program:

program Hello_World
  integer :: i ! Local variable
  integer :: num[*] ! scalar coarray
  if (this_image() == 1) then
    write(*,'(a)') 'Enter a number: '
    read(*,'(i80)') num
    ! Distribute information to other images
    do i = 2, num_images()
      num[i] = num
    end do
  end if
  sync all ! Barrier to make sure the data has arrived
  ! I/O from all nodes
  write(*,'(a,i0,a,i0)') 'Hello ',num,' from image ', this_image()
end program Hello_world

(Adapted from [1]).

Compiling, linking and running co-array fortran programs is different depending on whether you will run the program only on a single node (with 8 cores), or on several nodes, and depends on which compiler you are using, Intel, or GNU.

Intel compiler instructions for Coarray Fortran

Loading necessary modules

First, you need to load the module for version 12 or greater of the Intel compilers, as well as Intel MPI.

module load intel/2018.2 intelmpi/2018.2

There are two modes in which the intel compiler supports coarray fortran:

1. Single node usage

2. Multiple node usage

The way you compile and run for these two cases is different. However, we're working on making coarray fortran compilation and running more uniform among these two cases, as well as with the, as-yet-experimental, gfortran coarray support. See Uniformized Usage below.

Note: For multiple node usage, it makes sense to have to load the IntelMPI module, since Intel's implementation of Co-array Fortran uses MPI. However, the Intel MPI module is needed even for single-node usage, just in order to link successfully.

Single node usage

Compilation

ifort -O3 -xHost -coarray=shared -c [sourcefile] -o [objectfile]

Linking

ifort -coarray=shared [objectfile] -o [executable]

Running

To run this co-array program on one node with 16 images (co-array version for what openmp calls a thread and mpi calls a process), you simply put

./[executable]

in your job submission script. The reason that this gives 16 images is that HyperThreading is enabled on Niagara nodes, which makes it seem to the system as if there are 80 computing units on a node, even though physically there are only 40.

To control the number of images, you can change the FOR_COARRAY_NUM_IMAGES environment variable:

export FOR_COARRAY_NUM_IMAGES=2
./[executable]

This can be useful for testing.

An example submission script would look as follows:

#!/bin/bash
# SLURM submission script for SciNet Niagara (Intel Coarray Fortran)
#
#SBATCH --nodes=1
#SBATCH --time=1:00:00
#SBATCH --cpus-per-task=40
#SBATCH --job-name test

# DIRECTORY TO RUN - $SLURM_SUBMIT_DIR is directory job was submitted from
cd $SLURM_SUBMIT_DIR

# LOAD MODULES THAT THE APPLICATION WAS COMPILED WITH
module load intel/2018.2 intelmpi/2018.2

# RUN THE APPLICATION WITH 80 IMAGES
export FOR_COARRAY_NUM_IMAGES=80
./[executable]

Multiple nodes usage

Compilation

ifort -O3 -xHost -coarray=distributed -c [sourcefile] -o [objectfile]

Linking

ifort -coarray=distributed [objectfile] -o [executable]

Running

Because distributed co-array fortran is based on MPI, we need to launch the mpi processes on different nodes. The easiest way to do this is to set the number of images using FOR_COARRAY_NUM_IMAGES and then to use mpirun without an -np parameter:

export I_MPI_PROCESS_MANAGER=mpd
export FOR_COARRAY_NUM_IMAGES=160
mpirun ./[executable]

Note that the total number of images is set explicitly, and should not be given to mpirun. You can still pass other parameters to mpirun, though. Also the default process manager for INTELMPI is not MPD, however Coarray needs MPD which is why I_MPI_PROCESS_MANAGER needs to be set to mpd.

An example submission script would look as follows:

#!/bin/bash
#
#SBATCH --nodes=4
#SBATCH --time=1:00:00
#SBATCH --ntasks=160
#SBATCH --job-name test

# DIRECTORY TO RUN - $SLURM_SUBMIT_DIR is directory job was submitted from
cd $SLURM_SUBMIT_DIR

# LOAD MODULES THAT THE APPLICATION WAS COMPILED WITH
module load intel/2018.2 intelmpi/2018.2

# EXECUTION export COMMAND; FOR_COARRAY_NUM_IMAGES = nodes*ppn
export I_MPI_PROCESS_MANAGER=mpd
export FOR_COARRAY_NUM_IMAGES=160
mpirun ./[executable]

Uniformized Usage

If you load the addition module

module load caf/intel/any

you get access to a compilation and linking wrapper called caf and a wrapper for running the application called cafrun.

Compilation

caf -O3 -xhost -c [sourcefile] -o [objectfile]

Linking

caf [objectfile] -o [executable]

Running

To run this co-array program on one node with 40 images, you simply put

cafrun ./[executable]

This runs 40 images, not 80.

To control the number of images, you can change the run command to

cafrun -np 2 ./[executable]

This can be useful for testing.

To control the number of images per node, add the -N [images-per-node] option.

Note: currently, the uniformized mode doesn't explicitly utilize optimization opportunities offered by the single node mode, although it will work on one node.