Home
last modified time | relevance | path

Searched refs:KokkosCsrMatrix (Results 1 – 3 of 3) sorted by relevance

/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp30 using KokkosCsrMatrix = KokkosCsrMatrixType<DefaultMemorySpace>; typedef
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…
144 csrmat = KokkosCsrMatrix("csrmat", ncols, a_d, KokkosCsrGraph(j_d, i_d)); in Mat_SeqAIJKokkos()
149 …Mat_SeqAIJKokkos(const KokkosCsrMatrix &csr) : csrmat(csr) /* Shallow-copy csr's views to csrmat */ in Mat_SeqAIJKokkos()
169 …xKokkosView &j_d, const MatScalarKokkosView &a_d) : Mat_SeqAIJKokkos(KokkosCsrMatrix("csrmat", nro… in Mat_SeqAIJKokkos()
184 …void SetColSize(MatColIdxType n) { csrmat = KokkosCsrMatrix("csrmat", n, a_dual.view_device(), csr… in SetColSize()
204 csrmatT = KokkosCsrMatrix(); /* Overwrite with empty matrices */ in DestroyMatTranspose()
[all …]
H A Daijkok.kokkos.cxx262 …MatSeqAIJKokkosGenerateTransposeStructure(Mat A, MatRowMapKokkosView &perm_d, KokkosCsrMatrix &T_d) in MatSeqAIJKokkosGenerateTransposeStructure()
303 …PetscCallCXX(T_d = KokkosCsrMatrix("csrmatT", n, m, nz, MatScalarKokkosView("Ta", nz), Ti_d, Tj_d)… in MatSeqAIJKokkosGenerateTransposeStructure()
310 PETSC_INTERN PetscErrorCode MatSeqAIJKokkosGenerateTranspose_Private(Mat A, KokkosCsrMatrix *csrmat… in MatSeqAIJKokkosGenerateTranspose_Private()
315 KokkosCsrMatrix &T = akok->csrmatT; in MatSeqAIJKokkosGenerateTranspose_Private()
348 static PetscErrorCode MatSeqAIJKokkosGenerateHermitian_Private(Mat A, KokkosCsrMatrix *csrmatH) in MatSeqAIJKokkosGenerateHermitian_Private()
353 KokkosCsrMatrix &T = akok->csrmatH; in MatSeqAIJKokkosGenerateHermitian_Private()
414 KokkosCsrMatrix csrmat; in MatMultTranspose_SeqAIJKokkos()
444 KokkosCsrMatrix csrmat; in MatMultHermitianTranspose_SeqAIJKokkos()
496 KokkosCsrMatrix csrmat; in MatMultTransposeAdd_SeqAIJKokkos()
527 KokkosCsrMatrix csrmat; in MatMultHermitianTransposeAdd_SeqAIJKokkos()
[all …]
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx119 KokkosCsrMatrix Cd, Co; // C is in split form matrices (all in local column indcies)
120 KokkosCsrMatrix C1, C2, C3, C4; // intermediate mat products
121 …KokkosCsrMatrix C2_mid, C4_mid; // alias of C2, C4; share their a[], i[], but with different j…
125 KokkosCsrMatrix Fd, Fo; // F in split form
342 static PetscErrorCode MatMPIAIJKokkosReduceBegin(MPI_Comm comm, KokkosCsrMatrix A, KokkosCsrMatrix … in MatMPIAIJKokkosReduceBegin()
617 PetscCallCXX(mm->Fd = KokkosCsrMatrix("Fd", Fm, cend - cstart, Fdnz, Fda_d, Fdi_d, Fdj_d)); in MatMPIAIJKokkosReduceBegin()
618 …PetscCallCXX(mm->Fo = KokkosCsrMatrix("Fo", Fm, n2, Fonz, Foa_d, Foi_d, Foj_d)); // Fo's column si… in MatMPIAIJKokkosReduceBegin()
674 static PetscErrorCode MatMPIAIJKokkosReduceEnd(MPI_Comm comm, KokkosCsrMatrix A, KokkosCsrMatrix B,… in MatMPIAIJKokkosReduceEnd()
930 PetscCallCXX(mm->Fd = KokkosCsrMatrix("Fd", Fm, cend - cstart, Fdnz, Fda_d, Fdi_d, Fdj_d)); in MatMPIAIJKokkosBcastBegin()
931 PetscCallCXX(mm->Fo = KokkosCsrMatrix("Fo", Fm, n2, Fonz, Foa_d, Foi_d, Foj_d)); in MatMPIAIJKokkosBcastBegin()
[all …]