Home
last modified time | relevance | path

Searched refs:cstart (Results 1 – 25 of 32) sorted by relevance

12

/petsc/src/mat/tutorials/
H A Dex10.c18 PetscInt n, cstart, cend; in main() local
55 PetscCall(MatGetOwnershipRangeColumn(A, &cstart, &cend)); in main()
58 PetscCall(PetscRealView(cend - cstart, norms + cstart, PETSC_VIEWER_STDOUT_WORLD)); in main()
/petsc/src/mat/tests/
H A Dex151.c32 PetscInt i, rstart, rend, cstart, cend; in main() local
45 PetscCall(MatGetOwnershipRangeColumn(A, &cstart, &cend)); in main()
53 …PetscCall(ISCreateGeneral(PETSC_COMM_WORLD, cend - cstart, ixcol + cstart, PETSC_USE_POINTER, &isc… in main()
H A Dex194.c8 PetscInt i, j, m = 3, n = 2, rstart, rend, cstart, cend; in main() local
45 PetscCall(MatGetOwnershipRangeColumn(C, &cstart, &cend)); in main()
48 PetscCall(ISCreateStride(PETSC_COMM_WORLD, cend - cstart, cstart, 1, &iscol)); in main()
H A Dex211.c15 PetscInt i, cstart, cend, Bn = B->cmap->N; in ISGetSeqIS_SameColDist_Private() local
36 PetscCall(MatGetOwnershipRangeColumn(mat, &cstart, &cend)); in ISGetSeqIS_SameColDist_Private()
43 xarray[is_idx[i] - cstart] = (PetscScalar)is_idx[i]; in ISGetSeqIS_SameColDist_Private()
44 cmaparray[is_idx[i] - cstart] = (PetscScalar)(i + isstart); /* global index of iscol[i] */ in ISGetSeqIS_SameColDist_Private()
45 idx[i] = is_idx[i] - cstart; /* local index of iscol[i] */ in ISGetSeqIS_SameColDist_Private()
H A Dex123.c20 PetscInt N = 6, m = 8, M, rstart, cstart, i; in main() local
49 PetscCall(PetscLayoutGetRange(cmap, &cstart, NULL)); in main()
77 for (i = 0; i < n1; i++) j1[i] += cstart; in main()
78 for (i = 0; i < n2; i++) j2[i] += cstart; in main()
80 for (i = 0; i < n1; i++) j1[i] += cstart + m; in main()
81 for (i = 0; i < n2; i++) j2[i] += cstart + m; in main()
195 coo_j[nnz] = jA[j] + cstart; in main()
H A Dex166.c32 PetscInt i, rstart, rend, cstart, cend; in main() local
45 PetscCall(MatGetOwnershipRangeColumn(A, &cstart, &cend)); in main()
H A Dex114.c208 …PetscInt bs = 2, *imaxabsB, *imaxabsB2, rstart, rend, cstart, cend, ncols, col, Brows[2]… in main() local
234 PetscCall(MatGetOwnershipRangeColumn(A, &cstart, &cend)); in main()
236 … PetscCall(MatSetSizes(B, bs * (rend - rstart), bs * (cend - cstart), PETSC_DECIDE, PETSC_DECIDE)); in main()
H A Dex1.c12 PetscInt cstart, cend; in createMatsAndVecs() local
46 PetscCall(MatGetOwnershipRangeColumn(mat, &cstart, &cend)); in createMatsAndVecs()
53 for (PetscInt j = cstart; j < cend; j++) { in createMatsAndVecs()
/petsc/src/mat/impls/aij/mpi/
H A Dfdmpiaij.c34 PetscInt k, cstart, cend, l, row, col, nz, spidx, i, j; in MatFDColoringApply_BAIJ() local
93 PetscCall(VecGetOwnershipRange(x1, &cstart, &cend)); /* used by ghosted vscale */ in MatFDColoringApply_BAIJ()
107 …if (ctype == IS_COLORING_GLOBAL) w3_array -= cstart; /* shift pointer so global index can be used … in MatFDColoringApply_BAIJ()
115 vscale_array -= cstart; /* shift pointer so global index can be used */ in MatFDColoringApply_BAIJ()
121 vscale_array += cstart; in MatFDColoringApply_BAIJ()
123 if (ctype == IS_COLORING_GLOBAL) w3_array += cstart; in MatFDColoringApply_BAIJ()
185 PetscInt k, cstart, cend, l, row, col, nz; in MatFDColoringApply_AIJ() local
243 PetscCall(VecGetOwnershipRange(x1, &cstart, &cend)); /* used by ghosted vscale */ in MatFDColoringApply_AIJ()
265 …if (ctype == IS_COLORING_GLOBAL) w3_array -= cstart; /* shift pointer so global index can be used … in MatFDColoringApply_AIJ()
272 vscale_array -= cstart; /* shift pointer so global index can be used */ in MatFDColoringApply_AIJ()
[all …]
H A Dmmaij.c209 PetscInt i, j, n, nt, cstart, cend, no, *garray = ina->garray, *lindices, bs = inA->rmap->bs; in MatMPIAIJDiagonalScaleLocalSetUp() local
213 PetscCall(MatGetOwnershipRange(inA, &cstart, &cend)); in MatMPIAIJDiagonalScaleLocalSetUp()
218 … if (inA->rmap->mapping->indices[i] * bs >= cstart && inA->rmap->mapping->indices[i] * bs < cend) { in MatMPIAIJDiagonalScaleLocalSetUp()
227 for (j = 0; j < bs; j++) auglyrmapd[(r_rmapd[i] - 1) * bs + j - cstart] = i * bs + j; in MatMPIAIJDiagonalScaleLocalSetUp()
H A Dmpiov.c284 PetscInt is_id, tnz, an, bn, rstart, cstart, row, start, end, col, totalrows, *sbdata; in MatIncreaseOverlap_MPIAIJ_Send_Scalable() local
305 PetscCall(PetscLayoutGetRange(cmap, &cstart, NULL)); in MatIncreaseOverlap_MPIAIJ_Send_Scalable()
331 col = aj[k] + cstart; in MatIncreaseOverlap_MPIAIJ_Send_Scalable()
371 col = aj[k] + cstart; in MatIncreaseOverlap_MPIAIJ_Send_Scalable()
405 PetscInt tnz, an, bn, i, j, row, start, end, rstart, cstart, col, k, *indices_temp; in MatIncreaseOverlap_MPIAIJ_Local_Scalable() local
425 PetscCall(PetscLayoutGetRange(cmap, &cstart, NULL)); in MatIncreaseOverlap_MPIAIJ_Local_Scalable()
446 col = aj[k] + cstart; in MatIncreaseOverlap_MPIAIJ_Local_Scalable()
855 PetscInt start, end, val, max, rstart, cstart, *ai, *aj; in MatIncreaseOverlap_MPIAIJ_Local() local
868 cstart = C->cmap->rstart; in MatIncreaseOverlap_MPIAIJ_Local()
907 val = aj[k] + cstart; in MatIncreaseOverlap_MPIAIJ_Local()
[all …]
H A Dmpiaij.c534 PetscInt cstart = mat->cmap->rstart, cend = mat->cmap->rend, row, col; in MatSetValues_MPIAIJ() local
576 if (in[j] >= cstart && in[j] < cend) { in MatSetValues_MPIAIJ()
577 col = in[j] - cstart; in MatSetValues_MPIAIJ()
649 PetscInt cstart = mat->cmap->rstart, cend = mat->cmap->rend, col; in MatSetValues_MPIAIJ_CopyFromCSRFormat_Symbolic() local
663 if (mat_j[col] >= cstart && mat_j[col] < cend) { in MatSetValues_MPIAIJ_CopyFromCSRFormat_Symbolic()
664 aj[diag_so_far++] = mat_j[col] - cstart; in MatSetValues_MPIAIJ_CopyFromCSRFormat_Symbolic()
692 PetscInt cstart = mat->cmap->rstart, cend = mat->cmap->rend; in MatSetValues_MPIAIJ_CopyFromCSRFormat() local
709 if (mat_j[col] >= cstart && mat_j[col] < cend) { in MatSetValues_MPIAIJ_CopyFromCSRFormat()
710 aj[rowstart_diag + dnz_row] = mat_j[col] - cstart; in MatSetValues_MPIAIJ_CopyFromCSRFormat()
729 PetscInt cstart = mat->cmap->rstart, cend = mat->cmap->rend, row, col; in MatGetValues_MPIAIJ() local
[all …]
/petsc/src/mat/impls/baij/mpi/
H A Dbaijov.c348 PetscInt start, end, val, max, rstart, cstart, *ai, *aj; in MatIncreaseOverlap_MPIBAIJ_Local() local
354 cstart = c->cstartbs; in MatIncreaseOverlap_MPIBAIJ_Local()
370 val = aj[k] + cstart; in MatIncreaseOverlap_MPIBAIJ_Local()
409 PetscInt rstart, cstart, *ai, *aj, *bi, *bj, *garray, i, j, k; in MatIncreaseOverlap_MPIBAIJ_Receive() local
418 cstart = c->cstartbs; in MatIncreaseOverlap_MPIBAIJ_Receive()
475 val = aj[l] + cstart; in MatIncreaseOverlap_MPIBAIJ_Receive()
646 PetscInt cstart = c->cstartbs, *bmap = c->garray; local
963 if ((ctmp = bmap[cworkB[l]]) < cstart) cols[l] = ctmp;
967 for (l = 0; l < nzA; l++) cols[imark + l] = cstart + cworkA[l];
1033 PetscCall(PetscHMapIGetWithDefault(cmap_i, cstart + cworkA[k] + 1, 0, &tt));
[all …]
H A Dmmbaij.c203 PetscInt bs = inA->rmap->bs, i, n, nt, j, cstart, cend, no, *garray = ina->garray, *lindices; in MatMPIBAIJDiagonalScaleLocalSetUp() local
207 PetscCall(MatGetOwnershipRange(inA, &cstart, &cend)); in MatMPIBAIJDiagonalScaleLocalSetUp()
212 … if (inA->rmap->mapping->indices[i] * bs >= cstart && inA->rmap->mapping->indices[i] * bs < cend) { in MatMPIBAIJDiagonalScaleLocalSetUp()
221 for (j = 0; j < bs; j++) uglyrmapd[(r_rmapd[i] - 1) * bs + j - cstart] = i * bs + j; in MatMPIBAIJDiagonalScaleLocalSetUp()
H A Dmpibaij.c414 PetscInt rend = baij->rendbs, cstart = baij->cstartbs, stepval; in MatSetValuesBlocked_MPIBAIJ() local
450 if (in[j] >= cstart && in[j] < cend) { in MatSetValuesBlocked_MPIBAIJ()
451 col = in[j] - cstart; in MatSetValuesBlocked_MPIBAIJ()
806 PetscInt cstart = baij->cstartbs, *garray = baij->garray, row, col, Nbs = baij->Nbs; in MatCreateHashTable_MPIBAIJ_Private() local
829 col = aj[j] + cstart; in MatCreateHashTable_MPIBAIJ_Private()
1305 PetscInt *cmap, *idx_p, cstart = mat->cstartbs; in MatGetRow_MPIBAIJ() local
1350 if (cmap[cworkB[i] / bs] < cstart) v_p[i] = vworkB[i]; in MatGetRow_MPIBAIJ()
1363 … if (cmap[cworkB[i] / bs] < cstart) idx_p[i] = cmap[cworkB[i] / bs] * bs + cworkB[i] % bs; in MatGetRow_MPIBAIJ()
1368 for (i = 0; i < nzA; i++) idx_p[imark + i] = cstart * bs + cworkA[i]; in MatGetRow_MPIBAIJ()
2547 PetscInt m, rstart, cstart, cend; in MatMPIBAIJSetPreallocationCSR_MPIBAIJ() local
[all …]
/petsc/src/mat/impls/sell/mpi/
H A Dmmsell.c203 PetscInt i, n, nt, cstart, cend, no, *garray = ina->garray, *lindices; in MatMPISELLDiagonalScaleLocalSetUp() local
207 PetscCall(MatGetOwnershipRange(inA, &cstart, &cend)); in MatMPISELLDiagonalScaleLocalSetUp()
212 if (inA->rmap->mapping->indices[i] >= cstart && inA->rmap->mapping->indices[i] < cend) { in MatMPISELLDiagonalScaleLocalSetUp()
220 if (r_rmapd[i]) auglyrmapd[(r_rmapd[i] - 1) - cstart] = i; in MatMPISELLDiagonalScaleLocalSetUp()
H A Dmpisell.c155 PetscInt cstart = mat->cmap->rstart, cend = mat->cmap->rend, row, col; in MatSetValues_MPISELL() local
192 if (in[j] >= cstart && in[j] < cend) { in MatSetValues_MPISELL()
193 col = in[j] - cstart; in MatSetValues_MPISELL()
252 PetscInt cstart = mat->cmap->rstart, cend = mat->cmap->rend, row, col; in MatGetValues_MPISELL() local
263 if (idxn[j] >= cstart && idxn[j] < cend) { in MatGetValues_MPISELL()
264 col = idxn[j] - cstart; in MatGetValues_MPISELL()
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx342 …educeBegin(MPI_Comm comm, KokkosCsrMatrix A, KokkosCsrMatrix B, PetscInt cstart, PetscInt cend, co… in MatMPIAIJKokkosReduceBegin() argument
373 if (garray1[*it] < cstart) { // map local to global in MatMPIAIJKokkosReduceBegin()
456 buf[j] = Aj[Ai[i] + j - nzLeft] + cstart; // diag A, also in global in MatMPIAIJKokkosReduceBegin()
515 if (curColIdx >= cstart && curColIdx < cend) { in MatMPIAIJKokkosReduceBegin()
573 if (curColIdx >= cstart && curColIdx < cend) { in MatMPIAIJKokkosReduceBegin()
574 Fdj[Fdnz] = curColIdx - cstart; // easily convert to local in MatMPIAIJKokkosReduceBegin()
617 PetscCallCXX(mm->Fd = KokkosCsrMatrix("Fd", Fm, cend - cstart, Fdnz, Fda_d, Fdi_d, Fdj_d)); in MatMPIAIJKokkosReduceBegin()
674 …sReduceEnd(MPI_Comm comm, KokkosCsrMatrix A, KokkosCsrMatrix B, PetscInt cstart, PetscInt cend, co… in MatMPIAIJKokkosReduceEnd() argument
753 PetscInt cstart, cend; in MatMPIAIJKokkosBcastBegin() local
756 PetscCall(MatGetOwnershipRangeColumn(E, &cstart, &cend)); in MatMPIAIJKokkosBcastBegin()
[all …]
/petsc/src/dm/impls/network/
H A Dnetwork.c2325 …e MatSetDenseblock_private(PetscInt nrows, PetscInt *rows, PetscInt ncols, PetscInt cstart, Mat *J) in MatSetDenseblock_private() argument
2332 for (j = 0; j < ncols; j++) cols[j] = j + cstart; in MatSetDenseblock_private()
2338 …tUserblock_private(Mat Ju, PetscInt nrows, PetscInt *rows, PetscInt ncols, PetscInt cstart, Mat *J) in MatSetUserblock_private() argument
2351 col = cols[j] + cstart; in MatSetUserblock_private()
2359 …atSetblock_private(Mat Ju, PetscInt nrows, PetscInt *rows, PetscInt ncols, PetscInt cstart, Mat *J) in MatSetblock_private() argument
2363 PetscCall(MatSetUserblock_private(Ju, nrows, rows, ncols, cstart, J)); in MatSetblock_private()
2365 PetscCall(MatSetDenseblock_private(nrows, rows, ncols, cstart, J)); in MatSetblock_private()
2466 PetscInt cstart, ncols, j, e, v; in DMCreateMatrix_Network() local
2534 cstart = rstart; in DMCreateMatrix_Network()
2566 PetscCall(DMNetworkGetGlobalVecOffset(dm, edges[e], ALL_COMPONENTS, &cstart)); in DMCreateMatrix_Network()
[all …]
/petsc/src/tao/leastsquares/impls/brgn/
H A Dbrgn.c168 PetscInt i, n, cstart, cend; in GNComputeHessian() local
207 PetscCall(MatGetOwnershipRangeColumn(gn->parent->ls_jac, &cstart, &cend)); in GNComputeHessian()
209 for (i = 0; i < cend - cstart; i++) diag_ary[i] = cnorms[cstart + i] * cnorms[cstart + i]; in GNComputeHessian()
/petsc/src/mat/impls/hypre/
H A Dmhypre.c85 HYPRE_Int rstart, rend, cstart, cend; in MatHYPRE_CreateFromMat() local
92 cstart = (HYPRE_Int)A->cmap->rstart; in MatHYPRE_CreateFromMat()
99 PetscCallHYPRE(HYPRE_IJMatrixCreate(hA->comm, rstart, rend - 1, cstart, cend - 1, &hA->ij)); in MatHYPRE_CreateFromMat()
191 PetscInt i, *garray = pA->garray, *jj, cstart, *pjj; in MatHYPRE_IJMatrixCopyIJ_MPIAIJ() local
202 PetscCall(MatGetOwnershipRange(A, &cstart, NULL)); in MatHYPRE_IJMatrixCopyIJ_MPIAIJ()
221 for (i = 0; i < pdiag->nz; i++) hjj[i] = cstart + pjj[i]; in MatHYPRE_IJMatrixCopyIJ_MPIAIJ()
1749 PetscInt rstart, rend, cstart, cend, M, N; local
1766 cstart = hypre_ParCSRMatrixFirstColDiag(parcsr);
1773 PetscCall(MatSetSizes(T, PetscMax(rend - rstart + 1, 0), PetscMax(cend - cstart + 1, 0), M, N));
1778 PetscCallHYPRE(HYPRE_IJMatrixCreate(hA->comm, rstart, rend, cstart, cend, &hA->ij));
/petsc/src/binding/petsc4py/test/
H A Dtest_dmplex.py336 cstart, cend = plex.getHeightStratum(0)
345 self.assertTrue((newcend - newcstart) == 3 * (cend - cstart))
/petsc/src/binding/petsc4py/src/petsc4py/PETSc/
H A DDMLabel.pyx545 cdef PetscInt cstart = asInt(start), cend = asInt(end)
546 CHKERR(DMLabelFilter(self.dmlabel, cstart, cend))
/petsc/src/mat/impls/sbaij/mpi/
H A Dmpisbaij.c571 PetscInt rend = baij->rendbs, cstart = baij->cstartbs, stepval; in MatSetValuesBlocked_MPISBAIJ() local
612 if (in[j] >= cstart && in[j] < cend) { in MatSetValuesBlocked_MPISBAIJ()
613 col = in[j] - cstart; in MatSetValuesBlocked_MPISBAIJ()
1177 PetscInt *cmap, *idx_p, cstart = mat->rstartbs; in MatGetRow_MPISBAIJ() local
1224 if (cmap[cworkB[i] / bs] < cstart) v_p[i] = vworkB[i]; in MatGetRow_MPISBAIJ()
1237 … if (cmap[cworkB[i] / bs] < cstart) idx_p[i] = cmap[cworkB[i] / bs] * bs + cworkB[i] % bs; in MatGetRow_MPISBAIJ()
1242 for (i = 0; i < nzA; i++) idx_p[imark + i] = cstart * bs + cworkA[i]; in MatGetRow_MPISBAIJ()
/petsc/src/mat/impls/maij/
H A Dmaij.c927 PetscInt rstart, cstart, *garray, ii, k; in MatConvert_MPIMAIJ_MPIAIJ() local
949 cstart = dof * maij->A->cmap->rstart; in MatConvert_MPIMAIJ_MPIAIJ()
957 for (k = 0; k < ncols; k++) icols[k] = cstart + dof * cols[k] + j; in MatConvert_MPIMAIJ_MPIAIJ()

12