xref: /petsc/src/mat/impls/aij/seq/seqviennacl/viennaclmatimpl.h (revision a76eec0e025e246efca9d3ebd46f014f5abffee5)
1 #if !defined(__VIENNACLMATIMPL)
2 #define __VIENNACLMATIMPL
3 
4 #include "../src/mat/impls/aij/seq/aij.h"          /*I "petscmat.h" I*/
5 
6 /* Pulls in some ViennaCL includes as well as VIENNACL_WITH_OPENCL: */
7 #include <../src/vec/vec/impls/seq/seqviennacl/viennaclvecimpl.h>
8 
9 /* for everything else */
10 #include "viennacl/compressed_matrix.hpp"
11 
12 
13 typedef viennacl::compressed_matrix<PetscScalar>   ViennaCLAIJMatrix;
14 
15 
16 struct Mat_SeqAIJViennaCL {
17   ViennaCLAIJMatrix      *mat;  /* pointer to the matrix on the GPU */
18 };
19 
20 PETSC_INTERN PetscErrorCode MatViennaCLCopyToGPU(Mat);
21 PETSC_INTERN PetscErrorCode MatViennaCLCopyFromGPU(Mat, ViennaCLAIJMatrix*);
22 #endif
23