Actual source code: viennaclmatimpl.h
petsc-3.10.5 2019-03-28
Report Typos and Errors

      4:  #include <../src/mat/impls/aij/seq/aij.h>

      6: /* Pulls in some ViennaCL includes as well as VIENNACL_WITH_OPENCL: */
      7:  #include <../src/vec/vec/impls/seq/seqviennacl/viennaclvecimpl.h>

      9: /* for everything else */
     10: #include "viennacl/compressed_matrix.hpp"
     11: #include "viennacl/compressed_compressed_matrix.hpp"

     14: typedef viennacl::compressed_matrix<PetscScalar>   ViennaCLAIJMatrix;
     15: typedef viennacl::compressed_compressed_matrix<PetscScalar>   ViennaCLCompressedAIJMatrix;

     18: struct Mat_SeqAIJViennaCL {
     19:   Mat_SeqAIJViennaCL() : tempvec(NULL), mat(NULL), compressed_mat(NULL) {}
     20:   ViennaCLVector               *tempvec;
     21:   ViennaCLAIJMatrix            *mat;  /* pointer to the matrix on the GPU */
     22:   ViennaCLCompressedAIJMatrix  *compressed_mat; /* compressed CSR */
     23: };

     25: PETSC_INTERN PetscErrorCode MatViennaCLCopyToGPU(Mat);
     26: PETSC_INTERN PetscErrorCode MatViennaCLCopyFromGPU(Mat, ViennaCLAIJMatrix*);
     27: #endif