View page source

warp.fem.make_space_partition#

warp.fem.make_space_partition(
space_topology,
geometry_partition=None,
with_halo=True,
max_node_count=-1,
environment_first=False,
*,
device=None,
temporary_store=None,
)[source]#

Compute the subset of nodes from a function space topology that touch a geometry partition

Parameters:

  • space_topology (SpaceTopology) – the topology of the function space to consider.

  • geometry_partition (GeometryPartition | None) – The subset of the space geometry. If not provided, use the whole geometry.

  • with_halo (bool) – If True, include the halo nodes (nodes from exterior frontier cells to the partition). Halo nodes are not currently supported with environment_first.

  • environment_first (bool) – If True, order partition nodes by environment to make sure node indices are contiguous within each environment.

  • max_node_count (int) – if positive, will be used to limit the number of nodes to avoid device/host synchronization.

  • device – Warp device on which to perform and store computations

  • temporary_store (TemporaryStore)

Returns:

the resulting space partition

Return type:

SpacePartition