1 #if !defined(__PETSCVIENNACL_H) 2 #define __PETSCVIENNACL_H 3 4 #include <petscvec.h> 5 6 #if defined(PETSC_HAVE_CUDA) 7 #define VIENNACL_WITH_CUDA 8 #endif 9 10 #if defined(PETSC_HAVE_OPENCL) 11 #define VIENNACL_WITH_OPENCL 12 #endif 13 14 #if defined(PETSC_HAVE_OPENMP) 15 #define VIENNACL_WITH_OPENMP 16 #endif 17 18 #include <viennacl/forwards.h> 19 #include <viennacl/vector_proxy.hpp> 20 #include <viennacl/vector.hpp> 21 22 PETSC_EXTERN PetscErrorCode VecViennaCLGetArrayReadWrite(Vec v, viennacl::vector<PetscScalar> **a); 23 PETSC_EXTERN PetscErrorCode VecViennaCLRestoreArrayReadWrite(Vec v, viennacl::vector<PetscScalar> **a); 24 25 PETSC_EXTERN PetscErrorCode VecViennaCLGetArrayRead(Vec v, const viennacl::vector<PetscScalar> **a); 26 PETSC_EXTERN PetscErrorCode VecViennaCLRestoreArrayRead(Vec v, const viennacl::vector<PetscScalar> **a); 27 28 PETSC_EXTERN PetscErrorCode VecViennaCLGetArrayWrite(Vec v, viennacl::vector<PetscScalar> **a); 29 PETSC_EXTERN PetscErrorCode VecViennaCLRestoreArrayWrite(Vec v, viennacl::vector<PetscScalar> **a); 30 31 32 #endif 33