Searched refs:MatColIdxType (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.hpp | 19 using MatColIdxType = PetscInt; typedef
23 using KokkosCsrMatrixType = typename KokkosSparse::CrsMatrix<MatScalarType, MatColIdxType, MemorySp…
50 using MatColIdxKokkosDualView = Kokkos::DualView<MatColIdxType *>;
53 using KernelHandle = KokkosKernels::Experimental::KokkosKernelsHandle<MatRowMapType, MatColIdxType,…
173 MatColIdxType *j_host_data() { return j_dual.view_host().data(); } in j_host_data()
177 MatColIdxType *j_device_data() { return j_dual.view_device().data(); } in j_device_data()
179 MatColIdxType nrows() { return csrmat.numRows(); } in nrows()
180 MatColIdxType ncols() { return csrmat.numCols(); } in ncols()
184 …void SetColSize(MatColIdxType n) { csrmat = KokkosCsrMatrix("csrmat", n, a_dual.view_device(), csr… in SetColSize()
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 357 const MatColIdxType *Aj = Aj_h.data(), *Bj = Bj_h.data(); in MatMPIAIJKokkosReduceBegin()
543 MatColIdxType *Fdj = Fdj_h.data(), *Foj = Foj_h.data(); in MatMPIAIJKokkosReduceBegin()
864 MatColIdxType *F_NzLeft = F_NzLeft_h.data(); in MatMPIAIJKokkosBcastBegin()
886 MatColIdxType *Fdj = Fdj_h.data(), *Foj = Foj_h.data(), gid; in MatMPIAIJKokkosBcastBegin()
|