#include "petscvec.h" PetscErrorCode VecCreateMPICUDAWithArray(MPI_Comm comm,PetscInt bs,PetscInt n,PetscInt N,const PetscScalar array[],Vec *vv)Collective
comm | - the MPI communicator to use | |
bs | - block size, same meaning as VecSetBlockSize() | |
n | - local vector length, cannot be PETSC_DECIDE | |
N | - global vector length (or PETSC_DECIDE to have calculated) | |
array | - the user provided GPU array to store the vector values |
vv | - the vector |
If the user-provided array is NULL, then VecCUDAPlaceArray() can be used at a later stage to SET the array for storing the vector values.
PETSc does NOT free the array when the vector is destroyed via VecDestroy(). The user should not free the array until the vector is destroyed.