xref: /petsc/src/mat/impls/aij/seq/seqviennacl/viennaclmatimpl.h (revision a3430c569f2f14ee3f6a1c44cd55773bcfd7d13e)
1d67ff14aSKarl Rupp #if !defined(__VIENNACLMATIMPL)
2d67ff14aSKarl Rupp #define __VIENNACLMATIMPL
3d67ff14aSKarl Rupp 
4aaa7dc30SBarry Smith #include <../src/mat/impls/aij/seq/aij.h>          /*I "petscmat.h" I*/
50a9f7e6aSKarl Rupp 
63db98f88SKarl Rupp /* Pulls in some ViennaCL includes as well as VIENNACL_WITH_OPENCL: */
7d67ff14aSKarl Rupp #include <../src/vec/vec/impls/seq/seqviennacl/viennaclvecimpl.h>
8d67ff14aSKarl Rupp 
9d67ff14aSKarl Rupp /* for everything else */
10d67ff14aSKarl Rupp #include "viennacl/compressed_matrix.hpp"
11*a3430c56SKarl Rupp #include "viennacl/compressed_compressed_matrix.hpp"
12d67ff14aSKarl Rupp 
13d67ff14aSKarl Rupp 
14d67ff14aSKarl Rupp typedef viennacl::compressed_matrix<PetscScalar>   ViennaCLAIJMatrix;
15*a3430c56SKarl Rupp typedef viennacl::compressed_compressed_matrix<PetscScalar>   ViennaCLCompressedAIJMatrix;
16d67ff14aSKarl Rupp 
17d67ff14aSKarl Rupp 
18d67ff14aSKarl Rupp struct Mat_SeqAIJViennaCL {
19*a3430c56SKarl Rupp   ViennaCLVector               *tempvec;
20d67ff14aSKarl Rupp   ViennaCLAIJMatrix            *mat;  /* pointer to the matrix on the GPU */
21*a3430c56SKarl Rupp   ViennaCLCompressedAIJMatrix  *compressed_mat; /* compressed CSR */
22d67ff14aSKarl Rupp };
23d67ff14aSKarl Rupp 
24d67ff14aSKarl Rupp PETSC_INTERN PetscErrorCode MatViennaCLCopyToGPU(Mat);
25d67ff14aSKarl Rupp PETSC_INTERN PetscErrorCode MatViennaCLCopyFromGPU(Mat, ViennaCLAIJMatrix*);
26d67ff14aSKarl Rupp #endif
27