cudolfinx.context

CUDA Context management.

Functions

get_cuda_context()

Return the CUDA context, intializing it if needed.

get_device()

Return PETSc device.

cudolfinx.context.get_cuda_context()[source]

Return the CUDA context, intializing it if needed.

cudolfinx.context.get_device()[source]

Return PETSc device.