Home
last modified time | relevance | path

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

/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp18 using MatRowMapType = PetscInt; typedef
23 …arse::CrsMatrix<MatScalarType, MatColIdxType, MemorySpace, void /* MemoryTraits */, MatRowMapType>;
49 using MatRowMapKokkosDualView = Kokkos::DualView<MatRowMapType *>;
53 using KernelHandle = KokkosKernels::Experimental::KokkosKernelsHandle<MatRowMapType, MatColIdxType,…
128 MatRowMapKokkosViewHost i_h(const_cast<MatRowMapType *>(aseq->i), nrows + 1); in Mat_SeqAIJKokkos()
153 …MatRowMapKokkosView i_d(const_cast<MatRowMapType *>(csr.graph.row_map.data()), csr.graph.row_map.e… in Mat_SeqAIJKokkos()
172 MatRowMapType *i_host_data() { return i_dual.view_host().data(); } in i_host_data()
176 MatRowMapType *i_device_data() { return i_dual.view_device().data(); } in i_device_data()
181 MatRowMapType nnz() { return csrmat.nnz(); } in nnz()
186 void SetDiagonal(const MatRowMapType *diag) in SetDiagonal()
[all …]
H A Daijkok.kokkos.cxx268 MatRowMapType *Ti = Ti_h.data(); in MatSeqAIJKokkosGenerateTransposeStructure()
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx356 const MatRowMapType *Ai = Ai_h.data(), *Bi = Bi_h.data(); in MatMPIAIJKokkosReduceBegin()
476 MatRowMapType *Fdi, *Foi; in MatMPIAIJKokkosReduceBegin()
544 MatRowMapType *Fdjmap = Fdjmap_h.data(), *Fojmap = Fojmap_h.data(); in MatMPIAIJKokkosReduceBegin()
545 MatRowMapType *Fdjperm = Fdjperm_h.data(), *Fojperm = Fojperm_h.data(); in MatMPIAIJKokkosReduceBegin()
693 …kos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, Fdnz), KOKKOS_LAMBDA(const MatRowMapType i) { in MatMPIAIJKokkosReduceEnd()
695 for (MatRowMapType k = Fdjmap(i); k < Fdjmap(i + 1); k++) sum += rootBuf(Fdjperm(k)); in MatMPIAIJKokkosReduceEnd()
700 …kos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, Fonz), KOKKOS_LAMBDA(const MatRowMapType i) { in MatMPIAIJKokkosReduceEnd()
702 for (MatRowMapType k = Fojmap(i); k < Fojmap(i + 1); k++) sum += rootBuf(Fojperm(k)); in MatMPIAIJKokkosReduceEnd()
863 MatRowMapType *Fdi = Fdi_h.data(), *Foi = Foi_h.data(); in MatMPIAIJKokkosBcastBegin()