LFRic API

This section describes the LFRic application programming interface (API). This API explains what a user needs to write in order to make use of the LFRic API in PSyclone.

As with the majority of PSyclone APIs, the LFRic API specifies how a user needs to write the algorithm layer and the kernel layer to allow PSyclone to generate the PSy layer. These algorithm and kernel APIs are discussed separately in the following sections.

The LFRic API supports the Met Office’s finite element (hereafter FEM) based GungHo dynamical core (see Introduction). This dynamical core with atmospheric physics parameterisation schemes is a part of the Met Office LFRic modelling system [AFH+19], currently being developed in preparation for exascale computing in the 2020s. The LFRic repository and the associated wiki are hosted at the Met Office Science Repository Service. The code is BSD-licensed, however browsing the LFRic wiki and code repository requires login access to MOSRS. For more technical details on the implementation of LFRic, please see the LFRic documentation.

Algorithm

The general requirements for the structure of an Algorithm are explained in the Algorithm layer section. This section explains the LFRic-API-specific specialisations and extensions.

The LFRic API defines a set of objects, with specific meanings and data-structures, that can be provided as arguments to Kernels within invoke calls. These are: scalar, field, field vector, operator, column-wise operator, Quadrature, Halo Depth and Stencil Extents. The example below showcases the use of each of these arguments:

real(kind=r_def)           :: rscalar
integer(kind=i_def)        :: iscalar, halo_depth
logical(kind=l_def)        :: lscalar
integer(kind=i_def)        :: stencil_extent
type(field_type)           :: field1, field2, field3
type(field_type)           :: field5(3), field6(3)
type(integer_field_type)   :: field7
type(quadrature_type)      :: qr
type(operator_type)        :: operator1
type(columnwise_operator_type) :: cma_op1
...
call invoke( kernel1(field1, field2, operator1, qr),           &
             builtin1(rscalar, field2, field3),                &
             int_builtin2(iscalar, field7),                    &
             kernel2(field1, stencil_extent, field3, lscalar), &
             kernel3(field1, halo_depth),                      &
             assembly_kernel(cma_op1, operator1),              &
             name="some_calculation")

Each of these argument types is described in more detail in the next section.

The LFRic API has support for inter-grid kernels (those that map fields between grids of different resolution). At the Algorithm layer, an invoke of such kernels looks much like an invoke containing general-purpose kernels. The only restrictions to be aware of are that inter-grid kernels accept only field or field-vectors as arguments and that an invoke may not mix inter-grid kernels with any other kernel type.

Algorithm Argument Types

Scalar

In the LFRic API a scalar is a single-valued argument that is identified with GH_SCALAR metadata. Scalar arguments can have real, integer or logical data type in user-defined Kernels (logical data type is not supported in the LFRic Built-ins).

Field

LFRic API fields, identified with GH_FIELD metadata, represent FEM discretisations of various dynamical core prognostic and diagnostic variables. In FEM, variables are discretised by placing them into a function space (see Supported Function Spaces) from which they inherit a polynomial expansion via the basis functions of that space. Field values at points within a cell are evaluated as the sum of a set of basis functions multiplied by coefficients which are the data points. Points of evaluation are determined by a quadrature object (Quadrature) and are independent of the function space the field is on. Placement of field data points, also called degrees of freedom (hereafter “DoFs”), is determined by the function space the field is on. LFRic fields passed as arguments to any LFRic kernel can be of real or integer primitive type. In the LFRic infrastructure, these fields are represented by instances of the field_type and integer_field_type classes, respectively.

Field Vector

Depending on the function space a field lives on, the field data value at a point can be a scalar or a vector (see Supported Function Spaces for the list of scalar and vector function spaces). There is an additional option, called a field vector, to represent a bundle of either scalar- or vector-valued fields. Field vectors are represented as GH_FIELD*N where N is the size of the vector. The 3D coordinate field, for example, has (x, y, z) scalar values at the nodes and therefore has a vector size of 3.

Operator

Represents a matrix constructed on a per-cell basis using Local Matrix Assembly (LMA) and is identified with GH_OPERATOR metadata. In the LFRic infrastructure, operators are represented by instances of the operator_type class. LFRic operators can only have real-valued data in user-defined Kernels (LFRic Built-ins do not currently support operators).

Column-wise Operator

The LFRic API has support for the construction and use of column-wise/Column Matrix Assembly (CMA) operators whose metadata identifier is GH_COLUMNWISE_OPERATOR. In the LFRic infrastructure, column-wise operators are represented by instances of the columnwise_operator_type class. As for the LMA operators above, LFRic column-wise operators can only have real-valued data.

As the name suggests, these are operators constructed for a whole column of the mesh. These are themselves constructed from the Local Matrix Assembly (LMA) operators of each cell in the column. The rules governing Kernels that have CMA operators as arguments are given in the Kernel section below.

There are three recognised Kernel types involving CMA operations; construction, application (including inverse application) and matrix-matrix. The following example sketches-out what the use of such kernels might look like in the Algorithm layer:

use field_mod, only: field_type
use operator_mod, only : operator_type
use columnwise_operator_mod, only : columnwise_operator_type
type(field_type) :: field1, field2, field3
type(operator_type) :: lma_op1, lma_op2
type(columnwise_operator_type) :: cma_op1, cma_op2, cma_op3
real(kind=r_def) :: alpha
...
call invoke(                                                    &
        assembly_kernel(cma_op1, lma_op1, lma_op2),             &
        assembly_kernel2(cma_op2, lma_op1, lma_op2, field3),    &
        apply_kernel(field1, field2, cma_op1),                  &
        matrix_matrix_kernel(cma_op3, cma_op1, alpha, cma_op2), &
        apply_kernel(field3, field1, cma_op3),                  &
        name="cma_example")

The above invoke uses two LMA operators to construct the CMA operator cma_op1. A second CMA operator, cma_op2, is assembled from the same two LMA operators but also uses a field. The first of these CMA operators is then applied to field2 and the result stored in field1 (assuming that the metadata for apply_kernel specifies that it is the first field argument that is written to). The two CMA operators are then combined to produce a third, cma_op3. This is then applied to field1 and the result stored in field3.

Note that PSyclone identifies the type of kernels performing column-wise operations based on their arguments as described in metadata (see Rules for Kernels that work with CMA Operators below). The names of the kernels in the above example are purely illustrative and are not used by PSyclone when determining kernel type.

A full example of CMA operator construction is available in examples/lfric/eg7.

Quadrature

Kernels conforming to the LFRic API may require quadrature information (specified using e.g. gh_shape = gh_quadrature_XYoZ in the kernel metadata - see Section gh_shape and gh_evaluator_targets). This information must be passed to the kernel from the Algorithm layer in the form of one or more quadrature_type objects. These must be the last arguments passed to the kernel (with the exception of halo_depth - Halo Depth - if the kernel requires it) and must be provided in the same order that they are specified in the kernel metadata, e.g. if the metadata for kernel pressure_gradient_kernel_type specified gh_shape = gh_quadrature_XYoZ and that for kernel geopotential_gradient_kernel had gh_shape(2) = (\ gh_quadrature_XYoZ, gh_quadrature_face \) then the corresponding invoke would look something like:

...
qr_xyoz = quadrature_xyoz_type(nqp_exact, rule)
qr_face = quadrature_face_type(nqp_exact, ..., rule)
call invoke(pressure_gradient_kernel_type(rhs_tmp(igh_u), rho, theta, qr_xyoz), &
            geopotential_gradient_kernel_type(rhs_tmp(igh_u), geopotential, &
                                              qr_xyoz, qr_face))

These quadrature objects specify the set(s) of points at which the basis/differential-basis functions required by the kernel are to be evaluated.

Halo Depth

If a Kernel is written to iterate into the halo (has an OPERATES_ON of HALO_CELL_COLUMN or OWNED_AND_HALO_CELL_COLUMN) then the halo depth must be passed as a final, integer argument to the Kernel.

Stencil Extent

The metadata for a Kernel which operates on a cell-column may specify that a Kernel performs a stencil operation on a field. Any such metadata must provide a stencil type. See the meta_args section for more details. The supported stencil types are X1D, Y1D, XORY1D, CROSS, CROSS2D or REGION.

If a stencil operation is specified by the Kernel metadata, the Algorithm layer must provide the extent of the stencil (the maximum distance from the central cell that the stencil extends). The LFRic API expects this information to be added as an additional integer argument immediately after the relevant field when specifying the Kernel via an invoke.

For example:

integer(kind=i_def) :: extent = 2
call invoke(kernel(field1, field2, extent))

where field2 has kernel metadata specifying that it has a stencil access.

extent may also be passed as a literal. For example:

call invoke(kernel(field1, field2, 2))

where, again, field2 has kernel metadata specifying that it has a stencil access.

Note

The stencil extent specified in the Algorithm layer is not the same as the stencil size passed in to the Kernel. The latter contains the number of cells in the stencil which is dependent on both the stencil type and extent.

If the Kernel metadata specifies that the stencil is of type XORY1D (which means X1D or Y1D) then the algorithm layer must specify whether the stencil is X1D or Y1D for that particular kernel call. The LFRic API expects this information to be added as an additional argument immediately after the relevant stencil extent argument. The argument should be an integer with valid values being x_direction or y_direction, both being supplied by the LFRic infrastructure via the flux_direction_mod fortran module

For example:

use flux_direction_mod, only : x_direction
integer(kind=i_def) :: direction = x_direction
integer(kind=i_def) :: extent = 2
! ...
call invoke(kernel(field1, field2, extent, direction))

direction may also be passed as a literal. For example:

use flux_direction_mod, only : x_direction
integer(kind=i_def) :: extent = 2
! ...
call invoke(kernel(field1, field2, extent, x_direction))

If the stencil is of type CROSS2D then the arrays passed to the kernel are of different dimensions to those of other stencils. The CROSS2D stencil is designed for use when it is necessary for a kernel to know where the stencil cells are, relative to the current cell. For this reason, the stencil_size passed to the kernel is an array of length 4 containing sizes for each branch of the stencil. The stencil_size array is always ordered: West, South, East, North. This branch dimension is also part of the stencil_dofmap array making it possible to loop over each branch of the stencil individually. The invoke call for the CROSS2D stencil remains of the same form as for other stencils.

If certain fields use the same value of extent and/or direction then the same variable, or literal value can be provided.

For example:

call invoke(kernel1(field1, field2, extent,  field3, extent, direction), &
            kernel2(field1, field2, extent2, field4, extent, direction))

In the above example field2 and field3 in kernel1 and field4 in kernel2 will have the same extent value but field2 in kernel2 may have a different value. Similarly, field3 in kernel1 and field4 in kernel2 will have the same direction value.

An example of the use of stencils is available in examples/lfric/eg5.

There is currently no attempt to perform type checking in PSyclone so any errors in the type and/or position of arguments will not be picked up until compile time. However, PSyclone does check for the correct number of algorithm arguments. If the wrong number of arguments is provided then an exception is raised.

For example, running test 19.2 from the LFRic API test suite gives:

cd <PSYCLONEHOME>/src/psyclone/tests
psyclone test_files/dynamo0p3/19.2_single_stencil_broken.f90
"Generation Error: error: expected '5' arguments in the algorithm layer but found '4'.
Expected '4' standard arguments, '1' stencil arguments and '0' qr_arguments'"

Mixed Precision

The LFRic API supports the ability to specify the precision required by the model via precision variables. To make use of this, the code developer must declare scalars, fields and operators in the algorithm layer with the required LFRic-supported precision. In the current implementation there are two supported precisions for REAL data and one each for INTEGER and LOGICAL data. The actual precision used in the code can be set in a configuration file. For example, INTEGER data could be set to be 32-bit precision. As REAL data has more than one supported precision, different parts of the code can be configured to have different precision.

The table below gives the currently supported datatypes, their associated kernel metadata description and their precision:

Data Type

Kernel Metadata

Precision

REAL(R_DEF)

GH_SCALAR, GH_REAL

R_DEF

REAL(R_BL)

GH_SCALAR, GH_REAL

R_BL

REAL(R_PHYS)

GH_SCALAR, GH_REAL

R_PHYS

REAL(R_SOLVER)

GH_SCALAR, GH_REAL

R_SOLVER

REAL(R_TRAN)

GH_SCALAR, GH_REAL

R_TRAN

INTEGER(I_DEF)

GH_SCALAR, GH_INTEGER

I_DEF

LOGICAL(L_DEF)

GH_SCALAR, GH_LOGICAL

L_DEF

FIELD_TYPE

GH_FIELD, GH_REAL

R_DEF

R_BL_FIELD_TYPE

GH_FIELD, GH_REAL

R_BL

R_PHYS_FIELD_TYPE

GH_FIELD, GH_REAL

R_PHYS

R_SOLVER_FIELD_TYPE

GH_FIELD, GH_REAL

R_SOLVER

R_TRAN_FIELD_TYPE

GH_FIELD, GH_REAL

R_TRAN

INTEGER_FIELD_TYPE

GH_FIELD, GH_INTEGER

I_DEF

OPERATOR_TYPE

GH_OPERATOR, GH_REAL

R_DEF

R_SOLVER_OPERATOR_TYPE

GH_OPERATOR, GH_REAL

R_SOLVER

R_TRAN_OPERATOR_TYPE

GH_OPERATOR, GH_REAL

R_TRAN

COLUMNWISE_OPERATOR_TYPE

GH_COLUMNWISE_OPERATOR, GH_REAL

R_SOLVER

As can be seen from the above table, the kernel metadata does not capture all of the precision options. For example, from the metadata it is not possible to determine whether a REAL scalar, REAL field or REAL operator has precision R_DEF, R_SOLVER or R_TRAN.

If a scalar, field, or operator is specified with a particular precision in the algorithm layer then any associated kernels that it is passed to must have been written so that they support this precision. If a kernel needs to support data that can be stored with different precisions then appropriate precision-specific subroutines should be written. These precision-specific subroutine should be called via a generic interface (which lets Fortran choose the appropriate subroutine based on the precision of its argument(s)).

Below is a simple example of an algorithm code calling the same generic kernel twice with potentially different precision. The implementation of the generic kernel such that it supports both 32- and 64-bit precision is also shown. The use of LFRic names for precision in the algorithm code allows precision to be controlled in a simple way. For example, r_solver could be set to be 32-bits in one configuration and 64-bits in another:

program test

  use constants_mod,      only : r_def, r_solver
  use field_mod,          only : field_type
  use r_solver_field_mod, only : r_solver_field_type
  use example_mod,        only : example_type

  type(field_type)          :: field_r_def
  type(r_solver_field_type) :: field_r_solver
  real(kind=r_def)          :: x_r_def
  real(kind=r_solver)       :: x_r_solver

  call invoke( example_type(field_r_def, x_r_def), &
               example_type(field_r_solver, x_r_solver))

end program test

module example_mod

  use argument_mod
  use kernel_mod

  implicit none

  type, extends(kernel_type) :: example_type
    type(arg_type), dimension(2) :: meta_args = (/       &
         arg_type(gh_field,  gh_real, gh_readwrite, w3), &
         arg_type(gh_scalar, gh_real, gh_read )          &
         /)
     integer :: operates_on = cell_column
   contains
     procedure, nopass :: code => example_code
  end type example_type

  private
  public :: example_code

  interface example_code
    module procedure example_code_32
    module procedure example_code_64
  end interface example_code

contains

  subroutine example_code_32(..., field1, x, ...)
    real*4, dimension(...), intent(inout) :: field1
    real*4, intent(in) :: x
    print *, "32-bit example called"
  end subroutine example_code_32

  subroutine example_code_64(..., field1, x, ...)
    real*8, dimension(...), intent(inout) :: field1
    real*8, intent(in) :: x
    print *, "64-bit example called"
  end subroutine example_code_64

end module example_mod

In order to support mixed precision, PSyclone needs to know the precision (as specified in the algorithm layer) of any kernel arguments that are of a type that supports different precisions (e.g. GH_FIELD). The reason for this is that PSyclone needs to be able to declare data with the correct precision information within the PSy-layer to ensure that the correct flavour of kernels are called.

PSyclone must therefore determine this information from the algorithm layer. The rules for whether PSyclone requires information for particular LFRic datatypes and what it does with or without this information are given below:

Fields

PSyclone must be able to determine the datatype of a field from the algorithm layer declarations. If it is not able to do this, PSyclone will abort with a message that indicates the problem.

Supported field types, their Fortran datatype and precisions are outlined in the table below:

Field Type

Fortran Datatype

Precision

field_type

real

r_def

r_bl_field_type

real

r_bl

r_phys_field_type

real

r_phys

r_solver_field_type

real

r_solver

r_tran_field_type

real

r_tran

integer_field_type

integer

i_def

Field Vectors

In addition to fields, LFRic supports an abstract vector type for fields, used in the LFRic solver API. Please note that these structures are different from the field vector implementation of field bundles in the PSyclone LFRic API interface.

The LFRic abstract vector type has precision-specific implementations. If PSyclone finds such a specifically declared field vector argument in the algorithm layer, e.g. r_solver_field_vector_type, it will assume that the actual field being referenced is of the same datatype and precision (see above for details). The correspondence between the available field types and their vector implementations is given in the table below (note that only real-valued fields have abstract vector implementations for now):

Field Type

Field Vector Type

field_type

field_vector_type

r_bl_field_type

r_bl_field_vector_type

r_phys_field_type

r_phys_field_vector_type

r_solver_field_type

r_solver_field_vector_type

r_tran_field_type

r_tran_field_vector_type

If PSyclone finds an argument that is declared as an abstract_field_type then it will not know the actual type of the argument. For instance, the following algorithm layer code will cause PSyclone to raise an exception:

! ...
class (abstract_vector_type), intent(inout) :: x
! ...
select type (x)
type is (field_vector_type)
  call invoke(testkern_type(x%vector(1)))
class default
  print *,"Error"
end select
! ...

The suggested solution to this is to add a pointer variable to the code that is of the required type. This pointer can then be associated with the argument and passed into the routine:

! ...
class (abstract_vector_type), target, intent(inout) :: x
type(field_vector_type), pointer :: x_ptr
! ...
select type (x)
type is (field_vector_type)
  x_ptr => x
  call invoke(testkern_type(x_ptr%vector(1)))
class default
  print *,"Error"
end select
! ...

Scalars

It is not mandatory for PSyclone to be able to determine the datatype of a scalar from the algorithm layer. This constraint was considered to be too restrictive as PSyclone currently only examines the declarations in the same source file as the invoke when determining datatype. This means that if scalars are imported from other modules (as is often the case) then their datatype cannot be determined.

If the precision information for a scalar is found by PSyclone then this is used. If the scalar declaration is found and it contains no precision information then PSyclone will abort with a message that indicates the problem (since this violates LFRic coding standards). If no declaration information is found then default precision values are used, as specified in the PSyclone config file (r_def for real, i_def for integer and l_def for logical).

Supported precisions for scalars are outlined in the table below. If an unsupported scalar precision is found then PSyclone will abort with a message that indicates the problem.

Fortran Datatype

Supported Precision

real

r_def, r_bl, r_phys, r_solver, r_tran

integer

i_def

logical

l_def

LMA Operators

PSyclone must be able to determine the datatype of an LMA operator. If it is not able to do this, PSyclone will abort with a message that indicates the problem.

Supported LMA operator types, their Fortran datatype and precisions are outlined in the table below:

Operator Type

Fortran Datatype

Precision

operator_type

real

r_def

r_solver_operator_type

real

r_solver

r_tran_operator_type

real

r_tran

Column-wise Operators

It is not mandatory for PSyclone to be able to determine the datatype of a column-wise (CMA) operator. The reason for this is that only one datatype is supported, a columnwise_operator_type which contains real-valued data with precision r_solver. PSyclone can therefore simply add this datatype in the PSy-layer. However, if the datatype information is found in the algorithm layer and it is not of the expected type then PSyclone will abort with a message that indicates the problem.

Consistency

If PSyclone is able to determine the datatype of an LFRic datatype then PSyclone also checks that this datatype is consistent with the associated kernel metadata. If it is not consistent then PSyclone will abort with a message that indicates the problem.

PSy-layer

The general details of the PSy-layer are explained in the PSy layer section. This section describes any LFRic-specific issues.

Module name

The PSy-layer code is contained within a Fortran module. The name of the module is determined from the algorithm-layer name with “_psy” appended. The algorithm-layer name is the algorithm’s module name if it is a module, its subroutine name if it is a subroutine that is not within a module, or the program name if it is a program.

So, for example, if the algorithm code is contained within a module called “fred” then the PSy-layer module name will be “fred_psy”.

Argument Intents

LFRic fields, field vectors, operators and column-wise operators are objects that contain pointers to data rather than data. The data are accessed by proxies of these objects and modified in kernels. As the objects themselves are not modified in the PSy layer, their Fortran intents there are always intent(in).

The Fortran intent of scalars is still defined by their access metadata as they are actual data. This means intent(in) for GH_READ and intent(out) for GH_SUM (more details in meta_args section below).

The intent of other data structures is mandated by the relevant LFRic API rules described in sections below.

Kernel

The general requirements for the structure of a Kernel are explained in the Kernel layer section. In the LFRic API there are six different Kernel types; general purpose, CMA, inter-grid, domain, dof and Built-ins. In the case of built-ins, PSyclone generates the source of the kernels. This section explains the rules for the other five user-supplied kernel types and then goes on to describe their metadata and subroutine arguments.

Domain kernels are distinct from the other four user-supplied kernel types because they must be passed data for the whole domain rather than a single cell-column or dof. This permits the use of kernels that have not been written to conform to the single-column/dof approach which simplifies the integration with existing code. Obviously, any parallelisation in the ‘domain’ kernel must be consistent with that in the rest of the application. The motivation for such kernels in LFRic is that they allow existing, “i-first” physics code to be called from the PSy layer. Since those routines currently contain their own, i-first looping structure (and associated OpenMP parallelisation), the most efficient way to use them is to avoid enclosing them within a loop in the PSy layer. This is a temporary measure and these kernels will ultimately be replaced once the LFRic infrastructure has support for i-first kernels (https://code.metoffice.gov.uk/trac/lfric/ticket/2154). At that point the looping (and associated parallelisation) will be put back into the PSy layer.

Rules for all User-Supplied Kernels that Operate on Cell-Columns

In the following, ‘operator’ refers to both LMA and CMA operator types.

  1. A Kernel must have at least one argument that is a field, field vector, or operator. This rule reflects the fact that a Kernel operates on some subset of the whole domain (e.g. a cell-column) and is therefore designed to be called from within a loop that iterates over those subsets of the domain.

  2. The continuity of the iteration space of the Kernel is determined from the function space of the modified argument (see Section Supported Function Spaces below). If more than one argument is modified then the iteration space is taken to be the largest required by any of those arguments. E.g. if a Kernel writes to two fields, the first on W3 (discontinuous) and the second on W1 (continuous), then the iteration space of that Kernel will be determined by the field on the continuous space.

  3. If any of the modified arguments are declared with the generic function space metadata (e.g. ANY_SPACE_<n>, see Supported Function Spaces) and their actual space cannot be determined statically then the iteration space is assumed to be

    1. discontinuous for ANY_DISCONTINUOUS_SPACE_<n>;

    2. continuous for ANY_SPACE_<n> and ANY_W2. This assumption is always safe but leads to additional computation if the quantities being updated are actually on discontinuous function spaces.

  4. Operators do not have halo operations operating on them as they are either cell- (LMA) or column-based (CMA) and therefore act like discontinuous fields.

  5. Any Kernel that writes to an operator will have its iteration space expanded such that valid values for the operator are computed in the level-1 halo.

  6. Any Kernel that reads from an operator must not access halos beyond level 1. In this case PSyclone will check that the Kernel does not require values beyond the level-1 halo. If it does then PSyclone will abort.

  7. Any Kernel that takes an operator argument must not also take an integer-valued field as an argument.

Rules specific to General-Purpose Kernels without CMA Operators

  1. General-purpose kernels with operates_on = CELL_COLUMN accept arguments of any of the following types: field, field vector, LMA operator, scalar (real, integer or logical).

  2. A Kernel is permitted to write to more than one quantity (field or operator) and these quantities may be on the same or different function spaces.

  3. A Kernel may not write to a scalar argument. (Only built-ins are permitted to do this.) Any scalar arguments must therefore be declared in the metadata as GH_READ - see below.

Rules for Kernels that work with CMA Operators

The LFRic API has support for kernels that assemble, apply (or inverse-apply) column-wise/Column Matrix Assembly (CMA) operators. Such operators may also be used by matrix-matrix kernels. There are thus three types of CMA-related kernels. Since, by definition, CMA operators only act on data within a column, they have no horizontal dependencies. Therefore, kernels that write to them may be parallelised without colouring.

All three CMA-related kernel types must obey the following rules:

  1. Since a CMA operator only acts within a single column of data, stencil operations are not permitted.

  2. No vector quantities (e.g. GH_FIELD*3 - see below) are permitted as arguments.

  3. The kernel must operate on cell-columns.

There are then additional rules specific to each of the three CMA kernel types. These are described below.

Assembly

CMA operators are themselves constructed from Local-Matrix-Assembly (LMA) operators. Therefore, any kernel which assembles a CMA operator must obey the following rules:

  1. Have one or more LMA operators as read-only arguments.

  2. Have exactly one CMA operator argument which must have write access.

  3. Other types of argument (e.g. scalars or fields) are permitted but must be read-only.

Application and Inverse Application

Column-wise operators can only be applied to fields. CMA-Application kernels must therefore:

  1. Have a single CMA operator as a read-only argument.

  2. Have exactly two field arguments, one read-only and one that is written to.

  3. The function spaces of the read and written fields must match the from and to spaces, respectively, of the supplied CMA operator.

Matrix-Matrix

A kernel that has just column-wise operators as arguments and zero or more read-only scalars is identified as performing a matrix-matrix operation. In this case:

  1. Arguments must be CMA operators and, optionally, one or more scalars.

  2. Exactly one of the CMA arguments must be written to while all other arguments must be read-only.

Rules for Inter-Grid Kernels

  1. An inter-grid kernel is identified by the presence of a field or field-vector argument with the optional mesh_arg metadata element (see Inter-Grid Metadata).

  2. An invoke that contains one or more inter-grid kernels must not contain any other kernel types. (This restriction is an implementation decision and could be lifted in future if there is a need.)

  3. An inter-grid kernel is only permitted to have field or field-vector arguments.

  4. All inter-grid kernel arguments must have the mesh_arg metadata entry.

  5. An inter-grid kernel (and metadata) must have at least one field on each of the fine and coarse meshes. Specifying all fields as coarse or fine is forbidden.

  6. Fields on different meshes must always live on different function spaces.

  7. All fields on a given mesh must be on the same function space.

  8. An inter-grid kernel must operate on cell-columns.

A consequence of Rules 5-7 is that an inter-grid kernel will only involve two function spaces.

Rules for User-Supplied Kernels that Operate on the Domain

The rules for kernels that have operates_on = DOMAIN are a subset of those for kernels that operate on a CELL_COLUMN without CMA Operators. Specifically:

  1. Only scalar, field and field vector arguments are permitted.

  2. All fields must be on discontinuous function spaces.

  3. Stencil accesses are not permitted.

Rules for all User-Supplied Kernels that Operate on DoFs (DoF Kernels)

Kernels that have operates_on = DOF and LFRic Built-ins overlap significantly in their scope, and the conventions that DoF Kernels must follow are influenced by those for built-ins as a result. This includes metadata arguments and valid data types and access modes. Naming conventions for DoF Kernels should follow those for General-Purpose Kernels.

The list of rules for DoF Kernels is as follows:

  1. A DoF Kernel must have at least one argument that is a field. This rule reflects that a Kernel operates on some subset of the whole domain and is therefore designed to be called from within a loop that iterates over those subsets of the domain. Only fields (as opposed to e.g. field vectors or operators) are accepted for DoF Kernels because only they have a single value at each DoF.

  2. All Kernel arguments must be either fields or scalars (real- and/or integer-valued). DoF Kernels cannot accept operators.

  3. All field arguments to a given DoF Kernel must be on the same function space so they have the same number of DoFs.

  4. They must have at least one modified (i.e. written to) field argument. Unlike built-ins, this is not limited and more than one modified argument is allowed.

  5. A Kernel may not write to a scalar argument. (Only built-ins are permitted to do this.) Any scalar arguments must therefore be declared in the metadata as GH_READ - see below

  6. Kernels must be written to operate on a single DoF, such that field values at the same dof location/index can be provided to the Kernel within a loop over the DoFs of the function space of the field that is being updated.

Metadata

The code below outlines the elements of the LFRic API Kernel metadata, 1) ‘meta_args’, 2) ‘meta_funcs’, 3) ‘meta_reference_element’, 4) ‘meta_mesh’, 5) ‘gh_shape’ (gh_shape and gh_evaluator_targets), 6) ‘operates_on’ and 7) ‘procedure’:

type, public, extends(kernel_type) :: my_kernel_type
  type(arg_type) :: meta_args(...) = (/ ... /)
  type(func_type) :: meta_funcs(...) = (/ ... /)
  type(reference_element_data_type) :: meta_reference_element(...) = (/ ... /)
  type(mesh_data_type) :: meta_mesh(...) = (/ ... /)
  integer :: gh_shape = gh_quadrature_XYoZ
  integer :: operates_on = cell_column
contains
  procedure, nopass :: my_kernel_code
end type

These various metadata elements are discussed in order in the following sections.

meta_args

The meta_args array specifies information about data that the kernel code expects to be passed to it via its argument list. There is one entry in the meta_args array for each scalar, field, or operator passed into the Kernel and the order that these occur in the meta_args array must be the same as they are expected in the kernel code argument list. The entry must be of arg_type which itself contains metadata about the associated argument. The size of the meta_args array must correspond to the number of scalars, fields and operators passed into the Kernel.

Note

It makes no sense for a Kernel to have only scalar arguments (because the PSy layer will call a Kernel for each point in the spatial domain) and PSyclone will reject such Kernels.

For example, if there are a total of 2 scalar / field / operator entities being passed to the Kernel then the meta_args array will be of size 2 and there will be two arg_type entries:

type(arg_type) :: meta_args(2) = (/                                  &
     arg_type( ... ),                                                &
     arg_type( ... )                                                 &
     /)

Argument metadata (information contained within the brackets of an arg_type entry), describes either a scalar, a field or an operator (either LMA or CMA).

The first argument-metadata entry describes whether the data that is being passed is for a scalar (GH_SCALAR), a field (GH_FIELD) or an operator (either GH_OPERATOR for LMA or GH_COLUMNWISE_OPERATOR for CMA). This information is mandatory.

Additionally, argument metadata can be used to describe a vector of fields (see the Field Vector section for more details).

As an example, the following meta_args metadata describes 4 entries, the first is a scalar, the next two are fields and the fourth is an operator. The third entry is a field vector of size 3.

type(arg_type) :: meta_args(4) = (/                                  &
     arg_type(GH_SCALAR, GH_REAL, ...),                              &
     arg_type(GH_FIELD, GH_INTEGER, ... ),                           &
     arg_type(GH_FIELD*3, GH_REAL, ... ),                            &
     arg_type(GH_OPERATOR, GH_REAL, ...)                             &
     /)

The second item in a metadata entry describes the Fortran primitive (intrinsic) type of the data of a kernel argument. The currently supported values are GH_REAL, GH_INTEGER and GH_LOGICAL for real, integer and logical data, respectively. This information is mandatory. Valid data types for each LFRic API argument type are specified later in this section (see Valid Data Types).

The third component of argument metadata describes how the Kernel makes use of the data being passed into it (the way it is accessed within a Kernel). This information is mandatory. There are currently 6 possible values of this metadata GH_READ, GH_WRITE, GH_READWRITE, GH_INC, GH_READINC and GH_SUM. However, not all combinations of metadata entries are valid and PSyclone will raise an exception if an invalid combination is specified. Valid combinations are specified later in this section (see Valid Access Modes).

  • GH_READ indicates that the data is read and is unmodified.

  • GH_WRITE indicates the data is modified in the Kernel before (optionally) being read. If any shared DoFs are written to then different iterations of the Kernel must write the same value.

  • GH_READWRITE indicates that different iterations of a Kernel update quantities which do not share DoFs, such as operators and fields over discontinuous function spaces. If a Kernel modifies only discontinuous fields and/or operators there is no need for synchronisation or colouring when running such Kernels in parallel. However, modifying another field with a GH_INC access in a Kernel means that synchronisation or colouring is required for parallel runs.

  • GH_INC indicates that different iterations of a Kernel make contributions to shared values. For example, values at cell faces may receive contributions from cells on either side of the face. This means that such a Kernel needs appropriate synchronisation (or colouring) to run in parallel.

  • GH_READINC indicates that the data is first read and then subsequently incremented. Therefore this is equivalent to a GH_READ followed by a GH_INC.

  • GH_SUM is an example of a reduction and is the only reduction currently supported in PSyclone. This metadata indicates that values are summed over calls to Kernel code.

For example:

type(arg_type) :: meta_args(6) = (/                            &
     arg_type(GH_OPERATOR, GH_REAL,    GH_READ,      ... ),    &
     arg_type(GH_FIELD*3,  GH_REAL,    GH_WRITE,     ... ),    &
     arg_type(GH_FIELD,    GH_REAL,    GH_READWRITE, ... ),    &
     arg_type(GH_FIELD,    GH_INTEGER, GH_INC,       ... ),    &
     arg_type(GH_FIELD,    GH_REAL,    GH_READINC,   ... ),    &
     arg_type(GH_SCALAR,   GH_REAL,    GH_SUM)                 &
     /)

Warning

It is important that GH_INC is not incorrectly used in place of a GH_READINC access as it could result in the reading of data from a dirty outermost halo when run in parallel, giving incorrect results. The reason for this is that PSyclone does not add a halo exchange for the outermost modified halo level of a field before a loop that contains a GH_INC access to that field, i.e. a loop iterating to the level-n halo will result in a halo exchange to the level-(n-1) halo being added before the loop (which means no halo exchange is added when n==1). The reason this can be performed is because any computation in the outermost halo will be incorrect (will only compute partial sums) and PSyclone therefore sets this halo level to dirty after the loop has completed. There is, therefore, no reason to make the values of the incremented field clean for the outermost modified halo. However, this optimisation does require that any (dirty) data in the outermost modified halo does not result in exceptions. With some compilers an exception can occur for a field that has not yet had its outermost halo data written to, i.e. if the uninitialised data is read. To avoid this potential problem in user code it is recommended that a redundant computation transformation is added to compute all setval_c, setval_x and setval_random Built-in calls (see Built-ins) to the same halo depth as the associated GH_INC access - which is level-1 without any redundant computation transformations being applied to the associated loops. This will guarantee that all data has been initialised with a value before it is incremented and avoid any potential exceptions.

Note

In the LFRic API only Built-ins are permitted to write to scalar arguments (and hence perform reductions). Furthermore, this permission is currently restricted to real scalars (GH_SCALAR, GH_REAL) as the LFRic infrastructure does not yet support integer and logical reductions.

For a scalar, the argument metadata contains only these three entries. However, fields and operators require further entries specifying function-space information. The meaning of these further entries differs depending on whether a field or an operator is being described.

In the case of an operator, the fourth and fifth arguments describe the to and from function spaces respectively. In the case of a field the fourth argument specifies the function space that the field lives on. More details about the supported function spaces are in subsection Supported Function Spaces.

For example, the metadata for a kernel that applies a column-wise operator to a field might look like:

type(arg_type) :: meta_args(3) = (/                              &
     arg_type(GH_FIELD, GH_REAL, GH_INC, W1),                    &
     arg_type(GH_FIELD, GH_REAL, GH_READ, W2H),                  &
     arg_type(GH_COLUMNWISE_OPERATOR, GH_REAL, GH_READ, W1, W2H) &
     /)

In some cases a Kernel may be written so that it works for fields and/or operators from any type of a vector W2* space (all W2* spaces except for the W2*trace spaces, see Section Supported Function Spaces below). In this case the metadata should be specified as being ANY_W2.

Warning

In the current implementation it is assumed that all fields and/or operators specifying ANY_W2 within a kernel will use the same function space. It is up to the user to ensure this is the case as otherwise invalid code would be generated.

It may be that a Kernel is written such that a field and/or operators may be on/map-between any function space(s). In this case the metadata should be specified as being one of ANY_SPACE_1, …, ANY_SPACE_<nmax> (see Supported Function Spaces), with the number of spaces, <nmax>, being set in the PSyclone configuration file (see here for more details on this option).

If the generic function spaces are known to be discontinuous the metadata may be specified as being one of ANY_DISCONTINUOUS_SPACE_1, …, ANY_DISCONTINUOUS_SPACE_<nmax> in order to avoid unnecessary computation into the halos (see rules for user-supplied kernels above). The reason for having different names is that a Kernel might be written to allow 2 or more arguments to be able to support any function space but for a particular call the function spaces may have to be the same as each other. Again, <nmax> is the configurable number of generalised discontinuous function spaces.

In the example below, the first field entry supports any function space but it must be the same as the operator’s to function space. Similarly, the second field entry supports any function space but it must be the same as the operator’s from function space. Note, the metadata does not forbid ANY_SPACE_1 and ANY_SPACE_2 from being the same.

type(arg_type) :: meta_args(3) = (/                                    &
     arg_type(GH_FIELD,    GH_REAL, GH_INC,  ANY_SPACE_1),             &
     arg_type(GH_FIELD*3,  GH_REAL, GH_INC,  ANY_SPACE_2),             &
     arg_type(GH_OPERATOR, GH_REAL, GH_READ, ANY_SPACE_1, ANY_SPACE_2) &
     /)

Note also that the scope of this naming of any-space function spaces is restricted to the argument list of individual kernels. I.e. if an Invoke contains say, two kernel calls that each support arguments on any function space, e.g. ANY_SPACE_1, there is no requirement that these two function spaces be the same. Put another way, if an Invoke contained two calls of a kernel with arguments described by the above metadata then the first field argument passed to each kernel call need not be on the same space.

Valid Data Types

As mentioned earlier, the currently supported Fortran primitive (intrinsic) types for kernel argument data are real, integer and logical, described by the GH_REAL, GH_INTEGER and GH_LOGICAL metadata descriptors. Supported data types for each argument type are given in the table below (please note that field vectors follow the same rules as the LFRic fields):

Argument Type

Data Type

GH_SCALAR

GH_REAL, GH_INTEGER, GH_LOGICAL

GH_FIELD

GH_REAL, GH_INTEGER

GH_OPERATOR

GH_REAL

GH_COLUMNWISE_OPERATOR

GH_REAL

Valid Access Modes

As mentioned earlier, not all combinations of metadata are valid. Valid combinations for each argument type in user-defined Kernels are summarised here. All argument types (GH_SCALAR, GH_FIELD, GH_OPERATOR and GH_COLUMNWISE_OPERATOR) may be read within a Kernel and this is specified in metadata using GH_READ. At least one kernel argument must be listed as being modified. When data is modified in a user-supplied Kernel (i.e. a Kernel that operates on a CELL_COLUMN, see iteration space metadata) then the permitted access modes depend upon the argument type and the function space it is on:

Argument Type

Function Space

Access Type

GH_SCALAR

n/a

GH_READ

GH_FIELD

Discontinuous

GH_READ, GH_WRITE, GH_READWRITE

GH_FIELD

Continuous

GH_READ, GH_WRITE, GH_INC, GH_READINC

GH_OPERATOR

Any for both ‘to’ and ‘from’

GH_READ, GH_WRITE, GH_READWRITE

GH_COLUMNWISE_OPERATOR

Any for both ‘to’ and ‘from’

GH_READ, GH_WRITE, GH_READWRITE

Note that scalar arguments to user-defined Kernels must be read-only. Only Built-ins are permitted to modify scalar arguments. In practice this means that the only allowed access for the scalars in user-defined Kernels is GH_READ (see the allowed accesses for arguments in Built-ins in the section below).

Note also that a GH_FIELD argument that has GH_WRITE or GH_READWRITE as its access pattern must typically (see below) be on a horizontally-discontinuous function space (see Supported Function Spaces for the list of discontinuous function spaces). Parallelisation of the loop over the horizontal domain for a Kernel that updates such a field will not require colouring for either of the above cases (since there are no shared entities).

There is however an exception to this - certain Kernels may write to shared entities but each Kernel iteration is guaranteed to write the same value to a given shared DoF. In this case, provided that the first access to any such shared DoF is a write, the loop containing such a Kernel may be parallelised without colouring. Therefore, GH_WRITE access is permitted for GH_FIELD arguments on continuous function spaces. Obviously, care must be taken to ensure that the Kernel implementation satisfies the constraints just described as PSyclone cannot currently check this.

If a field is described as being on ANY_SPACE_*, there is currently no way to determine its continuity from the metadata (unless we can statically determine the space of the field being passed in). At the moment this type of a user-supplied Kernel is always treated as if it is updating a field that is on a function space that is continuous in the horizontal, even if it is not (see rules for user-supplied kernels above).

There is no restriction on the number and function spaces of other quantities that a general-purpose kernel can modify other than that it must modify at least one. The rules for kernels involving CMA operators, however, are stricter and only one argument may be modified (the CMA operator itself for assembly, a field for CMA-application and a CMA operator for matrix-matrix kernels). If a kernel writes to quantities on different function spaces then PSyclone generates loop bounds appropriate to the largest iteration space. This means that if a single kernel updates one quantity on a continuous function space and one on a discontinuous space then the resulting loop will include cells in the level-1 halo since they are required for a quantity on a continuous space. As a consequence, any quantities on a discontinuous space will then be computed redundantly in the level-1 halo. Currently PSyclone makes no attempt to take advantage of this (by e.g. setting the appropriate level-1 halo to ‘clean’).

PSyclone ensures that both CMA and LMA operators are computed (redundantly) out to the level-1 halo cells. This permits their use in kernels which modify quantities on continuous function spaces and also in subsequent redundant computation of other quantities on discontinuous function spaces. In conjunction with this, PSyclone also checks (when generating the PSy layer) that any kernels which read operator values do not do so beyond the level-1 halo. If any such accesses are found then PSyclone aborts.

Supported Function Spaces

As mentioned in the Field and Field Vector sections, the function space of an argument specifies how it maps onto the underlying topology and, additionally, whether the data at a point is a vector. In LFRic API the dimension of the basis function set for the scalar function spaces is 1 and for the vector function spaces is 3 (see the table in Rules for General-Purpose Kernels for the dimensions of the basis and differential basis functions).

Function spaces can share DoFs between cells in the horizontal, vertical or both directions. Depending on the function space and FEM order, the shared DoFs can lie on one or more cell entities (faces, edges and vertices) in each direction. This property is referred to as the continuity of a function space (horizontal, vertical or full). Alternatively, if there are no shared DoFs a function space is described as discontinuous (fully or in a particular direction).

The mixed FEM formulation is built on a foundation set of four function spaces described below.

  • W0 is the space of scalar functions with full continuity. The shared DoFs lie on cell vertices in the lowest order FEM and on all three entities in higher order FEM.

  • W1 is the space of vector functions with full continuity in the tangential direction only. In the lowest order FEM the shared DoFs lie on cell edges for each component, whereas in higher order they also lie on cell faces.

  • W2 is the space of vector functions with full continuity in the normal direction only. The shared DoFs lie on cell faces for each component.

  • W3 is the space of scalar functions with full discontinuity. All DoFs lie within the cell volume and are not shared across the cell boundaries.

Other spaces required for representation of scalar or component-wise vector variables are:

  • Wtheta is the space of scalar functions based on the vertical part of W2, discontinuous in the horizontal and continuous in the vertical;

  • W2H is the space of vector functions based on the horizontal part of W2, continuous in the horizontal and discontinuous in the vertical;

  • W2V is the space of vector functions based on the vertical part of W2, discontinuous in the horizontal and continuous in the vertical;

  • W2broken is the space of vector functions, locally identical to the W2 space. However, DoFs are topologically discontinuous in all directions despite their placement on cell faces;

  • W2trace is the space of scalar functions defined only on cell faces, resulting from taking the trace of a W2 space. DoFs are shared between faces, hence making this space fully continuous;

  • W2Htrace is the space of scalar functions defined only on cell faces in the horizontal, resulting from taking the trace of a W2H space. DoFs are shared between horizontal faces, hence making this space continuous in the horizontal and discontinuous in the vertical;

  • W2Vtrace is the space of scalar functions defined only on cell faces in the vertical, resulting from taking the trace of a W2V space. DoFs are shared between vertical faces, hence making this space discontinuous in the horizontal and continuous in the vertical;

  • Wchi is the space of scalar functions used to store coordinates in LFRic. It is fully discontinuous except for the coordinate order 0 when it becomes the W0 space (i.e. fully continuous). Please see the next section for more details on this function space.

In addition to the specific function space metadata, there are also three generic function space metadata descriptors mentioned in sections above:

  • ANY_SPACE_<n>, n = 1, 2, … nmax, for when the function space of the argument(s) cannot be determined and/or for when a Kernel has been written so that it works with fields on any of the available spaces (as mentioned in the meta_args section, the number of spaces, <nmax>, is configurable);

  • ANY_DISCONTINUOUS_SPACE_<n>, n = 1, 2, … nmax, for when the function space of the argument(s) cannot be determined but is known to be discontinuous and/or for when a Kernel has been written so that it works with fields on any of the discontinuous spaces (again, the number of spaces, <nmax>, is configurable);

  • ANY_W2 for any type of a vector W2* function space, i.e. W2, W2H, W2V and W2broken but not W2*trace spaces.

As mentioned previously , ANY_SPACE_<n> and ANY_W2 function space types are treated as continuous while ANY_DISCONTINUOUS_SPACE_<n> spaces are treated as discontinuous.

Note

The name and use of ANY_W2 metadata (e.g. continuity and vector or/and scalar basis of W2* spaces the metadata can represent) are being reviewed in PSyclone issue #540.

Since the LFRic API operates on columns of data, function spaces are categorised as continuous or discontinuous with regard to their continuity in the horizontal. For example, a GH_FIELD that specifies GH_INC as its access pattern (see :ref:lfric-kernel-valid-access: above) may be continuous in the vertical (and discontinuous in the horizontal), continuous in the horizontal (and discontinuous in the vertical), or continuous in both. In each case the code is the same. This principle of horizontal continuity also applies to the three generic ANY_*_* function space identifiers above. The valid metadata values for continuous and discontinuous function spaces are summarised in the table below.

Function Space Continuity

Function Space Name

Continuous

W0, W1, W2, W2H, W2trace, W2Htrace, ANY_W2, ANY_SPACE_<n>

Discontinuous

W2broken, W2V, W2Vtrace, W3, Wtheta, ANY_DISCONTINUOUS_SPACE_<n>

Horizontally discontinuous function spaces and fields over them will not need colouring so PSyclone does not perform it. If such attempt is made, PSyclone will raise a Generation Error in the Dynamo0p3ColourTrans transformation (see Transformations for more details on transformations). An example of fields iterating over a discontinuous function space Wtheta is given in examples/lfric/eg9, with the GH_READWRITE access descriptor denoting an update to the relevant fields. This example also demonstrates how to only colour loops over continuous function spaces when transformations are applied.

Read-Only Function Spaces

LFRic supports the concept of a read-only function space. A field on such a function space must not be modified by any kernels contained within invoke calls (i.e. within any code that PSyclone is responsible for). Further, a field on a read-only function space must contain clean halos in order to avoid any halo exchanges that would occur if the field is read within a kernel where redundant computation is performed.

The primary reason for including a read-only function space is that it does not need any halo-exchange support e.g. it does not require a routing table, which can reduce the memory footprint.

Currently Wchi is the only read-only function space in LFRic.

As a read-only function space is not modified, it does not matter whether it is classified as continuous or discontinuous. LFRic therefore treats read-only as a third category of function space.

Optional Field Metadata

A field entry in the meta_args array may have an optional fifth element. This element describes either a stencil access or, for inter-grid kernels, which mesh the field is on. Since an inter-grid kernel is not permitted to have stencil accesses, these two options are mutually exclusive. The metadata for each case is described in the following sections.

Stencil Metadata

Stencil metadata specifies that the corresponding field argument is accessed as a stencil operation within the Kernel. Stencil metadata only makes sense if the associated field is read within a Kernel i.e. it only makes sense to specify stencil metadata if the first entry is GH_FIELD and the second entry is GH_READ.

Stencil metadata is written in the following format:

STENCIL(type)

where type may be one of X1D, Y1D, XORY1D, CROSS, CROSS2D or REGION. As the stencil extent (the maximum distance from the central cell that the stencil extends) is not provided in the metadata, it is expected to be provided by the algorithm writer as part of the invoke call (see Section Stencil Extent). As there is currently no way to specify a fixed extent value for stencils in the Kernel metadata, Kernels must therefore be written to support different values of extent (i.e. stencils with a variable number of cells).

The XORY1D stencil type indicates that the Kernel can accept either X1D or Y1D stencils. In this case it is up to the algorithm developer to specify which of these it is from the algorithm layer as part of the invoke call (see Section Stencil Extent).

For example, the following stencil (with extent=2):

| 3 | 2 | 1 | 4 | 5 |

would be declared as:

STENCIL(X1D)

and the following stencil (with extent=2):

|   |   | 9 |   |   |
|   |   | 8 |   |   |
| 3 | 2 | 1 | 6 | 7 |
|   |   | 4 |   |   |
|   |   | 5 |   |   |

would be declared as:

STENCIL(CROSS)

The REGION stencil references a block of cells:

| 9 | 8 | 7 |
| 2 | 1 | 6 |
| 3 | 4 | 5 |

and would be declared as:

STENCIL(REGION)

Below is an example of stencil information within the full kernel metadata:

type(arg_type) :: meta_args(3) = (/                                  &
     arg_type(GH_FIELD,    GH_REAL, GH_INC,  W1),                    &
     arg_type(GH_FIELD,    GH_REAL, GH_READ, W2H, STENCIL(CROSS)),   &
     arg_type(GH_OPERATOR, GH_REAL, GH_READ, W1, W2H)                &
     /)

There is a full example of this distributed with PSyclone. It may be found in examples/lfric/eg5.

Inter-Grid Metadata

The alternative form of the optional fifth metadata argument for a field specifies which mesh the associated field is on. This is required for inter-grid kernels which perform prolongation or restriction operations on fields (or field vectors) existing on grids of different resolutions.

Mesh metadata is written in the following format:

mesh_arg=type

where type may be one of GH_COARSE or GH_FINE. Any kernel having a field argument with this metadata is assumed to be an inter-grid kernel and, as such, all of its other arguments (which must also be fields) must have it specified too. An example of the metadata for such a kernel is given below:

type(arg_type) :: meta_args(2) = (/                                      &
    arg_type(GH_FIELD, GH_REAL, GH_READWRITE, ANY_DISCONTINUOUS_SPACE_1, &
                                                    mesh_arg=GH_COARSE), &
    arg_type(GH_FIELD, GH_REAL, GH_READ,      ANY_DISCONTINUOUS_SPACE_2, &
                                                    mesh_arg=GH_FINE  )  &
    /)

Note that an inter-grid kernel must have at least one field (or field- vector) argument on each mesh type. Fields that are on different meshes cannot be on the same function space while those on the same mesh must also be on the same function space.

Column-wise Operators (CMA)

In this section we provide example metadata for each of the three recognised kernel types involving CMA operators.

Column-wise operators are constructed from cell-wise (local) operators. Therefore, in order to assemble a CMA operator, a kernel must have at least one read-only LMA operator, e.g.:

type(arg_type) :: meta_args(2) = (/                                                 &
     arg_type(GH_OPERATOR,            GH_REAL, GH_READ,  ANY_SPACE_1, ANY_SPACE_2), &
     arg_type(GH_COLUMNWISE_OPERATOR, GH_REAL, GH_WRITE, ANY_SPACE_1, ANY_SPACE_2)  &
     /)

CMA operators (and their inverse) are applied to fields. Therefore any kernel of this type must have one read-only CMA operator, one read-only field and a field that is updated, e.g.:

type(arg_type) :: meta_args(3) = (/                                               &
     arg_type(GH_FIELD,               GH_REAL, GH_INC,  ANY_SPACE_1),             &
     arg_type(GH_FIELD,               GH_REAL, GH_READ, ANY_SPACE_2),             &
     arg_type(GH_COLUMNWISE_OPERATOR, GH_REAL, GH_READ, ANY_SPACE_1, ANY_SPACE_2) &
     /)

Matrix-matrix kernels compute the product/linear combination of CMA operators. They must therefore have one such operator that is updated while the rest are read-only. They may also have read-only scalar arguments, e.g.:

type(arg_type) :: meta_args(3) = (/                                                 &
     arg_type(GH_COLUMNWISE_OPERATOR, GH_REAL, GH_WRITE, ANY_SPACE_1, ANY_SPACE_2), &
     arg_type(GH_COLUMNWISE_OPERATOR, GH_REAL, GH_READ,  ANY_SPACE_1, ANY_SPACE_2), &
     arg_type(GH_COLUMNWISE_OPERATOR, GH_REAL, GH_READ,  ANY_SPACE_1, ANY_SPACE_2), &
     arg_type(GH_SCALAR,              GH_REAL, GH_READ) /)

Note

The order with which arguments are specified in metadata for CMA kernels does not affect the process of identifying the type of kernel (whether it is assembly, matrix-matrix etc.)

meta_funcs

The (optional) second component of kernel metadata specifies whether any quadrature or evaluator data is required for a given function space. (If no quadrature or evaluator data is required then this metadata should be omitted.) Consider the following kernel metadata:

type, extends(kernel_type) :: testkern_operator_type
  type(arg_type), dimension(3) :: meta_args =                 &
      (/ arg_type(gh_operator, gh_real,    gh_write, w0, w0), &
         arg_type(gh_field*3,  gh_real,    gh_read,  w1),     &
         arg_type(gh_scalar,   gh_integer, gh_read)           &
      /)
  type(func_type) :: meta_funcs(2) =                          &
      (/ func_type(w0, gh_basis, gh_diff_basis)               &
         func_type(w1, gh_basis)                              &
      /)
  integer :: gh_shape = gh_quadrature_XYoZ
  integer :: operates_on = cell_column
contains
  procedure, nopass :: code => testkern_operator_code
end type testkern_operator_type

The arg_type component of this metadata describes a kernel that takes three arguments (an operator, a field and an integer scalar). Following the meta_args array we now have a meta_funcs array. This allows the user to specify that the kernel requires basis functions (gh_basis) and/or the differential of the basis functions (gh_diff_basis) on one or more of the function spaces associated with the arguments listed in meta_args. In this case we require both for the W0 function space but only basis functions for W1.

Note

Basis and differential basis functions for both real- and integer-valued field arguments have real values on the points on which these functions are required.

meta_reference_element

A kernel that requires properties of the reference element in LFRic specifies those properties through the meta_reference_element metadata entry. (If no reference element properties are required then this metadata should be omitted.) Consider the following example kernel metadata:

type, extends(kernel_type) :: testkern_type
  type(arg_type), dimension(2) :: meta_args =      &
      (/ arg_type(gh_field, gh_real, gh_read, w1), &
         arg_type(gh_field, gh_real, gh_inc,  w0) /)
  type(reference_element_data_type), dimension(2) ::               &
    meta_reference_element =                                       &
      (/ reference_element_data_type(normals_to_horizontal_faces), &
         reference_element_data_type(normals_to_vertical_faces) /)
contains
  procedure, nopass :: code => testkern_code
end type testkern_type

This metadata specifies that the testkern_type kernel requires two properties of the reference element. The supported properties are listed below:

Name

Description

normals_to_horizontal_faces

Array of normals pointing in the positive (x, y, z) axis direction for each horizontal face indexed as (component, face).

normals_to_vertical_faces

Array of normals pointing in the positive (x, y, z) axis direction for each vertical face indexed as (component, face).

normals_to_faces

Array of normals pointing in the positive (x, y, z) axis direction for each face indexed as (component, face).

outward_normals_to_horizontal_faces

Array of outward-pointing normals for each horizontal face indexed as (component, face).

outward_normals_to_vertical_faces

Array of outward-pointing normals for each vertical face indexed as (component, face).

outward_normals_to_faces

Array of outward-pointing normals for each face indexed as (component, face).

meta_mesh

A kernel that requires properties of the LFRic mesh object specifies those properties through the meta_mesh metadata entry. (If no mesh properties are required then this metadata should be omitted.) Consider the following example kernel metadata:

type, extends(kernel_type) :: testkern_type
  type(arg_type), dimension(2) :: meta_args =      &
      (/ arg_type(gh_field, gh_real, gh_read, w1), &
         arg_type(gh_field, gh_real, gh_inc,  w0) /)
  type(mesh_data_type), dimension(1) ::            &
    meta_mesh =                                    &
      (/ mesh_data_type(adjacent_face) /)
contains
  procedure, nopass :: code => testkern_code
end type testkern_type

This metadata specifies that the testkern_type kernel requires one property of the mesh. There is currently one supported property:

Name

Description

adjacent_face

Local ID of a neighbouring face in each horizontally-adjacent cell indexed as (face).

gh_shape and gh_evaluator_targets

If a kernel requires basis or differential-basis functions then the metadata must also specify the set of points on which these functions are required. This information is provided by the gh_shape component of the metadata. Currently PSyclone supports four shapes; gh_quadrature_XYoZ for Gaussian quadrature points, gh_quadrature_face for quadrature points on cell faces, gh_quadrature_edge for quadrature points on cell edges and gh_evaluator for evaluation at nodal points. If a kernel requires just one of these then gh_shape is an integer scalar. However, if more than one is required then gh_shape becomes a one-dimensional, integer array, e.g.:

integer :: gh_shape(2) = (/ gh_quadrature_face, gh_quadrature_edge /)

If a kernel requires an evaluator then there are two options: if an evaluator is required for multiple function spaces then these can be specified using the additional gh_evaluator_targets metadata entry. This entry is a one-dimensional, integer array containing the desired function spaces. For example, to request basis/differential-basis functions evaluated on both W0 and W1, the metadata would be:

integer :: gh_shape = gh_evaluator
integer :: gh_evaluator_targets(2) = (/W0, W1/)

The kernel must have an argument (field or operator) on each of the function spaces listed in gh_evaluator_targets. The default behaviour if gh_evaluator_targets is not specified is to provide evaluators for each function space associated with the quantities that the kernel is updating. All necessary data is extracted in the PSy layer and passed to the kernel(s) as required - nothing is required from the Algorithm layer. If a kernel requires quadrature on the other hand, the Algorithm writer must supply a quadrature_type object for each specified quadrature as the last argument(s) to the kernel (see Section Quadrature).

Note that it is an error for kernel metadata to specify a value for gh_shape if no basis or differential-basis functions are required. It is also an error to specify gh_evaluator_targets if the kernel does not require an evaluator (i.e. gh_shape != gh_evaluator).

operates_on

The fourth type of metadata provided is OPERATES_ON. This specifies that the Kernel has been written with the assumption that it is supplied with the specified data for each field/operator argument. The possible values for OPERATES_ON and their interpretation are summarised in the following table:

operates_on

Data passed for each field/operator argument

cell_column

Single column of cells from the owned region (except when performing an INC operation on continuous fields when it will include one level of halo cells).

halo_cell_column

Single column of cells exclusively from halo region.

owned_and_halo_cell_column

Single column of cells but iteration space will include both owned and halo regions.

dof

Single DoF .

domain

All columns of cells in the (sub-)domain.

(For a description of the concepts of ‘owned’ and ‘halo’ cells please see the dev_guide:lfric-developers.)

procedure

The fifth and final type of metadata is procedure metadata. This specifies the name of the Kernel subroutine that this metadata describes.

For example:

procedure, nopass :: my_kernel_subroutine

Subroutine

Rules for General-Purpose Kernels

The arguments to general-purpose kernels (those that do not involve either CMA operators or prolongation/restriction operations) that operate on cell-columns follow a set of rules which have been specified for the LFRic API. These rules are encoded in the generate() method within the ArgOrdering abstract class in the dynamo0p3.py file. The rules, along with PSyclone’s naming conventions, are:

  1. If an LMA operator is passed then include the cells argument. cells is an integer of kind i_def and has intent in.

  2. Include nlayers, the number of layers in a column. nlayers is an integer of kind i_def and has intent in. PSyclone will obtain the value of nlayers to use for a particular kernel from the first field or operator in the argument list.

  3. For each scalar/field/vector_field/operator in the order specified by the meta_args metadata:

    1. If the current entry is a scalar quantity then include the Fortran variable in the argument list. The intent is determined from the metadata (see meta_args for an explanation).

    2. If the current entry is a field then include the field array. The field array name is currently specified as being "field_"<argument_position>"_"<field_function_space>. A field array is a rank-1, real array with extent equal to the number of unique degrees of freedom for the space that the field is on. Its precision (kind) depends on how it is defined in the algorithm layer, see the Mixed Precision section for more details. This value is passed in separately. Again, the intent is determined from the metadata (see meta_args).

      1. If the field entry has a stencil access then add an integer (or if the stencil is of type CROSS2D, an integer rank-1 array of extent 4 and kind i_def) stencil-size argument with intent in. This will supply the number of cells in the stencil or, in the case of the CROSS2D stencil, the number of cells in each branch of the stencil.

      2. If the stencil is of type CROSS2D then an integer of kind i_def and intent in for the max branch length is needed. This is used in defining the dimensions of the stencil dofmap array and is required due to the varying length of the branches of the stencil when used on planar meshes.

      3. Also needed is a stencil dofmap array of type integer, kind i_def and intent in in either 2 or 3 dimensions. For a CROSS2D stencil the array needs dimensions of (number-of-dofs-in-cell, max-branch-length, 4). All other stencils need dimensions of (number-of-dofs-in-cell, stencil-size).

      4. If the field entry stencil access is of type XORY1D then add an additional integer direction argument of kind i_def and with intent in.

    3. If the current entry is a field vector then for each dimension of the vector, include a field array. The field array name is specified as being using "field_"<argument_position>"_"<field_function_space>"_v"<vector_position>. A field array in a field vector is declared in the same way as a field array (described in the previous step).

    4. If the current entry is an operator then first include an integer extent of kind i_def. The name of this extent is <operator_name>"_ncell_3d". Next include the operator. This is a rank-3, real array. Its precision (kind) depends on how it is defined in the algorithm layer, see the Mixed Precision section for more details. The extent of the first dimension is <operator_name>"_ncell_3d", and of the second and third dimension are the local degrees of freedom for the to and from function spaces, respectively. Again the intent is determined from the metadata (see meta_args).

  4. For each function space in the order they appear in the metadata arguments (the to function space of an operator is considered to be before the from function space of the same operator as it appears first in lexicographic order)

    1. Include the number of local degrees of freedom (i.e. number per-cell) for the function space. This is an integer of kind i_def and has intent in. The name of this argument is "ndf_"<field_function_space>.

    2. If there is a field on this space

      1. Include the unique number of degrees of freedom for the function space. This is an integer of kind i_def and has intent in. The name of this argument is "undf_"<field_function_space>.

      2. Include the dofmap for this function space. This is an integer array of kind i_def with intent in. It has one dimension sized by the local degrees of freedom for the function space.

    3. For each operation on the function space (basis, diff_basis), in the order specified in the metadata, pass real arrays of kind r_def with intent in. For each shape specified in the gh_shape metadata entry:

      1. If shape is gh_quadrature_* then the arrays are of rank four and are named "basis_"<field_function_space>_<quadrature_arg_name> or "diff_basis_"<field_function_space>_<quadrature_arg_name>, as appropriate:

        1. If shape is gh_quadrature_xyoz then the arrays have extent (dimension, number_of_dofs, np_xy, np_z).

        2. If shape is gh_quadrature_face or gh_quadrature_edge then the arrays have extent (dimension, number_of_dofs, np_xyz, nfaces or nedges).

      2. If shape is gh_evaluator then we pass one array for each target function space (i.e. as specified by gh_evaluator_targets). Each of these arrays are of rank three with extent (dimension, number_of_dofs, ndf_<target_function_space>). The name of the argument is "basis_"<field_function_space>"_on_"<target_function_space> or "diff_basis_"<field_function_space>"_on_"<target_function_space>, as appropriate.

      Here <quadrature_arg_name> is the name of the corresponding quadrature object being passed to the Invoke. dimension is 1 or 3 and depends upon the function space (see Supported Function Spaces above for more information) and whether or not it is a basis or a differential basis function (see the table below). number_of_dofs is the number of degrees of freedom (DoFs) associated with the function space and np_* are the number of points to be evaluated: i) *_xyz in all directions (3D); ii) *_xy in the horizontal plane (2D); iii) *_x, *_y in the horizontal (1D); and iv) *_z in the vertical (1D). nfaces and nedges are the number of horizontal faces/edges obtained from the appropriate quadrature object supplied to the Invoke.

      Function Type

      Dimension

      Function Space Name

      Basis

      1

      W0, W2trace, W2Htrace, W2Vtrace, W3, Wtheta, Wchi

      3

      W1, W2, W2H, W2V, W2broken, ANY_W2

      Differential Basis

      1

      W2, W2H, W2V, W2broken, ANY_W2

      3

      W0, W1, W2trace, W2Htrace, W2Vtrace, W3, Wtheta, Wchi

  5. If either the normals_to_horizontal_faces or outward_normals_to_horizontal_faces properties of the reference element are required then pass the number of horizontal faces of the reference element (nfaces_re_h). Similarly, if either the normals_to_vertical_faces or outward_normals_to_vertical_faces are required then pass the number of vertical faces (nfaces_re_v). This also holds for the normals_to_faces and outward_normals_to_faces where the number of all faces of the reference element (nfaces_re) is passed to the kernel. (All of these quantities are integers of kind i_def.) Then, in the order specified in the meta_reference_element metadata:

    1. For the normals_to_horizontal/vertical_faces, pass a rank-2 integer array of kind i_def with dimensions (3, nfaces_re_h/v).

    2. For the outward_normals_to_horizontal/vertical_faces, pass a rank-2 integer array of kind i_def with dimensions (3, nfaces_re_h/v).

    3. For normals_to_faces or outward_normals_to_faces pass a rank-2 integer array of kind i_def with dimensions (3, nfaces_re).

  6. If the adjacent_face mesh property is required then:

    1. If the number of horizontal cell faces obtained from the reference element (nfaces_re_h) is not already being passed to the kernel (due to rule 5 above) then supply it here. This is an integer of kind i_def.

    2. Pass a rank-1, integer array of kind i_def and extent nfaces_re_h.

  7. If Quadrature is required (gh_shape = gh_quadrature_*) then, for each shape in the order specified in the gh_shape metadata:

    1. Include integer, scalar arguments of kind i_def with intent in that specify the extent of the basis/diff-basis arrays:

      1. If gh_shape is gh_quadrature_XYoZ then pass np_xy_<quadrature_arg_name> and np_z_<quadrature_arg_name>.

      2. If gh_shape is gh_quadrature_face/_edge then pass nfaces/nedges_<quadrature_arg_name> and np_xyz_<quadrature_arg_name>.

    2. Include weights which are real arrays of kind r_def:

      1. If gh_quadrature_XYoZ pass in weights_xz_<quadrature_arg_name> (rank one, extent np_xy_<quadrature_arg_name>) and weights_z_<quadrature_arg_name> (rank one, extent np_z_<quadrature_arg_name>).

      2. If gh_quadrature_face/_edge pass in weights_xyz_<quadrature_arg_name> (rank two with extents [np_xyz_<quadrature_arg_name>, nfaces/nedges_<quadrature_arg_name>]).

Examples

For instance, if a kernel has only one written argument and requires an evaluator then its metadata might be:

type, extends(kernel_type) :: testkern_operator_type
   type(arg_type), dimension(2) :: meta_args =               &
        (/ arg_type(gh_operator, gh_real, gh_write, w0, w1), &
           arg_type(gh_field*3,  gh_real, gh_read,  w0) /)
   type(func_type) :: meta_funcs(1) =                        &
        (/ func_type(w0, gh_basis) /)
   integer :: operates_on = cell_column
   integer :: gh_shape = gh_evaluator
 contains
   procedure, nopass :: code => testkern_operator_code
end type testkern_operator_type

then we only pass the basis functions evaluated on W0 (the space of the written kernel argument). The subroutine arguments will therefore be:

subroutine testkern_operator_code(cell, nlayers, ncell_3d,        &
     local_stencil, xdata, ydata, zdata, ndf_w0, undf_w0, map_w0, &
     basis_w0_on_w0, ndf_w1)

where local_stencil is the operator, xdata, ydata etc. are the three components of the field vector and map_w0 is the dofmap for the W0 function space.

If instead, gh_evaluator_targets is specified in the metadata:

type, extends(kernel_type) :: testkern_operator_type
   type(arg_type), dimension(2) :: meta_args =               &
        (/ arg_type(gh_operator, gh_real, gh_write, w0, w1), &
           arg_type(gh_field*3,  gh_real, gh_read,  w0) /)
   type(func_type) :: meta_funcs(1) =               &
        (/ func_type(w0, gh_basis) /)
   integer :: operates_on = cell_column
   integer :: gh_shape = gh_evaluator
   integer :: gh_evaluator_targets(2) = (/W0, W1/)
 contains
   procedure, nopass :: code => testkern_operator_code
end type testkern_operator_type

then we will need to pass two sets of basis functions (evaluated at W0 and at W1):

subroutine testkern_operator_code(cell, nlayers, ncell_3d,        &
     local_stencil, xdata, ydata, zdata, ndf_w0, undf_w0, map_w0, &
     basis_w0_on_w0, basis_w0_on_w1, ndf_w1)

If the metadata specifies that a kernel requires both an evaluator and quadrature:

type, extends(kernel_type) :: testkern_operator_type
   type(arg_type), dimension(2) :: meta_args =               &
        (/ arg_type(gh_operator, gh_real, gh_write, w0, w1), &
           arg_type(gh_field*3,  gh_real, gh_read,  w0) /)
   type(func_type) :: meta_funcs(1) =                        &
        (/ func_type(w0, gh_basis) /)
   integer :: operates_on = cell_column
   integer :: gh_shape(2) = (/ gh_evaluator, gh_quadrature_face /)
 contains
   procedure, nopass :: code => testkern_operator_code
end type testkern_operator_type

then we will need to pass basis functions for both the evaluator and the quadrature (where qr_face is the name of the face-quadrature object passed to the Invoke):

subroutine testkern_operator_code(cell, nlayers, ncell_3d,              &
     local_stencil, xdata, ydata, zdata, ndf_w0, undf_w0, map_w0,       &
     basis_w0_on_w0, basis_w0_qr_face, ndf_w1,                          &
     np_xyz_qr_face, nfaces_qr_face, weights_xyz_qr_face)

If the metadata specifies that the kernel requires a property of the reference element:

type, extends(kernel_type) :: testkern_operator_type
   type(arg_type), dimension(2) :: meta_args =               &
        (/ arg_type(gh_operator, gh_real, gh_write, w0, w1), &
           arg_type(gh_field*3,  gh_real, gh_read,  w0) /)
   type(reference_element_data_type) :: meta_reference_element(1) =  &
        (/ reference_element_data_type(normals_to_horizontal_faces) /)
   integer :: operates_on = cell_column
 contains
   procedure, nopass :: code => testkern_operator_code
end type testkern_operator_type

then the kernel must be passed the number of faces of the reference element and the array of face normals in the specified direction (here horizontal):

subroutine testkern_operator_code(cell, nlayers, ncell_3d,        &
     local_stencil, xdata, ydata, zdata, ndf_w0, undf_w0, map_w0, &
     nfaces_re_h, normals_face_h)

Rules for CMA Kernels

Kernels involving CMA operators are restricted to just three types; assembly, application/inverse-application and matrix-matrix. We give the rules for each of these in the sections below.

Assembly

An assembly kernel requires the column-banded dofmap for both the to- and from-function spaces of the CMA operator being assembled as well as the number of DoFs for each of the dofmaps. The full set of rules is:

  1. Include the cell argument. cell is an integer of kind i_def``and has intent ``in.

  2. Include nlayers, the number of layers in a column. nlayers is an integer of kind i_def and has intent in.

  3. Include the total number of cells in the 2D mesh (including halos), ncell_2d, which is an integer of kind i_def with intent in.

  4. Include the total number of cells, ncell_3d, which is an integer of kind i_def with intent in.

  5. For each argument in the meta_args metadata array:

    1. If it is a LMA operator, include a real, 3-dimensional array. The first dimension is ncell_3d. The second and third dimension are the local degrees of freedom for the to and from spaces, respectively. The precision of the array depends on how it is defined in the algorithm layer, see the Mixed Precision section for more details;

    2. If it is a CMA operator, include a real, 3-dimensional array of kind r_solver. The first dimension is "bandwidth_"<operator_name>, the second is "nrow_"<operator_name>, and the third is ncell_2d.

      1. Include the number of rows in the banded matrix. This is an integer of kind i_def with intent in and is named as "nrow_"<operator_name>.

      2. If the from-space of the operator is not the same as the to-space then include the number of columns in the banded matrix. This is an integer of kind i_def with intent in and is named as "ncol_"<operator_name>.

      3. Include the bandwidth of the banded matrix. This is an integer of kind i_def with intent in and is named as "bandwidth_"<operator_name>.

      4. Include banded-matrix parameter alpha. This is an integer of kind i_def with intent in and is named as "alpha_"<operator_name>.

      5. Include banded-matrix parameter beta. This is an integer of kind i_def with intent in and is named as "beta_"<operator_name>.

      6. Include banded-matrix parameter gamma_m. This is an integer of kind i_def with intent in and is named as "gamma_m_"<operator_name>.

      7. Include banded-matrix parameter gamma_p. This is an integer of kind i_def with intent in and is named as "gamma_p_"<operator_name>.

    3. If it is a field or scalar argument then include arguments following the same rules as for general-purpose kernels.

  6. For each unique function space in the order they appear in the metadata arguments (the to function space of an operator is considered to be before the from function space of the same operator as it appears first in lexicographic order):

    1. Include the number of degrees of freedom per cell for the space. This is an integer of kind i_def with intent in. The name of this argument is "ndf_"<arg_function_space>.

    2. If there is a field on this space then:

      1. Include the unique number of degrees of freedom for the function space. This is an integer of kind i_def and has intent in. The name of this argument is "undf_"<field_function_space>.

      2. Include the dofmap for this space. This is an integer array of kind i_def with intent in. It has one dimension sized by the local degrees of freedom for the function space.

    3. If the CMA operator has this space as its to/from space then include the column-banded dofmap, the list of offsets for the to/from-space. This is an integer array of rank 2 and kind i_def. The first dimension is "ndf_"<arg_function_space> and the second is nlayers.

Application/Inverse-Application

A kernel applying a CMA operator requires the column-indirection dofmap for both the to- and from-function spaces of the CMA operator. Since it does not have any LMA operator arguments it does not require the ncell_3d and nlayers scalar arguments. (Since a column-wise operator is, by definition, assembled for a whole column, there is no loop over levels when applying it.) The full set of rules is then:

  1. Include the cell argument. cell is an integer of kind i_def and has intent in.

  2. Include the total number of cells in the 2D mesh (including halos), ncell_2d, which is an integer of kind i_def with intent in.

  3. For each argument in the meta_args metadata array:

    1. If it is a field, include the field array. This is a real array of rank 1. Its precision (kind) depends on how it is defined in the algorithm layer, see the Mixed Precision. The field array name is currently specified as being "field_"<argument_position>"_"<field_function_space>. The extent of the array is the number of unique degrees of freedom for the function space that the field is on. This value is passed in separately. The intent of the argument is determined from the metadata (see meta_args);

    2. If it is a CMA operator, include it and its associated parameters (see Rule 5 of CMA Assembly kernels).

  4. For each of the unique function spaces encountered in the metadata arguments (the to function space of an operator is considered to be before the from function space of the same operator as it appears first in lexicographic order):

    1. Include the number of degrees of freedom per cell for the associated function space. This is an integer of kind i_def with intent in. The name of this argument is "ndf_"<field_function_space>;

    2. Include the number of unique degrees of freedom for the associated function space. This is an integer of kind i_def with intent in. The name of this argument is "undf_"<field_function_space>;

    3. Include the dofmap for this function space. This is a rank-1 integer array of kind i_def with extent equal to the number of degrees of freedom of the space ("ndf_"<field_function_space>).

  5. Include the indirection map for the to-space of the CMA operator. This is a rank-1 integer array of kind i_def with extent nrow.

  6. If the from-space of the operator is not the same as the to-space then include the indirection map for the from-space of the CMA operator. This is a rank-1 integer array of kind i_def with extent ncol.

Matrix-Matrix

Does not require any dofmaps and also does not require the nlayers and ncell_3d scalar arguments. The full set of rules are then:

  1. Include the cell argument. cell is an integer of kind i_def and has intent in.

  2. Include the total number of cells in the 2D mesh (including halos), ncell_2d, which is an integer of kind i_def with intent in.

  3. For each CMA operator or scalar argument specified in metadata:

    1. If it is a CMA operator, include it and its associated parameters (see Rule 5 of CMA Assembly kernels);

    2. If it is a scalar argument include the corresponding Fortran variable in the argument list with intent in.

Rules for Inter-Grid Kernels

As already specified, inter-grid kernels are only permitted to take fields and/or field-vectors as arguments. Fields (and field-vectors) that are on different meshes must be on different function spaces. Fields on the same mesh must also be on the same function space.

Argument ordering follows the general pattern used for ‘normal’ kernels with field data being followed by dofmap data. The rules for arguments to inter-grid kernels are as follows:

  1. Include nlayers, the number of layers in a column. nlayers is an integer of kind i_def and has intent in.

  2. Include the cell_map for the current cell (column). This is an integer array of rank two, kind i_def and intent in which provides the mapping from the coarse to the fine mesh. It has extent (ncell_f_per_c_x, ncell_f_per_c_y).

  3. Include ncell_f_per_c_x, and ncell_f_per_c_y, the numbers of fine cells per coarse cell in the x and y directions, respectively. These are integers of kind i_def and have intent in.

  4. Include ncell_f, the number of cells (columns) in the fine mesh. This is an integer of kind i_def and has intent in.

  5. For each argument in the meta_args metadata array (which must be a field or field-vector):

    1. Pass in field data as done for a regular kernel.

  6. For each unique function space (of which there will currently be two) in the order in which they are encountered in the meta_args metadata array, include dofmap information:

    If the dofmap is associated with an argument on the fine mesh:

    1. Include ndf_fine, the number of DoFs per cell for the FS of the field on the fine mesh;

    2. Include undf_fine, the number of unique DoFs per cell for the FS of the field on the fine mesh;

    3. Include dofmap_fine, the whole dofmap for the fine mesh. This is an integer array of rank two and kind i_def with intent in. The extent of the first dimension is ndf_fine and that of the second is ncell_f.

    else, the dofmap is associated with an argument on the coarse mesh:

    1. Include undf_coarse, the number of unique DoFs for the coarse field. This is an integer of kind i_def with intent in;

    2. Include dofmap_coarse, the dofmap for the current cell (column) in the coarse mesh. This is an integer array of rank one, kind i_def``and has intent ``in.

Rules for Domain Kernels

The rules for kernels that have operates_on = DOMAIN are almost identical to those for general-purpose kernels (described above), allowing for the fact that they are not permitted any type of operator argument or any argument with a stencil access. The only difference is that, since the kernel operates on the whole domain, the number of columns in the mesh excluding those in the halo (ncell_2d_no_halos), must be passed in. This is provided as the second argument to the kernel (after nlayers). ncell_2d_no_halos is an integer of kind i_def with intent in.

Rules for DoF Kernels

The rules for kernels that have operates_on = DOF are similar to those for general-purpose kernels but, due to the restriction that only fields and scalars can be passed to them, are much fewer. The full set of rules, along with PSyclone’s naming conventions, are:

  1. Include df, the index of the single dof to be operated on. This is an integer of of kind i_def with intent in.

  2. For each scalar/field in the order specified by the meta_args metadata:

    1. If the current entry is a scalar quantity then include the Fortran variable in the argument list. The intent is determined from the metadata (see meta_args for an explanation).

    2. If the current entry is a field then include the field array. The field array name is currently specified as being "field_" <argument_position>. A field array is a rank-1, real array with extent equal to the number of unique degrees of freedom for the space that the field is on. Its precision (kind) depends on how it is defined in the algorithm layer, see the Mixed Precision section for more details. This value is passed in separately. Again, the intent is determined from the metadata (see meta_args).

Argument Intents

As described above, LFRic kernels read and/or update the data pointed to by objects such as fields or operators. This data is passed to the kernels as subroutine arguments and their Fortran intents usually follow the logic determined by their access modes.

  • GH_READ indicates intent(in) as the argument is only ever read from.

  • GH_WRITE (for discontinuous function spaces) indicates that the argument is only written to in a kernel. The field and operator arguments’ data in LFRic are always defined outside of a kernel so the argument intent for this access type is intent(inout).

  • GH_INC, GH_READINC and GH_READWRITE indicate intent(inout) as the arguments are updated (albeit in a different way due to different access to DoFs, see meta_args for more details).

Kernel Naming Conventions

LFRic development uses strict naming conventions related to kernels. While they are not a requirement for PSyclone itself, any LFRic development should follow these conventions (see e.g. LFRic examples in PSyclone):

Module name:

<base_name>_kernel_mod

Kernel type name:

<base_name>_kernel_type

Subroutine name:

<base_name>_code

The latest version of the LFRic coding style guidelines are availabe in this LFRic wiki page (requires login access to MOSRS, see the above introduction to the LFRic API).

Built-ins

The basic concept of a PSyclone Built-in is described in the Built-ins section. In the LFRic API, calls to Built-ins generally follow a convention that the field/scalar written to comes first in the argument list. LFRic Built-ins must conform to the following rules:

  1. They must have one and only one modified (i.e. written to) argument.

  2. They must operate on a DoF (operates_on = DOF metadata).

  3. There must be at least one field in the argument list. This is so that we know the number of DoFs to iterate over in the PSy layer.

  4. Kernel arguments must be either fields or scalars (real- and/or integer-valued).

  5. All field arguments to a given Built-in must be on the same function space. This is because all current Built-ins operate on DoFs and therefore all fields should have the same number. It also means that we can determine the number of DoFs uniquely when a scalar is written to;

  6. Built-ins that update real-valued fields can, in general, only read from other real-valued fields, but they can take both real and integer scalar arguments (see rule 8 for exceptions);

  7. Built-ins that update integer-valued fields can, in general, only read from other integer-valued fields and take integer scalar arguments (see rule 8 for exceptions);

  8. The only two exceptions from the rules 6) and 7) above regarding the same data type of “write” and “read” field arguments are Built-ins that convert field data from real to integer, real_to_int_X, and from integer to real, int_to_real_X.

The Built-ins supported for the LFRic API are listed in the related subsections, grouped first by the data type of fields they operate on (real-valued and integer-valued) and then by the mathematical operation they perform.

The field arguments in Built-ins are the derived types that represent the LFRic fields, however mathematical operations are actually performed on the data of the field proxies (e.g. field1_proxy%data(:)). For instance, X_plus_Y Built-in adds the values of two fields accessed via their proxies in a loop over DoFs:

DO df=loop0_start,loop0_stop
   field3_proxy%data(df) = field1_proxy%data(df) + field2_proxy%data(df)

where the precise values of the loop limits depend on the use of distributed memory, annexed DoFs or both.

As described in the PSy-layer Argument Intents section, the Fortran intent of LFRic field objects is always in (because it is only the data pointed to from within the object that is modified). The field or scalar that has its data modified by a Built-in is marked in bold.

For clarity, the calculation performed by each Built-in is described using Fortran array syntax without the details about field proxies. The actual implementation of the Built-in may change in future (e.g. it could be implemented by PSyclone generating a call to an optimised Maths library).

Metadata

The code below outlines the elements of the LFRic API Built-in metadata for the Built-ins that update a real-valued field, 1) ‘meta_args’, 2) ‘operates_on’ and 3) ‘procedure’:

type, public, extends(kernel_type) :: aX_plus_bY
   private
   type(arg_type) :: meta_args(5) = (/                              &
        arg_type(GH_FIELD,  GH_REAL  GH_WRITE, ANY_SPACE_1),        &
        arg_type(GH_SCALAR, GH_REAL, GH_READ              ),        &
        arg_type(GH_FIELD,  GH_REAL, GH_READ,  ANY_SPACE_1),        &
        arg_type(GH_SCALAR, GH_REAL, GH_READ              ),        &
        arg_type(GH_FIELD,  GH_REAL  GH_READ,  ANY_SPACE_1)         &
        /)
   integer :: operates_on = DOF
 contains
   procedure, nopass :: aX_plus_bY_code
end type aX_plus_bY

As can be seen, the metadata for a Built-in kernel is a subset of that for a user-defined Kernel with the exception that operates_on must be DOF instead of CELL_COLUMN.

The metadata for the LFRic Built-ins that update an integer-valued field is similar:

!> ifield3 = ifield1 + ifield2
type, public, extends(kernel_type) :: int_X_plus_Y
   private
   type(arg_type) :: meta_args(3) = (/                              &
        arg_type(GH_FIELD, GH_INTEGER, GH_WRITE, ANY_SPACE_1),      &
        arg_type(GH_FIELD, GH_INTEGER, GH_READ,  ANY_SPACE_1),      &
        arg_type(GH_FIELD, GH_INTEGER, GH_READ,  ANY_SPACE_1)       &
        /)
   integer :: operates_on = DOF
 contains
   procedure, nopass :: int_X_plus_Y_code
end type int_X_plus_Y

Valid Data Types and Access Modes

The allowed data types and accesses for arguments in LFRic Built-in kernels are a bit different than for the user-defined Kernels and are listed in the table below.

Argument Type

Data Type

Function Space

Access Type

GH_SCALAR

GH_INTEGER

n/a

GH_READ

GH_SCALAR

GH_REAL

n/a

GH_READ, GH_SUM

GH_FIELD

GH_REAL, GH_INTEGER

ANY_SPACE_<n>

GH_READ, GH_WRITE, GH_READWRITE

Note

Since the LFRic infrastructure does not currently support integer reductions, integer scalar arguments in Built-ins are restricted to having read-only access. Also, logical scalar arguments are not permitted.

Naming scheme

The supported Built-ins in the LFRic API are named according to the scheme presented below. Any new Built-in needs to comply with these rules.

  1. Ordering of arguments in Built-ins calls follows LHS (result) <- RHS (operation on arguments) direction, except where a Built-in returns the LHS result to one of the RHS arguments. In that case ordering of arguments remains as in the RHS expression, with the returning RHS argument written as close to the LHS as it can be without affecting the mathematical expression.

  2. Field names begin with upper case in short form (e.g. X, Y, Z) and any case in long form (e.g. Field1, field).

  3. Scalar names begin with lower case: e.g. a, b, are scalar1, scalar2. Special names for scalars are: constant (or c), innprod (inner/scalar product of two fields) and sumfld (sum of a field).

  4. Arguments in Built-ins variable declarations and constructs (PSyclone Fortran and Python definitions):

    1. Are always written in long form and lower case (e.g. field1, field2, scalar1, scalar2);

    2. LHS result arguments are always listed first;

    3. RHS arguments are listed in order of appearance in the mathematical expression, except when one of them is the LHS result.

  5. Built-ins names in Fortran consist of:

    1. RHS arguments in short form (e.g. X, Y, a, b) only;

    2. Descriptive name of mathematical operation on RHS arguments in the form <operationname>_<RHSargs> or <RHSargs>_<operationname>_<RHSargs>;

    3. Prefix "inc_" where the result is returned to one of the RHS arguments (i.e. "inc_"<RHSargs>_<operationname>_<RHSargs>);

    4. Prefix "int_" for the Built-in operations on the integer-valued field arguments (i.e. "int_inc_"<RHSargs>_<operationname>_<RHSargs>).

  6. Built-ins names in Python definitions are similar to their Fortran counterparts, with a few differences:

    1. Operators and RHS arguments are all in upper case (e.g. X, Y, A, B, Plus, Minus);

    2. There are no underscores;

    1. Common suffix is "Kern";

    1. Common prefix is "LFRic" for the Built-in operations on the real-valued arguments and "LFRicInt" for the Built-in operations on the integer-valued fields.

Built-in operations on real-valued fields

As described above, Built-ins that operate on real-valued fields mandate GH_REAL as the kernel metadata for fields and scalars.

The precision of fields and scalars, however, is determined by the algorithm layer via precision variables as described in the Mixed Precision section (see subsections on fields and scalars).

For instance, field and scalar declarations for the aX_plus_Y Built-in that operates on r_solver_field_type and uses r_solver scalar will be:

real(kind=r_solver), intent(in) :: ascalar
type(r_solver_field_type), intent(in) :: zfield, xfield, yfield

Mixing precisions is not explicitly forbidden, so we may have e.g. X_divideby_a Built-in where:

real(kind=r_def), intent(in) :: ascalar
type(r_tran_field_type), intent(in) :: yfield, xfield

Certain Built-ins are currently restricted in the precision of the arguments that they accept. Those that calculate the inner product and sum of a field are restricted to r_def precision because the scalar global reductions in the LFRic infrastructure are currently only able to support field_type and hence have r_def precision. In addition, all integer arguments to Built-ins are currently restricted to i_def precision.

Addition

Built-ins that add (scaled) real-valued fields and return the result as a real-valued field are denoted with the keyword plus.

X_plus_Y

X_plus_Y (field3, field1, field2)

Sums two fields and stores the result in the third field (Z = X + Y):

field3(:) = field1(:) + field2(:)
inc_X_plus_Y

inc_X_plus_Y (field1, field2)

Adds the second field to the first and returns it (X = X + Y):

field1(:) = field1(:) + field2(:)
a_plus_X

a_plus_X (field2, rscalar, field1)

Adds a real scalar value to all elements of a field and stores the result in another field (Y = a + X):

field2(:) = rscalar + field1(:)
inc_a_plus_X

inc_a_plus_X (rscalar, field)

Adds a real scalar value to all elements of a field and returns the field (X = a + X):

field(:) = rscalar + field(:)
aX_plus_Y

aX_plus_Y (field3, rscalar, field1, field2)

Performs Z = aX + Y:

field3(:) = rscalar*field1(:) + field2(:)
inc_aX_plus_Y

inc_aX_plus_Y (rscalar, field1, field2)

Performs X = aX + Y (increments the first field):

field1(:) = rscalar*field1(:) + field2(:)
inc_X_plus_bY

inc_X_plus_bY (field1, rscalar, field2)

Performs X = X + bY (increments the first field):

field1(:) = field1(:) + rscalar*field2(:)
aX_plus_bY

aX_plus_bY (field3, rscalar1, field1, rscalar2, field2)

Performs Z = aX + bY:

field3(:) = rscalar1*field1(:) + rscalar2*field2(:)
inc_aX_plus_bY

inc_aX_plus_bY (rscalar1, field1, rscalar2, field2)

Performs X = aX + bY (increments the first field):

field1(:) = rscalar1*field1(:) + rscalar2*field2(:)
aX_plus_aY

aX_plus_aY (field3, rscalar, field1, field2)

Performs Z = aX + aY = a(X + Y):

field3(:) = rscalar*(field1(:) + field2(:))

Subtraction

Built-ins which subtract (scaled) real-valued fields and return the result as a real-valued field are denoted with the keyword minus.

X_minus_Y

X_minus_Y (field3, field1, field2)

Subtracts the second field from the first and returns the result in the third field (Z = X - Y):

field3(:) = field1(:) - field2(:)
inc_X_minus_Y

inc_X_minus_Y (field1, field2)

Subtracts the second field from the first and returns it (X = X - Y):

field1(:) = field1(:) - field2(:)
a_minus_X

a_minus_X (field2, rscalar, field1)

Subtracts all elements of a field from a real scalar value and stores the result in another field (Y = a - X):

field2(:) = rscalar - field1(:)
inc_a_minus_X

inc_a_minus_X (rscalar, field)

Subtracts all elements of a field from a real scalar value and returns the field (X = a - X):

field(:) = rscalar - field(:)
X_minus_a

X_minus_a (field2, field1, rscalar)

Subtracts a real scalar value from all elements of a field and stores the result in another field (Y = X - a):

field2(:) = field1(:) - rscalar
inc_X_minus_a

inc_X_minus_a (field, rscalar)

Subtracts a real scalar value from all elements of a field and returns the field (X = X - a):

field(:) = field(:) - rscalar
aX_minus_Y

aX_minus_Y (field3, rscalar, field1, field2)

Performs Z = aX - Y:

field3(:) = rscalar*field1(:) - field2(:)
X_minus_bY

X_minus_bY (field3, field1, rscalar, field2)

Performs Z = X - bY:

field3(:) = field1(:) - rscalar*field2(:)
inc_X_minus_bY

inc_X_minus_bY (field1, rscalar, field2)

Performs X = X - bY (decrements the first field):

field1(:) = field1(:) - rscalar*field2(:)
aX_minus_bY

aX_minus_bY (field3, rscalar1, field1, rscalar2, field2)

Performs Z = aX - bY:

field3(:) = rscalar1*field1(:) - rscalar2*field2(:)

Multiplication

Built-ins which multiply (scaled) real-valued fields and return the result as a real-valued field are denoted with the keyword times.

X_times_Y

X_times_Y (field3, field1, field2)

Multiplies two fields DoF by DoF and returns the result in a third field (Z = X*Y):

field3(:) = field1(:)*field2(:)
inc_X_times_Y

inc_X_times_Y (field1, field2)

Multiplies the first field by the second and returns it (X = X*Y):

field1(:) = field1(:)*field2(:)
inc_aX_times_Y

inc_aX_times_Y (rscalar, field1, field2)

Performs X = a*X*Y (increments the first field):

field1(:) = rscalar*field1(:)*field2(:)

Scaling

Built-ins which scale real-valued fields are technically cases of multiplying a real-valued field by a real scalar and are hence also denoted with the keyword times.

a_times_X

a_times_X (field2, rscalar, field1)

Multiplies a field by a real scalar value and stores the result in another field (Y = a*X):

field2(:) = rscalar*field1(:)
inc_a_times_X

inc_a_times_X (rscalar, field)

Multiplies a field by a real scalar value and returns the field (X = a*X):

field(:) = rscalar*field(:)

Division

Built-ins which divide real-valued fields and return the result as a real-valued field are denoted with the keyword divideby.

X_divideby_Y

X_divideby_Y (field3, field1, field2)

Divides the first field by the second field, DoF by DoF, and stores the result in the third field (Z = X/Y):

field3(:) = field1(:)/field2(:)
inc_X_divideby_Y

inc_X_divideby_Y (field1, field2)

Divides the first field by the second and returns it (X = X/Y):

field1(:) = field1(:)/field2(:)
X_divideby_a

X_divideby_a (field2, field1, rscalar)

Divides each field element by a real scalar value and stores the result in another field (Y = X/a):

field2(:) = field1(:)/rscalar
inc_X_divideby_a

inc_X_divideby_a (field, rscalar)

Divides each field element by a real scalar value and returns the field (X = X/a):

field(:) = field(:)/rscalar

Inverse scaling

Built-ins which perform inverse scaling of real-valued fields are also denoted with the keyword divideby as they divide a real scalar by elements of a real-valued field.

a_divideby_X

a_divideby_X (field2, rscalar, field1)

Divides a real scalar value by each field element and stores the result in another field (Y = a/X):

field2(:) = rscalar/field1(:)
inc_a_divideby_X

inc_a_divideby_X (rscalar, field)

Divides a real scalar value by each field element and returns the field (X = a/X):

field(:) = rscalar/field(:)

Setting to a value

Built-ins which set real-valued field elements to some real value are denoted with the keyword setval.

setval_c

setval_c (field, constant)

Sets all elements of a field field to a real scalar constant (X = c):

field(:) = constant
setval_X

setval_X (field2, field1)

Sets a field field2 equal (DoF per DoF) to another field field1 (Y = X):

field2(:) = field1(:)
setval_random

setval_random (field)

Fills all elements of a field field using a sequence of real, pseudo-random numbers in the interval 0 <= x < 1:

do df = 1, ndofs
  field(df) = RAND()
end do

where RAND() is some function that returns a new pseudo-random number each time it is called.

Warning

This Built-in is implemented using the Fortran random_number intrinsic. Therefore no guarantee is made as to the quality of the sequence of pseudo-random numbers, especially when running in parallel.

Raising to power

Built-ins which raise real-valued field elements to an exponent are denoted with the keyword powreal for a real exponent or powint for an integer exponent.

inc_X_powreal_a

inc_X_powreal_a (field, rscalar)

Raises a field to a real scalar value and returns the field (X = X**a):

field(:) = field(:)**rscalar
inc_X_powint_n

inc_X_powint_n (field, iscalar)

Raises a field to an integer scalar value and returns the field (X = X**n):

field(:) = field(:)**iscalar

where iscalar is an integer scalar of i_def precision.

Inner product

Built-ins which calculate the inner product of two real-valued fields or of a real-valued field with itself and return the result as a real scalar are denoted with the keyword innerproduct.

Note

When used with distributed memory these Built-ins will trigger the addition of a global sum which may affect the performance and/or scalability of the code. Also, whilst the fields in these Built-ins can be of any supported real precision, the only currently supported precision for the global reductions in the LFRic infrastructure is r_def, hence the result will be converted accordingly.

X_innerproduct_Y

X_innerproduct_Y (innprod, field1, field2)

Computes the inner product of two fields, field1 and field2, i.e.:

innprod = SUM(field1(:)*field2(:))

where innprod is a real scalar of r_def precision.

X_innerproduct_X

X_innerproduct_X (innprod, field)

Computes the inner product of the field field1 by itself, i.e.:

innprod = SUM(field(:)*field(:))

where innprod is a real scalar of r_def precision.

Sum of elements

A Built-in which sums the elements of a real-valued field and returns the result as a real scalar is denoted with the keyword sum.

Note

When used with distributed memory this Built-in will trigger the addition of a global sum which may affect the performance and/or scalability of the code. Also, whilst the fields in these Built-ins can be of any supported real precision, the only currently supported precision for the global reductions in the LFRic infrastructure is r_def, hence the result will be converted accordingly.

sum_X

sum_X (sumfld, field)

Sums all of the elements of the field field and returns the result in the real scalar variable sumfld:

sumfld = SUM(field(:))

where sumfld is a real scalar of r_def precision.

Sign of elements

A Built-in which returns the sign of a real-valued field is denoted with the keyword sign.

sign_X

sign_X (field2, rscalar, field1)

Returns the sign of a real-valued field, e.g. in Fortran: Y = sign(a, X). Here a is a real scalar and Y and X are real-valued fields. The results are a for X >= 0 and -a for X < 0:

field2(:) = SIGN(rscalar, field1(:))

DoF-wise maximum of elements

Built-ins which return the DoF-wise maximum of a real scalar and a real-valued field are denoted with the keyword max.

max_aX

max_aX (field2, rscalar, field1)

Returns maximum of rscalar and each element of the field field1 as the second field field2 (Y = max(a, X)):

field2(:) = MAX(rscalar, field1(:))
inc_max_aX

inc_max_aX (rscalar, field)

Returns maximum of rscalar and each element of the field field in the same field (X = max(a, X)):

field(:) = MAX(rscalar, field(:))

DoF-wise minimum of elements

Built-ins which return the DoF-wise minimum of a real scalar and a real-valued field are denoted with the keyword min.

min_aX

min_aX (field2, rscalar, field1)

Returns minimum of rscalar and each element of the field field1 as the second field field2 (Y = min(a, X)):

field2(:) = MIN(rscalar, field1(:))
inc_min_aX

inc_min_aX (rscalar, field)

Returns minimum of rscalar and each element of the field field in the same field (X = min(a, X)):

field(:) = MIN(rscalar, field(:))

Conversion of real field elements

Built-ins which take a real field for conversion to a field of a different datatype or precision are denoted by the datatype that the input real field will be converted to. A Built-in that converts a real to an integer field is denoted by the phrase to_int. Likewise, a Built-in that converts a real to a real field is denoted by the phrase to_real.

real_to_int_X

real_to_int_X (ifield2, field1)

Converts real-valued field elements to integer-valued field elements, e.g. in Fortran this would be: Y = INT(X, kind=i_<prec>). Here Y is an integer-valued field and X is the real-valued field being converted:

ifield2(:) = INT(field1(:), kind=i_<prec>)

where ifield2 is currently the only supported integer-valued field type in LFRic (integer_field_type of i_def precision) and a real -valued field field1 can be of any supported precisions for GH_REAL fields (e.g. r_tran for r_tran_field_type).

real_to_real_X

real_to_real_X (field2, field1)

Converts real-valued field elements from a precision r_<prec> to real-valued field elements of a differing precision r_<prec>, e.g. in Fortran this would be: Y = REAL(X, kind=r_<prec>). Here Y and X are both real-valued fields, with X being converted to the precision of Y:

field2(:) = REAL(field1(:), kind=r_<prec>)

field2 and field1 are real-valued fields of any supported precisions for GH_REAL fields (e.g. r_tran for r_tran_field_type).

Built-in operations on integer-valued fields

The number of supported Built-in operations on the integer-valued fields is not as large as for their real counterparts as not all mathematical operations on integer-valued fields make sense.

As described above, Built-ins that operate on integer-valued fields mandate GH_INTEGER as the kernel metadata for fields and scalars. Both integer scalar arguments and integer-valued fields can only currently have i_def precision, as described in the Mixed Precision section.

For instance, field and scalar declarations for the X_minus_a Built-in will be:

integer(kind=i_def), intent(in) :: ascalar
type(integer_field_type), intent(in) :: yfield, xfield

Addition

Built-ins that add integer-valued fields and return the result as an integer-valued field are denoted with the keyword plus and the prefix int.

int_X_plus_Y

int_X_plus_Y (ifield3, ifield1, ifield2)

Sums two fields and stores the result in the third field (Z = X + Y):

ifield3(:) = ifield1(:) + ifield2(:)
int_inc_X_plus_Y

int_inc_X_plus_Y (ifield1, ifield2)

Adds the second field to the first and returns it (X = X + Y):

ifield1(:) = ifield1(:) + ifield2(:)
int_a_plus_X

int_a_plus_X (ifield2, iscalar, ifield1)

Adds an integer scalar value to all elements of a field and stores the result in another field (Y = a + X):

ifield2(:) = iscalar + ifield1(:)
int_inc_a_plus_X

int_inc_a_plus_X (iscalar, ifield)

Adds an integer scalar value to all elements of a field and returns the field (X = a + X):

ifield(:) = iscalar + ifield(:)

Subtraction

Built-ins which subtract integer-valued fields and return the result as an integer-valued field are denoted with the keyword minus and the prefix int.

int_X_minus_Y

int_X_minus_Y (ifield3, ifield1, ifield2)

Subtracts the second field from the first and returns the result in the third field (Z = X - Y):

ifield3(:) = ifield1(:) - ifield2(:)
int_inc_X_minus_Y

int_inc_X_minus_Y (ifield1, ifield2)

Subtracts the second field from the first and returns it (X = X - Y):

ifield1(:) = ifield1(:) - ifield2(:)
int_a_minus_X

int_a_minus_X (ifield2, iscalar, ifield1)

Subtracts all elements of a field from an integer scalar value and stores the result in another field (Y = a - X):

ifield2(:) = iscalar - ifield1(:)
int_inc_a_minus_X

int_inc_a_minus_X (iscalar, ifield)

Subtracts all elements of a field from an integer scalar value and returns the field (X = a - X):

ifield(:) = iscalar - ifield(:)
int_X_minus_a

int_X_minus_a (ifield2, ifield1, iscalar)

Subtracts an integer scalar value from all elements of a field and stores the result in another field (Y = X - a):

ifield2(:) =  ifield1(:) - iscalar
int_inc_X_minus_a

int_inc_X_minus_a (ifield, iscalar)

Subtracts an integer scalar value from all elements of a field and returns the field (X = X - a):

ifield(:) =  ifield(:) - iscalar

Multiplication

Built-ins which multiply integer-valued fields and return the result as an integer-valued field are denoted with the keyword times and the prefix int.

int_X_times_Y

int_X_times_Y (ifield3, ifield1, ifield2)

Multiplies two fields DoF by DoF and returns the result in a third field (Z = X*Y):

ifield3(:) = ifield1(:)*ifield2(:)
int_inc_X_times_Y

int_inc_X_times_Y (ifield1, ifield2)

Multiplies the first field by the second and returns it (X = X*Y):

ifield1(:) = ifield1(:)*ifield2(:)

Scaling

Built-ins which scale integer-valued fields are denoted with the keyword times and prefixed by the keyword int.

int_a_times_X

int_a_times_X (ifield2, iscalar, ifield1)

Multiplies a field by an integer scalar and stores the result in another field (Y = a*X):

ifield2(:) = iscalar*ifield1(:)
int_inc_a_times_X

int_inc_a_times_X (iscalar, ifield)

Multiplies a field by an integer scalar value and returns the field (X = a*X):

ifield(:) = iscalar*ifield(:)

Setting to a value

Built-ins which set integer-valued field elements to some integer value are denoted with the keyword setval and the prefix int.

int_setval_c

int_setval_c (ifield, constant)

Sets all elements of a field ifield to an integer scalar constant (X = c):

ifield(:) = constant
int_setval_X

int_setval_X (ifield2, ifield1)

Sets a field ifield2 equal (DoF per DoF) to another field ifield1 (Y = X):

ifield2(:) = ifield1(:)

Sign of elements

A Built-in which returns the sign of an integer-valued field is denoted with the keyword sign and the prefix int.

int_sign_X

int_sign_X (ifield2, iscalar, ifield1)

Returns the sign of an integer-valued field, e.g. in Fortran: Y = sign(a, X). Here a is an integer scalar and Y and X are integer-valued fields. The results are a for X >= 0 and -a for a < 0:

ifield2(:) = SIGN(iscalar, ifield1(:))

DoF-wise maximum of elements

Built-ins which return the DoF-wise maximum of an integer scalar and an integer-valued field are denoted with the keyword max.

int_max_aX

int_max_aX (ifield2, iscalar, ifield1)

Returns maximum of iscalar and each element of the field ifield1 as the second field ifield2 (Y = max(a, X)):

ifield2(:) = MAX(iscalar, ifield1(:))
int_inc_max_aX

int_inc_max_aX (iscalar, ifield)

Returns maximum of iscalar and each element of the field ifield in the same field (X = max(a, X)):

ifield(:) = MAX(iscalar, ifield(:))

DoF-wise minimum of elements

Built-ins which return the DoF-wise minimum of an integer scalar and an integer-valued field are denoted with the keyword min.

int_min_aX

int_min_aX (ifield2, iscalar, ifield1)

Returns minimum of iscalar and each element of the field ifield1 as the second field ifield2 (Y = min(a, X)):

ifield2(:) = MIN(iscalar, ifield1(:))
int_inc_min_aX

int_inc_min_aX (iscalar, ifield)

Returns minimum of iscalar and each element of the field ifield in the same field (X = min(a, X)):

ifield(:) = MIN(iscalar, ifield(:))

Conversion of integer to real field elements

A Built-in which takes an integer field and converts it to a real field is denoted by the phrase to_real.

int_to_real_X

int_to_real_X (field2, ifield1)

Converts integer-valued field elements to real-valued field elements, e.g. in Fortran this would be Y = REAL(X, kind=r_<prec>). Here Y is a real-valued field and X is the integer-valued field being converted:

field2(:) = REAL(ifield1(:), kind=r_<prec>)

where ifield1 is currently the only supported integer-valued field type in LFRic (integer_field_type of i_def precision). The real-valued field1 can be of any supported precisions for GH_REAL fields, hence r_<prec> is determined from the algorithm layer (e.g. r_solver for r_solver_field_type).

Boundary Conditions

In the LFRic API, boundary conditions for a field or LMA operator can be enforced by the algorithm developer by calling the Kernels enforce_bc_type or enforce_operator_bc_type, respectively. These kernels take a field or operator as input and apply boundary conditions. For example:

call invoke( kernel_type(field1, field2),      &
             enforce_bc_type(field1),          &
             kernel_with_op_type(field1, op1), &
             enforce_operator_bc_type(op1)     &
           )

The particular boundary conditions that are applied are not known by PSyclone, PSyclone simply recognises these kernels by their names and passes pre-specified dofmap and boundary_value arrays into the kernel implementations, the contents of which are set by the LFRic infrastructure.

Up to and including version 1.4.0 of PSyclone, boundary conditions were applied automatically after a call to matrix_vector_type if the field arguments were on a vector function space (one of W1, W2, W2H, W2V or W2broken). With the subsequent introduction of the ability to apply boundary conditions to operators this functionality is no longer required and has been removed.

Example eg4 in the examples/lfric directory includes a call to enforce_bc_kernel_type so can be used to see the boundary condition code that is added by PSyclone. See the README in the examples/lfric directory for instructions on how to run this example.

An example of applying boundary conditions to an operator is the kernel enforce_operator_bc_kernel_mod.F90 in the <PSYCLONEHOME>/src/psyclone/tests/test_files/dynamo0p3 directory. Since operators are discontinuous quantities, updating their values can be safely performed in parallel (see Section Kernel). The GH_READWRITE access is used for updating discontinuous operators (see subsection Valid Access Modes for more details).

Conventions

The naming of LFRic API kernels and associated entities (types, subroutines and modules) follows the PSyclone Fortran naming conventions (see Fortran Naming Conventions). However, PSyclone does not need this convention to be followed apart from the stub generator (see the Kernel-stub Generator Section ) where the name of the metadata to be parsed is determined from the module name.

The contents of the metadata is also usually declared private but this does not affect PSyclone.

Finally, the procedure metadata (located within the kernel metadata) usually has nopass specified but again this is ignored by PSyclone.

Configuration

The general and the LFRic-API-specific configuration options are described in the Configuration section.

Annexed DoFs

When a kernel operates on DoFs (rather than cell-columns) for a continuous field using distributed memory, PSyclone need only ensure that DoFs owned by a processor are computed. However, for continuous fields, shared DoFs at the boundary between processors must be replicated (as different cells share the same DoF). Only one processor can own a DoF, therefore processors will have continuous fields which contain DoFs that the processor does not own. These unowned DoFs are called annexed in the LFRic API and are a separate, but related, concept to field halos.

When a kernel that operates on a cell-column needs to read a continuous field then the annexed DoFs must be up-to-date on all processors. If they are not then a halo exchange must be added. Currently PSyclone defaults, for kernels which iterate over DoFs, to iterating over only owned DoFs. This behaviour can be changed by setting COMPUTE_ANNEXED_DOFS to true in the lfric section of the configuration file (see the Configuration section). PSyclone will then generate code to iterate over both owned and annexed DoFs, thereby reducing the number of halo exchanges required (at the expense of redundantly computing annexed DoFs). For more details please refer to the lfric-developers developers section.

Run-time Checks

PSyclone performs static consistency checks where possible. When this is not possible PSyclone can generate run-time checks. As there may be performance costs associated with run-time checks they may be switched on or off by the RUN_TIME_CHECKS option in the configuration file.

Currently run-time checks can be generated to:

  1. Check that a field with a read-only function space (see section Read-Only Function Spaces) is not modified by a kernel. This is enforced by checking that all fields that are marked (in kernel metadata) as being updated by a kernel are not on a read-only function space. A second check that is required for fields on read-only function spaces is to ensure that the halo is clean before it is accessed. This check is currently implemented within the LFRic infrastructure halo exchange call (that the PSyclone LFRic API places at appropriate locations). If the halo is clean then the halo exchange will not be called. However, if the halo is not clean then the resulting halo exchange call will cause the infrastructure to raise an error (because the field is on a read-only space).

  2. Check that the function space of a field is consistent with the kernel function space metadata that the field’s data is passed into. For example, if kernel metadata specifies that a field is on the W2 function space then a run-time check is added to ensure that the field object passed into the PSy layer is indeed on that space. For more general kernel function space metadata, such as ANY_DISCONTINUOUS_SPACE_* then a run-time check is added to ensure that the field is on one of the discontinuous function spaces supported in the LFRic API.

Supported Data Types and Default Kind

The LFRic API supports three Fortran primitive (intrinsic) data types, real, integer and logical (listed in the supported_fortran_datatypes section of the PSyclone configuration file). All three data types are used for scalars. Fields and field vectors are allowed to have real and integer data. Operators and column-wise operators are only allowed to have real data. These supported primitive types are linked to the respective kernel data type metadata descriptors, GH_REAL and GH_INTEGER.

The default kind (precision) for these supported data types is set to r_def, i_def and l_def, respectively, in the default_kind dictionary in the configuration file. These default values are defined in the LFRic infrastructure code.

Note

Whilst the logical Fortran primitive (intrinsic) data type is supported in the LFRic API for scalar arguments, it is not yet available for fields and operators. This will be added as required in future releases.

Precision Map

This gives the amount of storage (in bytes) associated with a particular LFRic precision. The values for ‘r_tran’, ‘r_solver’, ‘r_def’, ‘r_bl’ and ‘r_phys’ are set within LFRic infrastructure according to CPP ifdefs. The values given in the configuration file are the defaults. ‘l_def’ is included in the dictionary so that it contains a complete record of the various precision symbols used in LFRic.

Note

Storing the precision map in the LFRic API within PSyclone is a temporary measure which will yield to the LFRic infrastructure as the single source of precisions, as discussed in PSyclone issue #1941.

Number of Generalised ANY_*_SPACE Function Spaces

As outlined in the meta_args and the Supported Function Spaces sections above, the number of generalised ANY_SPACE_<n> and ANY_DISCONTINUOUS_SPACE_<n> function spaces can be set in the PSyclone configuration file.

The relevant parameters are NUM_ANY_SPACE and NUM_ANY_DISCONTINUOUS_SPACE, respectively. Their default values in the configuration file are 10 and their allowed values are positive non-zero integers. PSyclone will raise a ConfigurationError if a supplied value is invalid.

Transformations

This section describes the LFRic API-specific transformations. In cases, excepting Dynamo0p3RedundantComputationTrans, Dynamo0p3AsyncHaloExchangeTrans and Dynamo0p3KernelConstTrans, these transformations are specialisations of generic transformations described in the Transformations section. The difference between these transformations and the generic ones is that these perform LFRic API-specific checks to make sure the transformations are valid. In practice these transformations perform the required checks then call the generic ones internally.

The use of the LFRic API-specific transformations is exactly the same as the equivalent generic ones in all cases excepting LFRicLoopFuseTrans. In this case an additional optional argument same_space can be set when applying the transformation. The reason for this is to allow loop fusion when one or more of the iteration spaces is determined by a function space that is unknown by PSyclone at compile time. This is the case when the ANY_SPACE_<n> function space is specified in the Kernel metadata. Adding {"same_space": True} as option when applying the transformation allows the user to specify that the spaces are the same (see Standard Functionality for using options in transformations). This option should therefore be used with caution. PSyclone will raise an error if same_space is used when at least one of the function spaces is not ANY_SPACE_<n> or both spaces are not the same. In general, PSyclone will not allow loop fusion if it does not know the spaces are the same. The exception are loops over discontinuous spaces (see Supported Function Spaces for list of discontinuous function spaces) for which loop fusion is allowed (unless the loop bounds become different due to a prior transformation).

The Dynamo0p3RedundantComputationTrans and Dynamo0p3AsyncHaloExchange transformations are only valid for the LFRic API. This is because this API is currently the only one that supports distributed memory. An example of redundant computation can be found in examples/lfric/eg8 and an example of asynchronous halo exchanges can be found in examples/lfric/eg11.

The Dynamo0p3KernelConstTrans transformation is only valid for the LFRic API. This is because the properties that it makes constant are API specific.

The LFRic API-specific transformations currently available are given below. Early transformations include “Dynamo0p3” or “Dynamo” in their name to indicate that these transformations are only valid for this particular API. More recent transformations typically include “LFRic” in their name to indicate the same restriction. However, more importantly, transformations that are specific to LFRic reside in the LFRic-specific “psyclone.domain/lfric/transformations” directory. Note, the early LFRic API-specific transformations have not yet been migrated to this directory.

Note

Only the loop-colouring and OpenMP transformations are currently supported for loops that contain inter-grid kernels. Attempting to apply other transformation types will result in PSyclone raising an error.

class psyclone.domain.lfric.transformations.LFRicExtractTrans[source]

LFRic API application of ExtractTrans transformation to extract code into a stand-alone program. For example:

>>> from psyclone.parse.algorithm import parse
>>> from psyclone.psyGen import PSyFactory
>>>
>>> API = "lfric"
>>> FILENAME = "solver_alg.x90"
>>> ast, invokeInfo = parse(FILENAME, api=API)
>>> psy = PSyFactory(API, distributed_memory=False).create(invoke_info)
>>> schedule = psy.invokes.get('invoke_0').schedule
>>>
>>> from psyclone.domain.lfric.transformations import LFRicExtractTrans
>>> etrans = LFRicExtractTrans()
>>>
>>> # Apply LFRicExtractTrans transformation to selected Nodes
>>> etrans.apply(schedule.children[0:3])
>>> print(schedule.view())
apply(nodes, options=None)[source]

Apply this transformation to a subset of the nodes within a schedule - i.e. enclose the specified Nodes in the schedule within a single PSyData region. It first uses the CallTreeUtils to determine input- and output-parameters. If requested, it will then call the LFRicExtractDriverCreator to write the stand-alone driver program. Then it will call apply of the base class.

Parameters:
  • nodes (psyclone.psyir.nodes.Node or List[psyclone.psyir.nodes.Node]) – can be a single node or a list of nodes.

  • options (Optional[Dict[str, Any]]) – a dictionary with options for transformations.

  • options["prefix"] (str) – a prefix to use for the PSyData module name (prefix_psy_data_mod) and the PSyDataType (prefix_PSyDataType) - a “_” will be added automatically. It defaults to “extract”, resulting in e.g. extract_psy_data_mod.

  • options["create_driver"] (bool) – whether or not to create a driver program at code-generation time. If set, the driver will be created in the current working directory with the name “driver-MODULE-REGION.f90” where MODULE and REGION will be the corresponding values for this region. Defaults to False.

  • options["region_name"] (Tuple[str,str]) – an optional name to use for this PSyData area, provided as a 2-tuple containing a location name followed by a local name. The pair of strings should uniquely identify a region unless aggregate information is required (and is supported by the runtime library).

validate(node_list, options=None)[source]

Perform Dynamo0.3 API specific validation checks before applying the transformation.

Parameters:
  • node_list (List[psyclone.psyir.nodes.Node]) – the list of Node(s) we are checking.

  • options (Optional[Dict[str, Any]]) – a dictionary with options for transformations.

Raises:

TransformationError – if transformation is applied to a Loop over cells in a colour without its parent Loop over colours.

class psyclone.domain.lfric.transformations.LFRicLoopFuseTrans[source]

LFRic API specialisation of the base class in order to fuse two Dynamo loops after performing validity checks. For example:

>>> from psyclone.parse.algorithm import parse
>>> from psyclone.psyGen import PSyFactory
>>>
>>> API = "lfric"
>>> FILENAME = "alg.x90"
>>> ast, invokeInfo = parse(FILENAME, api=API)
>>> psy = PSyFactory(API, distributed_memory=False).create(invoke_info)
>>> schedule = psy.invokes.get('invoke_0').schedule
>>>
>>> from psyclone.domain.lfric.transformations import LFRicLoopFuseTrans
>>> ftrans =  LFRicLoopFuseTrans()
>>>
>>> ftrans.apply(schedule[0], schedule[1])
>>> print(schedule.view())

The optional argument same_space can be set as

>>> ftrans.apply(schedule[0], schedule[1], {"same_space": True})

when applying the transformation.

validate(node1, node2, options=None)[source]

Performs various checks to ensure that it is valid to apply the LFRicLoopFuseTrans transformation to the supplied loops.

Parameters:
  • node1 (psyclone.domain.lfric.LFRicLoop) – the first Loop to fuse.

  • node2 (psyclone.domain.lfric.LFRicLoop) – the second Loop to fuse.

  • options (Optional[Dict[str, Any]]) – a dictionary with options for transformations.

  • options["same_space"] (bool) – this optional flag, set to True, asserts that an unknown iteration space (i.e. ANY_SPACE) matches the other iteration space. This is set at the user’s own risk. If both iteration spaces are discontinuous the loops can be fused without having to use the same_space flag.

Raises:
  • TransformationError – if either of the supplied loops contains an inter-grid kernel.

  • TransformationError – if one or both function spaces have invalid names.

  • TransformationError – if the same_space flag was set, but does not apply because neither field is on ANY_SPACE or the spaces are not the same.

  • TransformationError – if one or more of the iteration spaces is unknown (ANY_SPACE) and the same_space flag is not set to True.

  • TransformationError – if the loops are over different spaces that are not both discontinuous and the loops both iterate over cells.

  • TransformationError – if the loops’ upper bound names are not the same.

  • TransformationError – if the halo-depth indices of two loops are not the same.

  • TransformationError – if each loop already contains a reduction.

  • TransformationError – if the first loop has a reduction and the second loop reads the result of the reduction.

class psyclone.domain.lfric.transformations.RaisePSyIR2LFRicKernTrans[source]

Raise a generic PSyIR representation of a kernel-layer routine and metadata to an LFRic version with specialised domain-specific nodes and symbols. This is currently limited to the specialisation of kernel metadata.

>>> from psyclone.configuration import Config
>>> from psyclone.domain.lfric.transformations import             RaisePSyIR2LFRicKernTrans
>>> from psyclone.psyir.frontend.fortran import FortranReader
>>> config = Config.get().api_conf("lfric")
>>> CODE = ("""
... MODULE example
... TYPE, EXTENDS(kernel_type) :: compute_cu
...   TYPE(arg_type), DIMENSION(4) :: meta_args = (/     &
...     arg_type(GH_FIELD, GH_REAL, GH_INC, W1),         &
...     arg_type(GH_FIELD, GH_REAL, GH_READ, W3),        &
...     arg_type(GH_FIELD, GH_REAL, GH_READ, W3),        &
...     arg_type(GH_FIELD, GH_REAL, GH_READ, W3)/)
...   INTEGER :: OPERATES_ON = CELL_COLUMN
... CONTAINS
...   PROCEDURE, NOPASS :: code => compute_cu_code
... END TYPE compute_cu
... contains
...   subroutine compute_cu_code()
...   end subroutine
... end module""")
>>> fortran_reader = FortranReader()
>>> kernel_container = fortran_reader.psyir_from_source(CODE)
>>> trans = RaisePSyIR2LFRicKernTrans()
>>> trans.apply(kernel_container, {"metadata_name": "compute_cu"})
apply(node, options=None)[source]

Raise the supplied language-level kernel to LFRic-specific kernel PSyIR. Specialises the kernel container to an LFRic-specific subclass, populates this subclass with the kernel metadata extracted from the metadata symbol as specified in metadata_name (which is supplied via the options argument) and removes the symbol from the symbol table.

Parameters:
  • node (psyclone.psyir.node.Container) – a kernel represented in generic PSyIR.

  • options (Optional[Dict[str: str]]) – a dictionary with options for transformations. This is expected to contain the metadata_name.

validate(node, options=None)[source]

Validate the supplied PSyIR tree.

Parameters:
  • node (psyclone.psyir.node.Container) – a PSyIR node that is the root of a PSyIR tree.

  • options (Optional[Dict[str: str]]) – a dictionary with options for transformations.

Raises:
  • TransformationError – if the supplied node is not a Container.

  • TransformationError – if the supplied node argument has a parent.

  • TransformationError – if the metadata name has not been provided in the options argument.

  • TransformationError – if the metadata name has not been set or does not exist in the code.

  • TransformationError – if the metadata symbol does not reside in a Container (as opposed to a FileContainer).

class psyclone.transformations.DynamoOMPParallelLoopTrans(omp_directive='do', omp_schedule='static')[source]

Dynamo-specific OpenMP loop transformation. Adds Dynamo specific validity checks. Actual transformation is done by the base class.

Parameters:
  • omp_directive (str) – choose which OpenMP loop directive to use. Defaults to “do”.

  • omp_schedule (str) – the OpenMP schedule to use. Must be one of ‘runtime’, ‘static’, ‘dynamic’, ‘guided’ or ‘auto’. Defaults to ‘static’.

validate(node, options=None)[source]

Perform LFRic-specific loop validity checks then call the validate method of the base class.

Parameters:
  • node (psyclone.psyir.nodes.Node) – the Node in the Schedule to check

  • options (Optional[Dict[str, Any]]) – a dictionary with options for transformations.

Raises:
  • TransformationError – if the supplied Node is not a LFRicLoop.

  • TransformationError – if the associated loop requires colouring.

class psyclone.transformations.Dynamo0p3AsyncHaloExchangeTrans[source]

Splits a synchronous halo exchange into a halo exchange start and halo exchange end. For example:

>>> from psyclone.parse.algorithm import parse
>>> from psyclone.psyGen import PSyFactory
>>> api = "lfric"
>>> ast, invokeInfo = parse("file.f90", api=api)
>>> psy=PSyFactory(api).create(invokeInfo)
>>> schedule = psy.invokes.get('invoke_0').schedule
>>> # Uncomment the following line to see a text view of the schedule
>>> # print(schedule.view())
>>>
>>> from psyclone.transformations import Dynamo0p3AsyncHaloExchangeTrans
>>> trans = Dynamo0p3AsyncHaloExchangeTrans()
>>> trans.apply(schedule.children[0])
>>> # Uncomment the following line to see a text view of the schedule
>>> # print(schedule.view())
apply(node, options=None)[source]

Transforms a synchronous halo exchange, represented by a HaloExchange node, into an asynchronous halo exchange, represented by HaloExchangeStart and HaloExchangeEnd nodes.

Parameters:
  • node (psyclone.psygen.HaloExchange) – a synchronous haloexchange node.

  • options (Optional[Dict[str, Any]]) – a dictionary with options for transformations.

property name
Returns:

the name of this transformation as a string.

Return type:

str

validate(node, options)[source]

Internal method to check whether the node is valid for this transformation.

Parameters:
  • node (psyclone.psygen.HaloExchange) – a synchronous Halo Exchange node

  • options (Optional[Dict[str, Any]]) – a dictionary with options for transformations.

Raises:

TransformationError – if the node argument is not a HaloExchange (or subclass thereof)

class psyclone.transformations.Dynamo0p3ColourTrans[source]

Split a Dynamo 0.3 loop over cells into colours so that it can be parallelised. For example:

>>> from psyclone.parse.algorithm import parse
>>> from psyclone.psyGen import PSyFactory
>>> import transformations
>>> import os
>>> import pytest
>>>
>>> TEST_API = "lfric"
>>> _,info=parse(os.path.join(os.path.dirname(os.path.abspath(__file__)),
>>>              "tests", "test_files", "dynamo0p3",
>>>              "4.6_multikernel_invokes.f90"),
>>>              api=TEST_API)
>>> psy = PSyFactory(TEST_API).create(info)
>>> invoke = psy.invokes.get('invoke_0')
>>> schedule = invoke.schedule
>>>
>>> ctrans = Dynamo0p3ColourTrans()
>>> otrans = DynamoOMPParallelLoopTrans()
>>>
>>> # Colour all of the loops
>>> for child in schedule.children:
>>>     ctrans.apply(child)
>>>
>>> # Then apply OpenMP to each of the colour loops
>>> for child in schedule.children:
>>>     otrans.apply(child.children[0])
>>>
>>> # Uncomment the following line to see a text view of the schedule
>>> # print(schedule.view())

Colouring in the LFRic (Dynamo 0.3) API is subject to the following rules:

  • Only kernels which operate on ‘CELL_COLUMN’s and which increment a field on a continuous function space require colouring. Kernels that update a field on a discontinuous function space will cause this transformation to raise an exception. Kernels that only write to a field on a continuous function space also do not require colouring but are permitted.

  • A kernel may have at most one field with ‘GH_INC’ access.

  • A separate colour map will be required for each field that is coloured (if an invoke contains >1 kernel call).

apply(node, options=None)[source]

Performs LFRic-specific error checking and then uses the parent class to convert the Loop represented by node into a nested loop where the outer loop is over colours and the inner loop is over cells of that colour.

Parameters:
  • node (psyclone.domain.lfric.LFRicLoop) – the loop to transform.

  • options – a dictionary with options for transformations. :type options: Optional[Dict[str, Any]]

class psyclone.transformations.Dynamo0p3KernelConstTrans[source]

Modifies a kernel so that the number of dofs, number of layers and number of quadrature points are fixed in the kernel rather than being passed in by argument.

>>> from psyclone.parse.algorithm import parse
>>> from psyclone.psyGen import PSyFactory
>>> api = "lfric"
>>> ast, invokeInfo = parse("file.f90", api=api)
>>> psy=PSyFactory(api).create(invokeInfo)
>>> schedule = psy.invokes.get('invoke_0').schedule
>>> # Uncomment the following line to see a text view of the schedule
>>> # print(schedule.view())
>>>
>>> from psyclone.transformations import Dynamo0p3KernelConstTrans
>>> trans = Dynamo0p3KernelConstTrans()
>>> for kernel in schedule.coded_kernels():
>>>     trans.apply(kernel, number_of_layers=150)
>>>     kernel_schedule = kernel.get_kernel_schedule()
>>>     # Uncomment the following line to see a text view of the
>>>     # symbol table
>>>     # print(kernel_schedule.symbol_table.view())
apply(node, options=None)[source]

Transforms a kernel so that the values for the number of degrees of freedom (if a valid value for the element_order arg is provided), the number of quadrature points (if the quadrature arg is set to True) and the number of layers (if a valid value for the number_of_layers arg is provided) are constant in a kernel rather than being passed in by argument.

The “cellshape”, “element_order” and “number_of_layers” arguments are provided to mirror the namelist values that are input into an LFRic model when it is run.

Quadrature support is currently limited to XYoZ in ths transformation. In the case of XYoZ the number of quadrature points (for horizontal and vertical) are set to the element_order + 3 in the LFRic infrastructure so their value is derived.

Parameters:
  • node (psyclone.domain.lfric.LFRicKern) – a kernel node.

  • options (Optional[Dict[str, Any]]) – a dictionary with options for transformations.

  • options["cellshape"] (str) – the shape of the cells. This is provided as it helps determine the number of dofs a field has for a particular function space. Currently only “quadrilateral” is supported which is also the default value.

  • options["element_order"] (int) – the order of the cell. In combination with cellshape, this determines the number of dofs a field has for a particular function space. If it is set to None (the default) then the dofs values are not set as constants in the kernel, otherwise they are.

  • options["number_of_layers"] (int) – the number of vertical layers in the LFRic model mesh used for this particular run. If this is set to None (the default) then the nlayers value is not set as a constant in the kernel, otherwise it is.

  • options["quadrature"] (bool) – whether the number of quadrature points values are set as constants in the kernel (True) or not (False). The default is False.

property name
Returns:

the name of this transformation as a string.

Return type:

str

validate(node, options=None)[source]

This method checks whether the input arguments are valid for this transformation.

Parameters:
  • node (psyclone.domain.lfric.LFRicKern) – a dynamo 0.3 kernel node.

  • options (Optional[Dict[str, Any]]) – a dictionary with options for transformations.

  • options["cellshape"] (str) – the shape of the elements/cells.

  • options["element_order"] (int) – the order of the elements/cells.

  • options["number_of_layers"] (int) – the number of layers to use.

  • options["quadrature"] (bool) – whether quadrature dimension sizes should or shouldn’t be set as constants in a kernel.

Raises:

TransformationError – if the node argument is not a dynamo 0.3 kernel, the cellshape argument is not set to “quadrilateral”, the element_order argument is not a 0 or a positive integer, the number of layers argument is not a positive integer, the quadrature argument is not a boolean, neither element order nor number of layers arguments are set (as the transformation would then do nothing), or the quadrature argument is True but the element order is not provided (as the former needs the latter).

class psyclone.transformations.Dynamo0p3OMPLoopTrans(omp_schedule='static')[source]

LFRic (Dynamo 0.3) specific orphan OpenMP loop transformation. Adds Dynamo-specific validity checks.

Parameters:

omp_schedule (str) – the OpenMP schedule to use. Must be one of ‘runtime’, ‘static’, ‘dynamic’, ‘guided’ or ‘auto’. Defaults to ‘static’.

apply(node, options=None)[source]

Apply LFRic (Dynamo 0.3) specific OMPLoopTrans.

Parameters:
  • node (psyclone.psyir.nodes.Node) – the Node in the Schedule to check.

  • options (Optional[dict[str, Any]]) – a dictionary with options for transformations and validation.

  • options["reprod"] (bool) – indicating whether reproducible reductions should be used. By default the value from the config file will be used.

validate(node, options=None)[source]

Perform LFRic (Dynamo 0.3) specific loop validity checks for the OMPLoopTrans.

Parameters:
  • node (psyclone.psyir.nodes.Node) – the Node in the Schedule to check

  • options (Optional[Dict[str, Any]]) – a dictionary with options for transformations and validation.

  • options["reprod"] (bool) – indicating whether reproducible reductions should be used. By default the value from the config file will be used.

Raises:

TransformationError – if an OMP loop transform would create incorrect code.

class psyclone.transformations.Dynamo0p3RedundantComputationTrans[source]

This transformation allows the user to modify a loop’s bounds so that redundant computation will be performed. Redundant computation can result in halo exchanges being modified, new halo exchanges being added or existing halo exchanges being removed.

  • This transformation should be performed before any parallelisation transformations (e.g. for OpenMP) to the loop in question and will raise an exception if this is not the case.

  • This transformation can not be applied to a loop containing a reduction and will again raise an exception if this is the case.

  • This transformation can only be used to add redundant computation to a loop, not to remove it.

  • This transformation allows a loop that is already performing redundant computation to be modified, but only if the depth is increased.

apply(loop, options=None)[source]

Apply the redundant computation transformation to the loop loop. This transformation can be applied to loops iterating over ‘cells or ‘dofs’. if depth is set to a value then the value will be the depth of the field’s halo over which redundant computation will be performed. If depth is not set to a value then redundant computation will be performed to the full depth of the field’s halo.

Parameters:
  • loop (psyclone.psyGen.LFRicLoop) – the loop that we are transforming.

  • options (Optional[Dict[str, Any]]) – a dictionary with options for transformations.

  • options["depth"] (int) – the depth of the stencil. Defaults to None.

validate(node, options=None)[source]

Perform various checks to ensure that it is valid to apply the RedundantComputation transformation to the supplied node

Parameters:
  • node (psyclone.psyir.nodes.Node) – the supplied node on which we are performing validity checks

  • options (Optional[Dict[str, Any]]) – a dictionary with options for transformations.

  • options["depth"] (int) – the depth of the stencil if the value is provided and None if not.

Raises:
  • TransformationError – if the parent of the loop is a psyclone.psyir.nodes.Directive.

  • TransformationError – if the parent of the loop is not a psyclone.psyir.nodes.Loop or a psyclone.psyGen.LFRicInvokeSchedule.

  • TransformationError – if the parent of the loop is a psyclone.psyir.nodes.Loop but the original loop does not iterate over ‘colour’.

  • TransformationError – if the parent of the loop is a psyclone.psyir.nodes.Loop but the parent does not iterate over ‘colours’.

  • TransformationError – if the parent of the loop is a psyclone.psyir.nodes.Loop but the parent’s parent is not a psyclone.psyGen.LFRicInvokeSchedule.

  • TransformationError – if this transformation is applied when distributed memory is not switched on.

  • TransformationError – if the loop does not iterate over cells, dofs or colour.

  • TransformationError – if the loop contains a kernel that operates on halo cells.

  • TransformationError – if the transformation is setting the loop to the maximum halo depth but the loop already computes to the maximum halo depth.

  • TransformationError – if the transformation is setting the loop to the maximum halo depth but the loop contains a stencil access (as this would result in the field being accessed beyond the halo depth).

  • TransformationError – if the supplied depth value is not an integer.

  • TransformationError – if the supplied depth value is less than 1.

  • TransformationError – if the supplied depth value is not greater than 1 when a continuous loop is modified as this is the minimum valid value.

  • TransformationError – if the supplied depth value is not greater than the existing depth value, as we should not need to undo existing transformations.

  • TransformationError – if a depth value has been supplied but the loop has already been set to the maximum halo depth.