Searched refs:MatRowMapKokkosView (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.hpp | 33 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 */ 116 …MatRowMapKokkosView transpose_perm; // A permutation array making Ta(i) = Aa… 153 …MatRowMapKokkosView 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 D | aijkok.kokkos.cxx | 262 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 D | mpiaijkok.kokkos.cxx | 611 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()
|