xref: /petsc/include/petscviennacl.h (revision a496304597bacff3545e802853d69e8765312868)
1*a4963045SJacob Faibussowitsch #pragma once
204afb09dSKarl Rupp 
304afb09dSKarl Rupp #include <petscvec.h>
40cd40bc2SKarl Rupp 
50cd40bc2SKarl Rupp #if defined(PETSC_HAVE_CUDA)
60cd40bc2SKarl Rupp   #define VIENNACL_WITH_CUDA
70cd40bc2SKarl Rupp #endif
80cd40bc2SKarl Rupp 
90cd40bc2SKarl Rupp #if defined(PETSC_HAVE_OPENCL)
100cd40bc2SKarl Rupp   #define VIENNACL_WITH_OPENCL
110cd40bc2SKarl Rupp #endif
120cd40bc2SKarl Rupp 
130cd40bc2SKarl Rupp #if defined(PETSC_HAVE_OPENMP)
140cd40bc2SKarl Rupp   #define VIENNACL_WITH_OPENMP
150cd40bc2SKarl Rupp #endif
160cd40bc2SKarl Rupp 
17269c5266SKarl Rupp #include <viennacl/forwards.h>
18269c5266SKarl Rupp #include <viennacl/vector_proxy.hpp>
1904afb09dSKarl Rupp #include <viennacl/vector.hpp>
2004afb09dSKarl Rupp 
21f2d70e9dSBarry Smith PETSC_EXTERN PetscErrorCode VecViennaCLGetArray(Vec v, viennacl::vector<PetscScalar> **a);
22f2d70e9dSBarry Smith PETSC_EXTERN PetscErrorCode VecViennaCLRestoreArray(Vec v, viennacl::vector<PetscScalar> **a);
2304afb09dSKarl Rupp 
2404afb09dSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLGetArrayRead(Vec v, const viennacl::vector<PetscScalar> **a);
2504afb09dSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLRestoreArrayRead(Vec v, const viennacl::vector<PetscScalar> **a);
2604afb09dSKarl Rupp 
2704afb09dSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLGetArrayWrite(Vec v, viennacl::vector<PetscScalar> **a);
2804afb09dSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLRestoreArrayWrite(Vec v, viennacl::vector<PetscScalar> **a);
2904afb09dSKarl Rupp 
30a83f5ae2SKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecCreateSeqViennaCLWithArray(MPI_Comm, PetscInt, PetscInt, const viennacl::vector<PetscScalar> *, Vec *);
3167d63736SKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecCreateSeqViennaCLWithArrays(MPI_Comm, PetscInt, PetscInt, const PetscScalar *, const viennacl::vector<PetscScalar> *, Vec *);
32a83f5ae2SKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecCreateMPIViennaCLWithArray(MPI_Comm, PetscInt, PetscInt, PetscInt, const viennacl::vector<PetscScalar> *, Vec *);
3367d63736SKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecCreateMPIViennaCLWithArrays(MPI_Comm, PetscInt, PetscInt, PetscInt, const PetscScalar *, const viennacl::vector<PetscScalar> *, Vec *);
34a83f5ae2SKaushik Kulkarni 
35a83f5ae2SKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecViennaCLPlaceArray(Vec, const viennacl::vector<PetscScalar> *);
36a83f5ae2SKaushik Kulkarni PETSC_EXTERN PetscErrorCode VecViennaCLResetArray(Vec);
37