warp.fem.make_space_partition#

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

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

Either space_topology or space must be provided (and will be considered in that order).

Parameters:

  • space (FunctionSpace | None) – (deprecated) the function space defining the topology if space_topology is None.

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

  • space_topology (SpaceTopology | None) – the topology of the function space to consider. If None, deduced from space.

  • with_halo (bool) – if True, include the halo nodes (nodes from exterior frontier cells to the partition)

  • 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