Lines Matching refs:KokkosCsrMatrix

30 using KokkosCsrMatrix     = KokkosCsrMatrixType<DefaultMemorySpace>;
35 using MatScalarKokkosView = KokkosCsrMatrix::values_type::non_const_type;
43 using ConstMatScalarKokkosView = KokkosCsrMatrix::values_type::const_type;
111   KokkosCsrMatrix  csrmat;       /* The CSR matrix, used to call KK functions */
114   KokkosCsrMatrix     csrmatT, csrmatH;                     /* Transpose and Hermitian of the matrix (built on demand) */
144     csrmat = KokkosCsrMatrix("csrmat", ncols, a_d, KokkosCsrGraph(j_d, i_d));
148   /* Construct with a KokkosCsrMatrix. For performance, only i, j are copied to host, but not the matrix values. */
149   Mat_SeqAIJKokkos(const KokkosCsrMatrix &csr) : csrmat(csr) /* Shallow-copy csr's views to csrmat */
169   Mat_SeqAIJKokkos(PetscInt nrows, PetscInt ncols, PetscInt nnz, const MatRowMapKokkosView &i_d, const MatColIdxKokkosView &j_d, const MatScalarKokkosView &a_d) : Mat_SeqAIJKokkos(KokkosCsrMatrix("csrmat", nrows, ncols, nnz, a_d, i_d, j_d)) { }
184   void SetColSize(MatColIdxType n) { csrmat = KokkosCsrMatrix("csrmat", n, a_dual.view_device(), csrmat.graph); }
204     csrmatT = KokkosCsrMatrix(); /* Overwrite with empty matrices */
205     csrmatH = KokkosCsrMatrix();
220 PETSC_INTERN PetscErrorCode MatSeqAIJKokkosGetKokkosCsrMatrix(Mat, KokkosCsrMatrix *);
221 PETSC_INTERN PetscErrorCode MatCreateSeqAIJKokkosWithKokkosCsrMatrix(MPI_Comm, KokkosCsrMatrix, Mat *);
222 PETSC_INTERN PetscErrorCode PrintCsrMatrix(const KokkosCsrMatrix &csrmat);
225 PETSC_INTERN PetscErrorCode MatSeqAIJKokkosGenerateTranspose_Private(Mat, KokkosCsrMatrix *);