Home
last modified time | relevance | path

Searched refs:rowOffsets (Results 1 – 5 of 5) sorted by relevance

/petsc/src/ksp/pc/impls/bjacobi/bjkokkos/
H A Dbjkokkoskernels.kokkos.cxx178 if (!jac->rowOffsets) { in PCApply_BJKOKKOSKERNELS()
179 jac->rowOffsets = new IntView("rowOffsets", Nsolves / Nsolves_team, Nloc + 1); // same grids in PCApply_BJKOKKOSKERNELS()
187 IntView &rowOffsets = *jac->rowOffsets; in PCApply_BJKOKKOSKERNELS() local
203 rowOffsets(blkID / Nsolves_team, rowb + 1 - start) = n; // save sizes in PCApply_BJKOKKOSKERNELS()
218 rowOffsets(graphID, 0) = 0; in PCApply_BJKOKKOSKERNELS()
219 for (int i = 0; i < Nloc; ++i) rowOffsets(graphID, i + 1) += rowOffsets(graphID, i); in PCApply_BJKOKKOSKERNELS()
233 …if (fill_idx && blkID % Nsolves_team == 0) colIndices(graphID, rowOffsets(graphID, rowb - start) +… in PCApply_BJKOKKOSKERNELS()
234 batch_values(blkID, rowOffsets(graphID, rowb - start) + i) = val; in PCApply_BJKOKKOSKERNELS()
255 …xValueView, IntView, XYType, KrylovHandleType>(batch_values, inv_diag, rowOffsets, colIndices, bat… in PCApply_BJKOKKOSKERNELS()
H A Dbjkokkos.kokkos.cxx1028 if (jac->rowOffsets) delete jac->rowOffsets; in PCReset_BJKOKKOS()
1033 jac->rowOffsets = NULL; in PCReset_BJKOKKOS()
1217 jac->rowOffsets = NULL; in PCCreate_BJKOKKOS()
/petsc/src/ksp/pc/impls/amgx/
H A Damgx.cxx264 const PetscInt *rowOffsets; in PCSetUp_AMGX() local
265 …PetscCall(MatGetRowIJ(amgx->localA, 0, PETSC_FALSE, PETSC_FALSE, &amgx->nLocalRows, &rowOffsets, &… in PCSetUp_AMGX()
270 …PetscCallCUDA(cudaMemcpy(&amgx->nnz, &rowOffsets[amgx->nLocalRows], sizeof(int), cudaMemcpyDefault… in PCSetUp_AMGX()
272 amgx->nnz = rowOffsets[amgx->nLocalRows]; in PCSetUp_AMGX()
296 …Rows, (int)amgx->nLocalRows, (int)amgx->nnz, amgx->bSize, amgx->bSize, rowOffsets, colIndices, amg… in PCSetUp_AMGX()
302 …PetscCall(MatRestoreRowIJ(amgx->localA, 0, PETSC_FALSE, PETSC_FALSE, &nlr, &rowOffsets, &colIndice… in PCSetUp_AMGX()
/petsc/include/petsc/private/
H A Dpcbjkokkosimpl.h60 IntView *rowOffsets; member
/petsc/src/dm/impls/plex/
H A Dplextree.c2081 …PetscInt *offsets, *newOffsets, *offsetsCopy, *newOffsetsCopy, *rowOffsets, *numD, *num… in DMPlexComputeInterpolatorTree() local
2133 …elds + 1, &newOffsets, maxFields + 1, &newOffsetsCopy, maxFields + 1, &rowOffsets, maxFields + 1, … in DMPlexComputeInterpolatorTree()
2698 rowOffsets[0] = 0; in DMPlexComputeInterpolatorTree()
2707 rowOffsets[f + 1] = pInd[numColIndices + f]; in DMPlexComputeInterpolatorTree()
2751 PetscInt numInRows = rowOffsets[f + 1] - rowOffsets[f]; in DMPlexComputeInterpolatorTree()
2766 PetscInt numInRows = rowOffsets[f + 1] - rowOffsets[f]; in DMPlexComputeInterpolatorTree()
2808 PetscCall(PetscFree7(offsets, offsetsCopy, newOffsets, newOffsetsCopy, rowOffsets, numD, numO)); in DMPlexComputeInterpolatorTree()
3521 PetscInt *offsets, *offsetsCopy, *rowOffsets; in DMPlexComputeInjectorTree() local
3546 PetscCall(PetscMalloc3(maxFields, &offsets, maxFields, &offsetsCopy, maxFields, &rowOffsets)); in DMPlexComputeInjectorTree()
3568 rowOffsets[0] = 0; in DMPlexComputeInjectorTree()
[all …]