psyclone.dynamo0p3.DynGlobalSum Class Reference

Inheritance diagram for psyclone.dynamo0p3.DynGlobalSum:

Collaboration diagram for psyclone.dynamo0p3.DynGlobalSum:

Public Member Functions
def	__init__ (self, scalar, parent=None)
def	gen_code (self, parent)
Public Member Functions inherited from psyclone.psyGen.GlobalSum
def	scalar (self)
def	dag_name (self)
def	args (self)
def	node_str (self, colour=True)

Detailed Description

Dynamo specific global sum class which can be added to and
manipulated in a schedule.

:param scalar: the kernel argument for which to perform a global sum.
:type scalar: :py:class:`psyclone.dynamo0p3.DynKernelArgument`
:param parent: the parent node of this node in the PSyIR.
:type parent: :py:class:`psyclone.psyir.nodes.Node`

:raises GenerationError: if distributed memory is not enabled.
:raises InternalError: if the supplied argument is not a scalar.
:raises GenerationError: if the scalar is not of "real" intrinsic type.

Definition at line 3632 of file dynamo0p3.py.

Member Function Documentation

gen_code()

def psyclone.dynamo0p3.DynGlobalSum.gen_code	(		self,
			parent
	)

Dynamo-specific code generation for this class.

:param parent: f2pygen node to which to add AST nodes.
:type parent: :py:class:`psyclone.f2pygen.SubroutineGen`

Definition at line 3668 of file dynamo0p3.py.

    def gen_code(self, parent):
        '''
        Dynamo-specific code generation for this class.
 
        :param parent: f2pygen node to which to add AST nodes.
        :type parent: :py:class:`psyclone.f2pygen.SubroutineGen`
 
        '''
        name = self._scalar.name
        # Use InvokeSchedule SymbolTable to share the same symbol for all
        # GlobalSums in the Invoke.
        sum_name = self.ancestor(InvokeSchedule).symbol_table.\
            find_or_create_tag("global_sum").name
        sum_type = self._scalar.data_type
        sum_mod = self._scalar.module_name
        parent.add(UseGen(parent, name=sum_mod, only=True,
                          funcnames=[sum_type]))
        parent.add(TypeDeclGen(parent, datatype=sum_type,
                               entity_decls=[sum_name]))
        parent.add(AssignGen(parent, lhs=sum_name+"%value", rhs=name))
        parent.add(AssignGen(parent, lhs=name, rhs=sum_name+"%get_sum()"))
 
 

References psyclone.domain.lfric.arg_index_to_metadata_index.ArgIndexToMetadataIndex._scalar(), psyclone.domain.lfric.metadata_to_arguments_rules.MetadataToArgumentsRules._scalar(), psyclone.psyGen.GlobalSum._scalar, and psyclone.psyir.nodes.node.Node.ancestor().

Here is the call graph for this function:

Here is the caller graph for this function:

---

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