xref: /petsc/include/petscviennacl.h (revision 26bd150190f26c623f12d3ed48c77abbffd51c93)
1*26bd1501SBarry Smith #if !defined(PETSCVIENNACL_H)
2*26bd1501SBarry Smith #define PETSCVIENNACL_H
304afb09dSKarl Rupp 
404afb09dSKarl Rupp #include <petscvec.h>
50cd40bc2SKarl Rupp 
60cd40bc2SKarl Rupp #if defined(PETSC_HAVE_CUDA)
70cd40bc2SKarl Rupp #define VIENNACL_WITH_CUDA
80cd40bc2SKarl Rupp #endif
90cd40bc2SKarl Rupp 
100cd40bc2SKarl Rupp #if defined(PETSC_HAVE_OPENCL)
110cd40bc2SKarl Rupp #define VIENNACL_WITH_OPENCL
120cd40bc2SKarl Rupp #endif
130cd40bc2SKarl Rupp 
140cd40bc2SKarl Rupp #if defined(PETSC_HAVE_OPENMP)
150cd40bc2SKarl Rupp #define VIENNACL_WITH_OPENMP
160cd40bc2SKarl Rupp #endif
170cd40bc2SKarl Rupp 
18269c5266SKarl Rupp #include <viennacl/forwards.h>
19269c5266SKarl Rupp #include <viennacl/vector_proxy.hpp>
2004afb09dSKarl Rupp #include <viennacl/vector.hpp>
2104afb09dSKarl Rupp 
2204afb09dSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLGetArrayReadWrite(Vec v, viennacl::vector<PetscScalar> **a);
2304afb09dSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLRestoreArrayReadWrite(Vec v, viennacl::vector<PetscScalar> **a);
2404afb09dSKarl Rupp 
2504afb09dSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLGetArrayRead(Vec v, const viennacl::vector<PetscScalar> **a);
2604afb09dSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLRestoreArrayRead(Vec v, const viennacl::vector<PetscScalar> **a);
2704afb09dSKarl Rupp 
2804afb09dSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLGetArrayWrite(Vec v, viennacl::vector<PetscScalar> **a);
2904afb09dSKarl Rupp PETSC_EXTERN PetscErrorCode VecViennaCLRestoreArrayWrite(Vec v, viennacl::vector<PetscScalar> **a);
3004afb09dSKarl Rupp 
3104afb09dSKarl Rupp 
3204afb09dSKarl Rupp #endif
33