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

VecCUDAGetArrayWrite

Provides write access to the CUDA buffer inside a vector.

Synopsis

#include "petscvec.h"   
PETSC_EXTERN PetscErrorCode VecCUDAGetArrayWrite(Vec v, PetscScalar **a)
The data pointed to by the device pointer is uninitialized. The user may not read from this data. Furthermore, the entire array needs to be filled by the user to obtain well-defined behaviour. The device memory will be allocated by this function if it hasn't been allocated previously. This is analogous to intent(out) in Fortran.

The device pointer needs to be released with VecCUDARestoreArrayWrite(). When the pointer is released the host data of the vector is marked as out of data. Subsequent access of the host data with e.g. VecGetArray() incurs a device to host data transfer.

Input Parameter

v - the vector

Output Parameter

a - the CUDA pointer

Fortran note

This function is not currently available from Fortran.

See Also

VecCUDARestoreArrayWrite(), VecCUDAGetArray(), VecCUDAGetArrayRead(), VecCUDAGetArrayWrite(), VecGetArray(), VecGetArrayRead()

Level

advanced

Location

src/vec/vec/interface/rvector.c

Examples

src/snes/tutorials/ex47cu.cu.html

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