psyclone.dynamo0p3.DynCMAOperators Class Reference

Inheritance diagram for psyclone.dynamo0p3.DynCMAOperators:

Collaboration diagram for psyclone.dynamo0p3.DynCMAOperators:

Public Member Functions
def	__init__ (self, node)
def	initialise (self, parent)
Public Member Functions inherited from psyclone.domain.lfric.lfric_collection.LFRicCollection
def	declarations (self, parent)

Static Public Attributes
list	cma_same_fs_params
list	cma_diff_fs_params

Detailed Description

Holds all information on the Column-Matrix-Assembly operators
required by an Invoke or Kernel stub.

:param node: either an Invoke schedule or a single Kernel object.
:type node: :py:class:`psyclone.dynamo0p3.DynSchedule` or \
            :py:class:`psyclone.domain.lfric.LFRicKern`

Definition at line 1777 of file dynamo0p3.py.

Member Function Documentation

initialise()

def psyclone.dynamo0p3.DynCMAOperators.initialise	(		self,
			parent
	)

Generates the calls to the LFRic infrastructure that look-up
the various components of each CMA operator. Adds these as
children of the supplied parent node.

:param parent: f2pygen node representing the PSy-layer routine.
:type parent: :py:class:`psyclone.f2pygen.SubroutineGen`

Reimplemented from psyclone.domain.lfric.lfric_collection.LFRicCollection.

Definition at line 1867 of file dynamo0p3.py.

    def initialise(self, parent):
        '''
        Generates the calls to the LFRic infrastructure that look-up
        the various components of each CMA operator. Adds these as
        children of the supplied parent node.
 
        :param parent: f2pygen node representing the PSy-layer routine.
        :type parent: :py:class:`psyclone.f2pygen.SubroutineGen`
 
        '''
        # If we have no CMA operators then we do nothing
        if not self._cma_ops:
            return
 
        parent.add(CommentGen(parent, ""))
        parent.add(CommentGen(parent,
                              " Look-up information for each CMA operator"))
        parent.add(CommentGen(parent, ""))
 
        const = LFRicConstants()
        suffix = const.ARG_TYPE_SUFFIX_MAPPING["gh_columnwise_operator"]
 
        for op_name in self._cma_ops:
            # First, assign a pointer to the array containing the actual
            # matrix.
            cma_name = self._symbol_table.lookup_with_tag(
                f"{op_name}:{suffix}").name
            parent.add(AssignGen(parent, lhs=cma_name, pointer=True,
                                 rhs=self._cma_ops[op_name]["arg"].
                                 proxy_name_indexed+"%columnwise_matrix"))
            # Then make copies of the related integer parameters
            for param in self._cma_ops[op_name]["params"]:
                param_name = self._symbol_table.find_or_create_tag(
                    f"{op_name}:{param}:{suffix}").name
                parent.add(AssignGen(parent, lhs=param_name,
                                     rhs=self._cma_ops[op_name]["arg"].
                                     proxy_name_indexed+"%"+param))
 

References psyclone.dynamo0p3.DynCMAOperators._cma_ops, psyclone.domain.lfric.lfric_collection.LFRicCollection._invoke, psyclone.nemo.NemoInvokeSchedule._invoke, psyclone.psyGen.InvokeSchedule._invoke, psyclone.domain.lfric.lfric_collection.LFRicCollection._symbol_table, psyclone.dynamo0p3.DynMeshes._symbol_table, psyclone.dynamo0p3.HaloDepth._symbol_table, psyclone.psyGen.HaloExchange._symbol_table, psyclone.psyir.backend.sympy_writer.SymPyWriter._symbol_table, and psyclone.psyir.nodes.scoping_node.ScopingNode._symbol_table.

Member Data Documentation

cma_diff_fs_params

list psyclone.dynamo0p3.DynCMAOperators.cma_diff_fs_params

static

Initial value:

=  ["nrow", "ncol", "bandwidth", "alpha",
                          "beta", "gamma_m", "gamma_p"]

Definition at line 1793 of file dynamo0p3.py.

cma_same_fs_params

list psyclone.dynamo0p3.DynCMAOperators.cma_same_fs_params

static

Initial value:

=  ["nrow", "bandwidth", "alpha",
                          "beta", "gamma_m", "gamma_p"]

Definition at line 1789 of file dynamo0p3.py.

---

The documentation for this class was generated from the following file:

• /home/docs/checkouts/readthedocs.org/user_builds/psyclone-ref/checkouts/latest/src/psyclone/dynamo0p3.py