warp.fem.SpacePartition

class warp.fem.SpacePartition(space_topology, geo_partition)

Partition of a function space over a geometry partition.

A space partition associates nodes of a function space topology with a geometry partition, enabling distributed computation over subdomains.

Parameters:

• space_topology (SpaceTopology)

• geo_partition (GeometryPartition)

__init__(space_topology, geo_partition)

Parameters:

• space_topology (SpaceTopology)

• geo_partition (GeometryPartition)

Methods

__init__(space_topology, geo_partition)
fill_partition_arg(arg, device)	Fill partition arguments for device functions.
interior_node_count()	Return number of interior nodes in this partition.
node_count()	Return number of nodes in this partition.
owned_node_count()	Return number of nodes in this partition, excluding exterior halo.
partition_arg_value(device)
partition_node_index(args, space_node_index)	Return the index in the partition of a function space node, or NULL_NODE_INDEX if it does not exist.
rebuild([device, temporary_store])	Rebuild the space partition indices
space_node_indices()	Return the global function space indices for nodes in this partition

Attributes

name	Name of the space partition.
space_topology	Topology of the function space being partitioned.
geo_partition	Partition of the geometry controlling how to partition the space.

class PartitionArg

Structure containing arguments to be passed to device functions.

space_topology: SpaceTopology

Topology of the function space being partitioned.

geo_partition: GeometryPartition

Partition of the geometry controlling how to partition the space.

node_count()

Return number of nodes in this partition.

owned_node_count()

Return number of nodes in this partition, excluding exterior halo.

Return type:

int

interior_node_count()

Return number of interior nodes in this partition.

Return type:

int

space_node_indices()

Return the global function space indices for nodes in this partition

Return type:

array

rebuild(device=None, temporary_store=None)

Rebuild the space partition indices

Parameters:

• device (Optional)

• temporary_store (TemporaryStore | None)

partition_arg_value(device)

fill_partition_arg(arg, device)

Fill partition arguments for device functions.

static partition_node_index(args, space_node_index)

Return the index in the partition of a function space node, or NULL_NODE_INDEX if it does not exist.

Parameters:

• args (PartitionArg)

• space_node_index (int)

property name: str

Name of the space partition.