Home
last modified time | relevance | path

Searched refs:rstart (Results 1 – 25 of 198) sorted by relevance

12345678

/petsc/src/mat/tests/
H A Dex56.c8 PetscInt bs = 3, m = 4, n = 6, i, j, val = 10, row[2], col[3], eval, rstart; in main() local
29 if (flg && (size != 1)) rstart = m * ((rank + 1) % size); in main()
30 else rstart = m * (rank); in main()
32 row[0] = rstart + 0; in main()
33 row[1] = rstart + 2; in main()
34 col[0] = rstart + 0; in main()
35 col[1] = rstart + 1; in main()
36 col[2] = rstart + 3; in main()
63 col[0] = rstart * bs + 0; in main()
64 col[1] = rstart * bs + 1; in main()
[all …]
H A Dex133.c8 PetscInt bs = 3, m = 4, i, j, val = 10, row[2], col[3], rstart; in main() local
18 rstart = 0; in main()
20 row[0] = rstart + 0; in main()
21 row[1] = rstart + 2; in main()
22 col[0] = rstart + 0; in main()
23 col[1] = rstart + 1; in main()
24 col[2] = rstart + 3; in main()
H A Dex59.c8 PetscInt i, j, m = 3, n = 2, rstart, rend; in main() local
31 PetscCall(MatGetOwnershipRange(C, &rstart, &rend)); in main()
32 for (i = rstart; i < rend; i++) { in main()
48 PetscCall(MatGetOwnershipRange(C, &rstart, &rend)); in main()
50 PetscCall(ISCreateStride(PETSC_COMM_WORLD, (rend - rstart) / 2, rstart, 2, &isrow)); in main()
52 PetscCall(ISCreateStride(PETSC_COMM_WORLD, (rend - rstart) / 2, rstart, 2, &iscol)); in main()
H A Dex181.c8 PetscInt i, j, m = 3, n = 2, rstart, rend; in main() local
30 PetscCall(MatGetOwnershipRange(C, &rstart, &rend)); in main()
31 for (i = rstart; i < rend; i++) { in main()
46 PetscCall(MatGetOwnershipRange(C, &rstart, &rend)); in main()
50 PetscCall(ISCreateStride(PETSC_COMM_WORLD, 1, rstart, 1, &isrow)); in main()
52 PetscCall(ISCreateStride(PETSC_COMM_WORLD, rend - rstart, rstart, 1, &isrow)); in main()
H A Dex194.c8 PetscInt i, j, m = 3, n = 2, rstart, rend, cstart, cend; in main() local
28 PetscCall(MatGetOwnershipRange(C, &rstart, &rend)); in main()
29 for (i = rstart; i < rend; i++) { in main()
44 PetscCall(MatGetOwnershipRange(C, &rstart, &rend)); in main()
47 …PetscCall(ISCreateStride(PETSC_COMM_WORLD, rend - rstart > 0 ? rend - rstart - 1 : 0, rstart, 1, &… in main()
H A Dex225.c104 PetscInt rstart, rend, nzA, nzB; in main() local
110 PetscCall(MatGetOwnershipRange(A, &rstart, &rend)); in main()
111 for (i = rstart; i < rend; i++) { in main()
124 PetscCall(MatGetOwnershipRange(A, &rstart, &rend)); in main()
125 …PetscCall(PetscCalloc3((rend - rstart) * 6, &valuesA, (rend - rstart) * 6, &valuesB, rend - rstart… in main()
126 for (i = rstart; i < rend; i++) rows[i - rstart] = i; in main()
128 PetscCall(MatGetValues(A, rend - rstart, rows, 6, cols, valuesA)); in main()
129 PetscCall(MatGetValues(B, rend - rstart, rows, 6, cols, valuesB)); in main()
131 for (i = 0; i < (rend - rstart); i++) { in main()
133 … PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Error MatGetValues %" PetscInt_FMT, i + rstart); in main()
H A Dex58.c8 PetscInt m = 7, n, i, rstart, rend, cols[3]; in main() local
22 PetscCall(MatGetOwnershipRange(A, &rstart, &rend)); in main()
23 if (!rstart) { in main()
28 PetscCall(MatSetValues(A, 1, &rstart, 2, cols, v, INSERT_VALUES)); in main()
29 rstart++; in main()
42 for (i = rstart; i < rend; i++) { in main()
H A Dex211.c11 PetscInt ncols, isstart, *idx, m, rstart, count; in ISGetSeqIS_SameColDist_Private() local
58 rstart = mat->rmap->rstart; in ISGetSeqIS_SameColDist_Private()
61 for (i = 0; i < m; i++) idx[i] = is_idx[i] - rstart; in ISGetSeqIS_SameColDist_Private()
128 PetscInt i, j, m = 3, n = 2, rstart, rend; in main() local
148 PetscCall(MatGetOwnershipRange(C, &rstart, &rend)); in main()
149 for (i = rstart; i < rend; i++) { in main()
163 PetscCall(MatGetOwnershipRange(C, &rstart, &rend)); in main()
165 PetscCall(ISCreateStride(PETSC_COMM_WORLD, (rend - rstart) / 2, rstart, 2, &isrow)); in main()
167 PetscCall(ISCreateStride(PETSC_COMM_WORLD, (rend - rstart) / 2, rstart, 2, &iscol)); in main()
H A Dex73.c135 PetscInt rstart, i, *nzd, *nzo, nzl, nzmax = 0, *ncols, nrow, j; in main() local
141 PetscCall(MatGetOwnershipRange(B, &rstart, NULL)); in main()
144 PetscCall(MatGetRow(B, i + rstart, &nzl, &cols, NULL)); in main()
146 if (cols[j] >= rstart && cols[j] < rstart + n) { in main()
155 PetscCall(MatRestoreRow(B, i + rstart, &nzl, &cols, NULL)); in main()
163 PetscCall(MatGetRow(B, i + rstart, &nzl, &cols, &vals)); in main()
168 nrow = 2 * (i + rstart); in main()
170 nrow = 2 * (i + rstart) + 1; in main()
172 PetscCall(MatRestoreRow(B, i + rstart, &nzl, &cols, &vals)); in main()
H A Dex139.c27 …PetscInt i, j, k, l, rstart, rend, m, n, top_bs, row_bs, col_bs, nlocblocks, *idx, n… in main() local
60 PetscCall(MatGetOwnershipRange(J, &rstart, &rend)); in main()
62 nlocblocks = (rend - rstart) / top_bs + 2; in main()
64 for (i = 0; i < nlocblocks; i++) idx[i] = (rstart / top_bs + i - 1 + m / top_bs) % (m / top_bs); in main()
73 nrowblocks = (rend - rstart) / row_bs; in main()
74 ncolblocks = (rend - rstart) / col_bs; in main()
76 for (i = 0; i < nrowblocks; i++) ridx[i] = i + ((i > nrowblocks / 2) ^ !rstart); in main()
77 for (i = 0; i < ncolblocks; i++) cidx[i] = i + ((i < ncolblocks / 2) ^ !!rstart); in main()
H A Dex16.c9 PetscInt M, N, rstart, rend, lda, i, j; in CheckValues() local
15 PetscCall(MatGetOwnershipRange(A, &rstart, &rend)); in CheckValues()
16 for (i = rstart; i < rend; i++) { in CheckValues()
18 PetscInt ii = i - rstart, jj = j; in CheckValues()
34 PetscInt i, j, M = 4, N = 3, rstart, rend; in main() local
88 PetscCall(MatGetOwnershipRange(A, &rstart, &rend)); in main()
89 PetscCall(PetscMalloc1((rend - rstart) * N, &array)); in main()
90 for (i = 0; i < (rend - rstart) * N; i++) array[i] = (PetscReal)1; in main()
H A Dex44.c15 PetscInt M, N, rstart, rend, i, j; in CheckValuesAIJ() local
21 PetscCall(MatGetOwnershipRange(A, &rstart, &rend)); in CheckValuesAIJ()
22 for (i = rstart; i < rend; i++) { in CheckValuesAIJ()
37 PetscInt rstart, rend, i, j; in main() local
50 PetscCall(MatGetOwnershipRange(A, &rstart, &rend)); in main()
51 for (i = rstart; i < rend; i++) { in main()
H A Dex45.c15 PetscInt M, N, rstart, rend, i, j; in CheckValuesAIJ() local
21 PetscCall(MatGetOwnershipRange(A, &rstart, &rend)); in CheckValuesAIJ()
22 for (i = rstart; i < rend; i++) { in CheckValuesAIJ()
37 PetscInt rstart, rend, i, j; in main() local
51 PetscCall(MatGetOwnershipRange(A, &rstart, &rend)); in main()
52 for (i = rstart; i < rend; i++) { in main()
H A Dex183.c29 PetscInt m, n, N, bs, rstart, rend, i, j, k, total_subdomains, hash, nsubdomains = 1; in main() local
72 PetscCall(MatGetOwnershipRange(A, &rstart, &rend)); in main()
74 for (i = rstart; i < rend; i++) { in main()
95 PetscCall(MatGetOwnershipRange(A, &rstart, &rend)); in main()
97 PetscCall(PetscMalloc2(rend - rstart, &rowindices, rend - rstart, &colindices)); in main()
99 for (j = rstart; j < rend; ++j) { in main()
105 rowindices[j - rstart] = idx % N; in main()
106 colindices[j - rstart] = (idx + m) % N; in main()
108 PetscCall(ISCreateGeneral(subcomm, rend - rstart, rowindices, PETSC_COPY_VALUES, &rowis[0])); in main()
109 PetscCall(ISCreateGeneral(subcomm, rend - rstart, colindices, PETSC_COPY_VALUES, &colis[0])); in main()
H A Dex236.c9 PetscInt i, j, k, n = 3, vstart, rstart, rend, margin; in main() local
23 PetscCall(MatGetOwnershipRange(A, &rstart, &rend)); in main()
41 for (i = rstart; i < rend - margin; i++) { in main()
42 for (j = rstart; j < rend; j++) { in main()
/petsc/src/dm/impls/composite/
H A Dpackm.c64 PetscInt rstart, rend, *indices; in DMCreateMatrix_Composite_AIJ() local
71 PetscCall(MatGetOwnershipRange(*J, &rstart, &rend)); in DMCreateMatrix_Composite_AIJ()
75 …for (i = rstart; i < rend; i++) PetscCall(MatSetValues(*J, 1, &i, mA, indices, values, INSERT_VALU… in DMCreateMatrix_Composite_AIJ()
88 PetscInt nc, rstart, *ccols, maxnc; in DMCreateMatrix_Composite_AIJ() local
98 PetscCall(MatGetOwnershipRange(Atmp, &rstart, NULL)); in DMCreateMatrix_Composite_AIJ()
104 PetscCall(MatGetRow(Atmp, rstart + i, &nc, NULL, NULL)); in DMCreateMatrix_Composite_AIJ()
106 PetscCall(MatRestoreRow(Atmp, rstart + i, &nc, NULL, NULL)); in DMCreateMatrix_Composite_AIJ()
110 PetscCall(MatGetRow(Atmp, rstart + i, &nc, &cols, NULL)); in DMCreateMatrix_Composite_AIJ()
117 PetscCall(MatPreallocateSet(com->rstart + next->rstart + i, nc, ccols, dnz, onz)); in DMCreateMatrix_Composite_AIJ()
118 PetscCall(MatRestoreRow(Atmp, rstart + i, &nc, &cols, NULL)); in DMCreateMatrix_Composite_AIJ()
[all …]
/petsc/src/vec/is/is/tests/
H A Dex6.c9 PetscInt i, j, n, cnt = 0, rstart, rend; in main() local
19 rstart = 3 * (size + 2) * rank; /* start of local range */ in main()
28 PetscCall(PetscCalloc1(rend - rstart, &mask)); in main()
32 PetscCall(ISGeneralSetIndicesFromMask(is[0], rstart, rend, mask)); in main()
40 for (j = 0; j < size; j++) indices[cnt++] = rstart + i * (size + 2) + j; in main()
50 PetscCall(ISComplement(is[0], rstart, rend, &isc)); in main()
/petsc/src/mat/graphops/color/impls/minpack/
H A Dcolor.c38 PetscInt bs = 1, rstart, rend, N_loc, nc; in MatColoringApply_SL() local
87 rstart = mat->rmap->rstart / bs; in MatColoringApply_SL()
89 N_loc = rend - rstart; /* number of local nodes */ in MatColoringApply_SL()
93 for (i = rstart; i < rend; i++) colors_loc[i - rstart] = iscoloring_seq->colors[i]; in MatColoringApply_SL()
138 PetscInt bs = 1, rstart, rend, N_loc, nc; in MatColoringApply_LF() local
189 rstart = mat->rmap->rstart / bs; in MatColoringApply_LF()
191 N_loc = rend - rstart; /* number of local nodes */ in MatColoringApply_LF()
195 for (i = rstart; i < rend; i++) colors_loc[i - rstart] = iscoloring_seq->colors[i]; in MatColoringApply_LF()
241 PetscInt bs = 1, rstart, rend, N_loc, nc; in MatColoringApply_ID() local
291 rstart = mat->rmap->rstart / bs; in MatColoringApply_ID()
[all …]
/petsc/src/vec/is/sf/tests/
H A Dex23.c14 PetscInt rstart, rend; in main() local
24 PetscCall(VecGetOwnershipRange(x, &rstart, &rend)); in main()
25 PetscCall(ISCreateStride(PETSC_COMM_WORLD, n, rstart, 1, &ix)); in main()
27 for (PetscInt i = rstart; i < rend; i++) indices[i - rstart] = i / 2; in main()
42 xh[i] = xh[i + n] = i + rstart; in main()
43 yh[i] = yh[i + n] = i + rstart; in main()
H A Dex22.c13 PetscInt rstart, rend; in main() local
26 PetscCall(VecGetOwnershipRange(x, &rstart, &rend)); in main()
27 PetscCall(ISCreateStride(PETSC_COMM_WORLD, n, rstart, 1, &ix)); in main()
29 for (PetscInt i = rstart; i < rend; i++) indices[i - rstart] = i / 4; in main()
/petsc/src/vec/vec/tests/
H A Dex38.c10 PetscInt i, rstart, rend, idxs[3]; in main() local
26 PetscCall(VecGetOwnershipRange(X, &rstart, &rend)); in main()
29 for (i = 0; i < rend - rstart; i++) x[i] = rstart + i; in main()
45 PetscCall(ISCreateStride(comm, (rend - rstart) / 3 + 3 * (rank > size / 2), rstart, 1, &is0)); in main()
46 PetscCall(ISComplement(is0, rstart, rend, &is1)); in main()
/petsc/src/mat/graphops/color/impls/natural/
H A Dnatural.c15 PetscInt rstart, rend, N_loc, nc; in MatColoringApply_Natural() local
46 rstart = mat->rmap->rstart / bs; in MatColoringApply_Natural()
48 N_loc = rend - rstart; /* number of local nodes */ in MatColoringApply_Natural()
52 for (i = rstart; i < rend; i++) colors_loc[i - rstart] = iscoloring_seq->colors[i]; in MatColoringApply_Natural()
/petsc/src/mat/graphops/order/
H A Dsorder.c176 PetscInt i, rstart, rend, *idx; in MatGetOrdering() local
181 PetscCall(MatGetOwnershipRange(mat, &rstart, &rend)); in MatGetOrdering()
184 PetscCall(PetscMalloc1(rend - rstart, &idx)); in MatGetOrdering()
185 for (i = 0; i + rstart < rend; i++) idx[i] = rstart + lidx[i]; in MatGetOrdering()
188 …PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)mat), rend - rstart, idx, PETSC_OWN_POINTER… in MatGetOrdering()
192 PetscCall(PetscMalloc1(rend - rstart, &idx)); in MatGetOrdering()
193 for (i = 0; i + rstart < rend; i++) idx[i] = rstart + lidx[i]; in MatGetOrdering()
196 …PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)mat), rend - rstart, idx, PETSC_OWN_POINTER… in MatGetOrdering()
/petsc/src/ksp/ksp/tests/
H A Dex60.c13 PetscInt M = 8, N = 8, m, n, rstart, rend, r; in main() local
36 PetscCall(MatGetOwnershipRange(A, &rstart, &rend)); in main()
37 for (r = rstart; r < rend; ++r) { in main()
78 PetscCall(PetscMalloc1(rend - rstart, &rows)); in main()
79 for (r = rstart; r < rend; ++r) rows[r - rstart] = r; in main()
80 PetscCall(ISCreateGeneral(PETSC_COMM_SELF, rend - rstart, rows, PETSC_OWN_POINTER, &is)); in main()
/petsc/src/vec/vec/tutorials/
H A Dex8.c14 PetscInt i, ng, *gindices, rstart, rend, M; in main() local
43 PetscCall(VecGetOwnershipRange(x, &rstart, &rend)); in main()
44 ng = rend - rstart + 2; in main()
46 gindices[0] = rstart - 1; in main()

12345678