petsc-3.14.6 2021-03-30
Report Typos and Errors

MatDenseCUDAGetArray

Provides access to the CUDA buffer inside a dense matrix. The array must be restored with MatDenseCUDARestoreArray() when no longer needed.

Synopsis

#include "petscmat.h"  
PetscErrorCode MatDenseCUDAGetArray(Mat A, PetscScalar **a)
Not Collective

Input Parameters

A - the matrix

Output Parameters

array - the GPU array in column major order

Notes

Data can be copied to the GPU due to operations done on the CPU. If you need write only access, use MatDenseCUDAGetArrayWrite(). For read-only access, use MatDenseCUDAGetArrayRead().

See Also

MatDenseCUDAGetArrayRead(), MatDenseCUDARestoreArray(), MatDenseCUDARestoreArrayWrite(), MatDenseCUDAGetArrayWrite(), MatDenseCUDARestoreArrayRead()

Level

developer

Location

src/mat/impls/dense/mpi/mpidense.c

Implementations

MatDenseCUDAGetArray_MPIDenseCUDA in src/mat/impls/dense/mpi/mpidense.c
MatDenseCUDAGetArray_SeqDenseCUDA in src/mat/impls/dense/seq/cuda/densecuda.cu

Index of all Mat routines
Table of Contents for all manual pages
Index of all manual pages