warp.fem.LinearGeometryPartition#

class warp.fem.LinearGeometryPartition(
geometry,
partition_rank,
partition_count,
device=None,
temporary_store=None,
)[source]#
Parameters:
__init__(
geometry,
partition_rank,
partition_count,
device=None,
temporary_store=None,
)[source]#

Creates a geometry partition by uniformly partionning cell indices

Parameters:
  • geometry (Geometry) – the geometry to partition

  • partition_rank (int) – the index of the partition being created

  • partition_count (int) – the number of partitions that will be created over the geometry

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

  • temporary_store (TemporaryStore)

Methods

__init__(geometry, partition_rank, ...[, ...])

Creates a geometry partition by uniformly partionning cell indices

boundary_side_count()

cell_arg_value(device)

cell_count()

compute_side_indices_from_cells(...[, ...])

fill_cell_arg(args, device)

fill_side_arg(args, device)

frontier_side_count()

side_arg_value(device)

side_count()

Attributes

CellArg

SideArg

boundary_side_index

cell_index

frontier_side_index

name

partition_cell_index

side_index

side_to_cell_arg

geometry

Underlying geometry

cell_count()[source]#
Return type:

int

CellArg = <warp._src.codegen.Struct object>#
fill_cell_arg(args, device)[source]#
Parameters:

args (CellArg)

cell_index = <Function LinearGeometryPartition__cell_index(args: warp._src.fem.geometry.partition.LinearGeometryPartition.CellArg, partition_cell_index: int)>#
Parameters:
  • args (CellArg)

  • partition_cell_index (int)

partition_cell_index = <Function LinearGeometryPartition__partition_cell_index(args: warp._src.fem.geometry.partition.LinearGeometryPartition.CellArg, cell_index: int)>#
Parameters:
  • args (CellArg)

  • cell_index (int)