Home
last modified time | relevance | path

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

/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp33 using MatRowMapKokkosView = KokkosCsrGraph::row_map_type::non_const_type; typedef
59 MatRowMapKokkosView iL_d, iU_d, iLt_d, iUt_d; /* rowmap for L, U, L^t, U^t of A=LU */
116MatRowMapKokkosView transpose_perm; // A permutation array making Ta(i) = Aa…
153MatRowMapKokkosView i_d(const_cast<MatRowMapType *>(csr.graph.row_map.data()), csr.graph.row_map.e… in Mat_SeqAIJKokkos()
169 …Mat_SeqAIJKokkos(PetscInt nrows, PetscInt ncols, PetscInt nnz, const MatRowMapKokkosView &i_d, con… in Mat_SeqAIJKokkos()
H A Daijkok.kokkos.cxx262 static PetscErrorCode MatSeqAIJKokkosGenerateTransposeStructure(Mat A, MatRowMapKokkosView &perm_d,… in MatSeqAIJKokkosGenerateTransposeStructure()
335 MatRowMapKokkosView perm; in MatSeqAIJKokkosGenerateTranspose_Private()
373 MatRowMapKokkosView perm; in MatSeqAIJKokkosGenerateHermitian_Private()
716 MatRowMapKokkosView ai, bi, ci; in MatSeqAIJKokkosMergeMats()
747 auto ci = MatRowMapKokkosView("i", ai.extent(0)); in MatSeqAIJKokkosMergeMats()
1297 MatRowMapKokkosView Xi = xkok->i_dual.view_device(), Yi = ykok->i_dual.view_device(); in MatAXPY_SeqAIJKokkos()
1702 …factors->iUt_d = MatRowMapKokkosView("factors->iUt_d", n + 1); // KK requires this view to be init… in MatSeqAIJKokkosTransposeSolveCheck()
1727 …kosView, ConstMatScalarKokkosView, MatRowMapKokkosView, MatColIdxKokkosView, MatScalarKokkosView, in MatSeqAIJKokkosTransposeSolveCheck()
1731 …PetscCallCXX(sort_crs_matrix<DefaultExecutionSpace, MatRowMapKokkosView, MatColIdxKokkosView, MatS… in MatSeqAIJKokkosTransposeSolveCheck()
1739 …factors->iLt_d = MatRowMapKokkosView("factors->iLt_d", n + 1); // KK requires this view to be init… in MatSeqAIJKokkosTransposeSolveCheck()
[all …]
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx611 MatRowMapKokkosView Fdi_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Fdi_h); in MatMPIAIJKokkosReduceBegin()
614 MatRowMapKokkosView Foi_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Foi_h); in MatMPIAIJKokkosReduceBegin()
925 MatRowMapKokkosView Fdi_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Fdi_h); in MatMPIAIJKokkosBcastBegin()
927 MatRowMapKokkosView Foi_d = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), Foi_h); in MatMPIAIJKokkosBcastBegin()