Home
last modified time | relevance | path

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

/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp35 using MatScalarKokkosView = KokkosCsrMatrix::values_type::non_const_type; typedef
61 MatScalarKokkosView aL_d, aU_d, aLt_d, aUt_d; /* matrix values */
70 MatScalarKokkosView D_d;
169 …st MatRowMapKokkosView &i_d, const MatColIdxKokkosView &j_d, const MatScalarKokkosView &a_d) : Mat… in Mat_SeqAIJKokkos()
227 PETSC_INTERN PetscErrorCode MatSeqAIJGetKokkosView(Mat, MatScalarKokkosView *);
228 PETSC_INTERN PetscErrorCode MatSeqAIJRestoreKokkosView(Mat, MatScalarKokkosView *);
229 PETSC_INTERN PetscErrorCode MatSeqAIJGetKokkosViewWrite(Mat, MatScalarKokkosView *);
230 PETSC_INTERN PetscErrorCode MatSeqAIJRestoreKokkosViewWrite(Mat, MatScalarKokkosView *);
H A Daijkok.kokkos.cxx303 …PetscCallCXX(T_d = KokkosCsrMatrix("csrmatT", n, m, nz, MatScalarKokkosView("Ta", nz), Ti_d, Tj_d)… in MatSeqAIJKokkosGenerateTransposeStructure()
715 MatScalarKokkosView aa, ba, ca; in MatSeqAIJKokkosMergeMats()
746 auto ca = MatScalarKokkosView("a", aa.extent(0) + ba.extent(0)); in MatSeqAIJKokkosMergeMats()
1200 PetscErrorCode MatSeqAIJGetKokkosView(Mat A, MatScalarKokkosView *kv) in MatSeqAIJGetKokkosView()
1214 PetscErrorCode MatSeqAIJRestoreKokkosView(Mat A, MatScalarKokkosView *kv) in MatSeqAIJRestoreKokkosView()
1224 PetscErrorCode MatSeqAIJGetKokkosViewWrite(Mat A, MatScalarKokkosView *kv) in MatSeqAIJGetKokkosViewWrite()
1237 PetscErrorCode MatSeqAIJRestoreKokkosViewWrite(Mat A, MatScalarKokkosView *kv) in MatSeqAIJRestoreKokkosViewWrite()
1264 MatScalarKokkosView Ya; in MatAXPY_SeqAIJKokkos()
1395 MatScalarKokkosView Aa; in MatSetValuesCOO_SeqAIJKokkos()
1704 factors->aUt_d = MatScalarKokkosView(NoInit("factors->aUt_d"), factors->aU_d.extent(0)); in MatSeqAIJKokkosTransposeSolveCheck()
[all …]
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx124 MatScalarKokkosView rootBuf, leafBuf;
599 mm->leafBuf = MatScalarKokkosView(NoInit("leafBuf"), nleaves); in MatMPIAIJKokkosReduceBegin()
600 mm->rootBuf = MatScalarKokkosView(NoInit("rootBuf"), nroots); in MatMPIAIJKokkosReduceBegin()
610 MatScalarKokkosView Fda_d(NoInit("Fda_d"), Fdnz); in MatMPIAIJKokkosReduceBegin()
613 MatScalarKokkosView Foa_d(NoInit("Foa_d"), Fonz); in MatMPIAIJKokkosReduceBegin()
918 mm->rootBuf = MatScalarKokkosView(NoInit("rootBuf"), nroots); in MatMPIAIJKokkosBcastBegin()
919 mm->leafBuf = MatScalarKokkosView(NoInit("leafBuf"), nleaves); in MatMPIAIJKokkosBcastBegin()
924 MatScalarKokkosView Fda_d(NoInit("Fda_d"), Fdnz), Foa_d(NoInit("Foa_d"), Fonz); in MatMPIAIJKokkosBcastBegin()
962 MatScalarKokkosView &rootBuf = mm->rootBuf; in MatMPIAIJKokkosBcastBegin()
963 MatScalarKokkosView &leafBuf = mm->leafBuf; in MatMPIAIJKokkosBcastBegin()
[all …]