implementations.datatype_classes.cupy_mesh module

class comp2_cupy_mesh(init, val=0.0)[source]

Bases: object

RHS data type for cupy_meshes with 2 components

comp1

first part

Type:

cupy_mesh.cupy_mesh

comp2

second part

Type:

cupy_mesh.cupy_mesh

class cupy_mesh(init, val=0.0, offset=0, buffer=None, strides=None, order=None)[source]

Bases: ndarray

CuPy-based datatype for serial or parallel meshes.

bcast(root=None, comm=None)[source]

Routine for broadcasting values

Parameters:

  • root (int) – process with value to broadcast

  • comm – communicator

Returns:

broadcasted values

property comm

Getter for the communicator

irecv(source=None, tag=None, comm=None)[source]

Routine for receiving in time

Parameters:

  • source (int) – source rank

  • tag (int) – communication tag

  • comm – communicator

Returns:

None

isend(dest=None, tag=None, comm=None)[source]

Routine for sending data forward in time (non-blocking)

Parameters:

  • dest (int) – target rank

  • tag (int) – communication tag

  • comm – communicator

Returns:

request handle

class imex_cupy_mesh(init, val=0.0)[source]

Bases: object

RHS data type for cupy_meshes with implicit and explicit components

This data type can be used to have RHS with 2 components (here implicit and explicit)

impl

implicit part

Type:

cupy_mesh.cupy_mesh

expl

explicit part

Type:

cupy_mesh.cupy_mesh