Home
last modified time | relevance | path

Searched refs:i2 (Results 1 – 25 of 34) sorted by relevance

12

/petsc/src/mat/impls/baij/seq/
H A Dbaij.c224 PetscInt m = a->mbs, i, i2, nz, bs = A->rmap->bs, bs2 = bs * bs, k, j, idx, it; in MatSOR_SeqBAIJ() local
257 i2 = 1; in MatSOR_SeqBAIJ()
263 s[0] = b[i2]; in MatSOR_SeqBAIJ()
268 t[i2] = s[0]; in MatSOR_SeqBAIJ()
270 x[i2] = xw[0]; in MatSOR_SeqBAIJ()
272 i2 += 1; in MatSOR_SeqBAIJ()
279 i2 = 2; in MatSOR_SeqBAIJ()
285 s[0] = b[i2]; in MatSOR_SeqBAIJ()
286 s[1] = b[i2 + 1]; in MatSOR_SeqBAIJ()
294 t[i2] = s[0]; in MatSOR_SeqBAIJ()
[all …]
/petsc/src/vec/is/is/utils/
H A Disdiff.c35 const PetscInt *i1, *i2; in ISDifference() local
70 PetscCall(ISGetIndices(is2, &i2)); in ISDifference()
73 if (i2[i] < imin || i2[i] > imax) continue; in ISDifference()
74 PetscCall(PetscBTClear(mask, i2[i] - imin)); in ISDifference()
76 PetscCall(ISRestoreIndices(is2, &i2)); in ISDifference()
123 const PetscInt *i1, *i2; in ISSum() local
143 PetscCall(ISGetIndices(is2, &i2)); in ISSum()
153 while (p2 < n2 && i2[p2] < i1[p1]) { in ISSum()
162 if (i2[p2] == i1[p1]) { in ISSum()
174 while (p1 < n1 && i1[p1] < i2[p2]) { in ISSum()
[all …]
/petsc/src/tao/unconstrained/tutorials/
H A Drosenbrock1f.F90109 PetscInt i, nn, ind(0:1), i2
116 i2 = 2
135 PetscCall(MatSetValues(H, i2, ind, i2, ind, reshape(v, [i2*i2]), INSERT_VALUES, ierr))
168 PetscInt i2, i1
176 i2 = 2
197 PetscCallA(MatCreateSeqBAIJ(PETSC_COMM_SELF, i2, n, n, i1, PETSC_NULL_INTEGER_ARRAY, H, ierr))
/petsc/src/snes/tutorials/
H A Dex1f.F9075 PetscInt idx(2), i2
84 i2 = 2
99 PetscCall(MatSetValues(B, i2, idx, i2, idx, A, INSERT_VALUES, ierr))
165 PetscInt its, i2, i20
190 i2 = 2
206 PetscCallA(VecCreateSeq(PETSC_COMM_SELF, i2, x, ierr))
212 PetscCallA(MatSetSizes(J, PETSC_DECIDE, PETSC_DECIDE, i2, i2, ierr))
/petsc/src/dm/impls/plex/tests/
H A Dex2f90.F9015 PetscInt i0, i1, i2, i3, i6, i7
21 i2 = 2
45 PetscCallA(DMPlexSetConeSize(dm, i6, i2, ierr))
46 PetscCallA(DMPlexSetConeSize(dm, i7, i2, ierr))
47 PetscCallA(DMPlexSetConeSize(dm, i8, i2, ierr))
48 PetscCallA(DMPlexSetConeSize(dm, i9, i2, ierr))
49 PetscCallA(DMPlexSetConeSize(dm, i10, i2, ierr))
/petsc/src/mat/impls/aij/seq/bas/
H A Dspbas.c144 static int spbas_row_order_icol(PetscInt i1, PetscInt i2, PetscInt *irow_in, PetscInt *icol_in, Pet… in spbas_row_order_icol() argument
148 PetscInt nnz2 = irow_in[i2 + 1] - irow_in[i2]; in spbas_row_order_icol()
150 PetscInt *icol2 = &icol_in[irow_in[i2]]; in spbas_row_order_icol()
162 if (icol1[j] - i1 < icol2[j] - i2) return -1; in spbas_row_order_icol()
163 if (icol1[j] - i1 > icol2[j] - i2) return 1; in spbas_row_order_icol()
182 PetscInt i, i1, i2; /* Loop counters for (partly) sorted arrays */ in spbas_mergesort_icols() local
211 i2 = istart + istep; in spbas_mergesort_icols()
212 i2end = i2 + istep; in spbas_mergesort_icols()
217 …if (i1 < i1end && i2 < i2end && spbas_row_order_icol(ihlp2[i1], ihlp2[i2], irow_in, icol_in, col_i… in spbas_mergesort_icols()
220 } else if (i2 < i2end) { in spbas_mergesort_icols()
[all …]
/petsc/src/mat/impls/aij/seq/
H A Dinode.c57 PetscInt *tns, *tvc, *ns_row = a->inode.size_csr, *ns_col, nsz, i1, i2; in MatGetRowIJ_SeqAIJ_Inode_Symmetric() local
77 for (i2 = 0; i2 < nsz; ++i2, ++col) tvc[col] = i1; in MatGetRowIJ_SeqAIJ_Inode_Symmetric()
92 i2 = tvc[col]; in MatGetRowIJ_SeqAIJ_Inode_Symmetric()
93 while (i2 < i1 && j < jmax) { /* 1.[-xx-d-xx--] 2.[-xx-------],off-diagonal elements */ in MatGetRowIJ_SeqAIJ_Inode_Symmetric()
95 ia[i2 + 1]++; in MatGetRowIJ_SeqAIJ_Inode_Symmetric()
96 i2++; /* Start col of next node */ in MatGetRowIJ_SeqAIJ_Inode_Symmetric()
97 while ((j < jmax) && ((col = *j + ishift) < tns[i2])) ++j; in MatGetRowIJ_SeqAIJ_Inode_Symmetric()
98 i2 = tvc[col]; in MatGetRowIJ_SeqAIJ_Inode_Symmetric()
100 if (i2 == i1) ia[i2 + 1]++; /* now the diagonal element */ in MatGetRowIJ_SeqAIJ_Inode_Symmetric()
122 i2 = tvc[col]; in MatGetRowIJ_SeqAIJ_Inode_Symmetric()
[all …]
/petsc/src/mat/tests/
H A Dex75.c12 …PetscInt n, col[3], n1, block, row, i, j, i2, j2, Ii, J, rstart, rend, bs = 1, mbs = 16, d_nz =… in main() local
134 PetscCall(MatGetSize(A, &i2, &j2)); in main()
135 i -= i2; in main()
143 PetscCall(MatGetLocalSize(A, &i2, &j2)); in main()
144 i2 -= i; in main()
146 if (i2 || j2) { in main()
183 PetscCall(MatGetOwnershipRange(A, &i2, &j2)); in main()
184 i2 -= rstart; in main()
186 if (i2 || j2) { in main()
H A Dex266.c22 …PetscInt i2[] = {3, 4, 1, 10, 0, 1, 1, 2, 1, 1, 2, 2, 3, 3, 4, 4, 1, 2, 5, 5, 6, 4, 17, 0, 1, 1, … in main() local
33 {i2, j2, PETSC_STATIC_ARRAY_LENGTH(i2)} in main()
H A Dex90.c48 PetscInt i1[] = {0, 3, 5}, i2[] = {0, 2, 5}; in main() local
63 …PetscCall(MatCreateMPIAIJWithArrays(comm, 2, 2, PETSC_DETERMINE, PETSC_DETERMINE, rank ? i2 : i1, … in main()
H A Dex254.c24 …PetscInt i2[] = {3, 4, 1, 10, 0, 1, 1, 2, 1, 1, 2, 2, 3, 3, 4, 4, 1, 2, 5, 5, 6, 4, 17, 0, 1, 1, … in main() local
35 {i2, j2, PETSC_STATIC_ARRAY_LENGTH(i2)} in main()
H A Dex123.c16 PetscInt i2[] = {7, 6, 2, 0, 4, 1, 1, 2, 1, -1, -1}; in main() local
74 for (i = 0; i < n2; i++) i2[i] += rstart; in main()
126 PetscCall(PetscArraycpy(it, i2, n2)); in main()
141 PetscCall(PetscArraycpy(it, i2, n2)); in main()
/petsc/src/mat/impls/kaij/
H A Dkaij.c863 PetscInt i, j, k, i2, bs, bs2, nz; in MatSOR_SeqKAIJ() local
898 i2 = bs; in MatSOR_SeqKAIJ()
910 PetscKernel_w_gets_w_minus_Ar_times_v(bs, bs, w, T, &t[i2]); in MatSOR_SeqKAIJ()
911 for (k = 0; k < bs; k++) t[i2 + k] += b[i2 + k]; in MatSOR_SeqKAIJ()
913 PetscCall(PetscArraycpy(t + i2, b + i2, bs)); in MatSOR_SeqKAIJ()
915 for (k = 0; k < bs; k++) t[i2 + k] -= v[j] * x[vi[j] * bs + k]; in MatSOR_SeqKAIJ()
918 PetscCall(PetscArraycpy(t + i2, b + i2, bs)); in MatSOR_SeqKAIJ()
921 PetscKernel_w_gets_Ar_times_v(bs, bs, t + i2, idiag, y); in MatSOR_SeqKAIJ()
922 for (j = 0; j < bs; j++) x[i2 + j] = omega * y[j]; in MatSOR_SeqKAIJ()
925 i2 += bs; in MatSOR_SeqKAIJ()
[all …]
/petsc/src/ksp/pc/tests/
H A Dex9f.F9015 PetscInt i, n, col(3), its, i1, i2, i3
35 i2 = 2
68 PetscCallA(MatSetValues(A, i1, [i], i2, col, value, INSERT_VALUES, ierr))
74 PetscCallA(MatSetValues(A, i1, [i], i2, col, value, INSERT_VALUES, ierr))
H A Dex5.c277 PetscInt i, n, N, i2; in interpolate() local
287 i2 = 2 * i; in interpolate()
288 y[i2] += .5 * x[i]; in interpolate()
289 y[i2 + 1] += x[i]; in interpolate()
290 y[i2 + 2] += .5 * x[i]; in interpolate()
299 PetscInt i, n, N, i2; in restrct() local
310 i2 = 2 * i; in restrct()
311 b[i] = (r[i2] + 2.0 * r[i2 + 1] + r[i2 + 2]); in restrct()
/petsc/src/ksp/ksp/tutorials/
H A Dex51.c265 PetscInt i1, j1, i2, j2; in Form2DElementMass() local
270 for (i2 = 0; i2 < (P + 1); ++i2) { in Form2DElementMass()
274 indx2 = j2 * (P + 1) + i2; in Form2DElementMass()
275 indx3 = (j2 * (P + 1) + j1) * (P + 1) * (P + 1) + (i2 * (P + 1) + i1); in Form2DElementMass()
289 PetscInt i1, j1, i2, j2; in Form2DElementStiffness() local
294 for (i2 = 0; i2 < (P + 1); ++i2) { in Form2DElementStiffness()
298 indx2 = j2 * (P + 1) + i2; in Form2DElementStiffness()
299 indx3 = (j2 * (P + 1) + j1) * (P + 1) * (P + 1) + (i2 * (P + 1) + i1); in Form2DElementStiffness()
H A Dex1f.F9053 PetscInt i, n, col(3), its, i1, i2, i3
71 i2 = 2
107 PetscCallA(MatSetValues(A, i1, [i], i2, col, value, INSERT_VALUES, ierr))
113 PetscCallA(MatSetValues(A, i1, [i], i2, col, value, INSERT_VALUES, ierr))
/petsc/src/vec/vec/tests/
H A Dex61.c16 …PetscInt i2[] = {7, 7, 8, 8, 9, 16, 17, 9, 10, 1, 1, -2, 2, 3, 3, 14, 4, 5, 10, 13, 9, 9, 10, 1, 0… in main() local
24 {i2, PETSC_STATIC_ARRAY_LENGTH(i2)} in main()
/petsc/src/ksp/ksp/tests/
H A Dex15.c10 PetscInt i, n = 10, col[3], its, i1, i2; in main() local
27 PetscCall(VecGetOwnershipRange(u, &i1, &i2)); in main()
28 for (i = i1; i < i2; i++) { in main()
H A Dex10.c85 PetscInt i, j, k, i1, i2, j_1, j2, k1, k2, h1, h2, shiftx, shifty, shiftz; in GetElasticityMatrix() local
118 for (i2 = 0; i2 < 3; i2++) { in GetElasticityMatrix()
119 r2 = base + i2 * shiftx + j2 * shifty + k2 * shiftz; in GetElasticityMatrix()
/petsc/src/ksp/ksp/impls/gmres/fgmres/ftn-custom/
H A Dzmodpcff.c19 static PetscErrorCode ourmodify(KSP ksp, PetscInt i, PetscInt i2, PetscReal d, PetscCtx ctx) in ourmodify() argument
21 … PetscInt *, PetscInt *, PetscReal *, void *, PetscErrorCode *), (&ksp, &i, &i2, &d, _ctx, &ierr)); in ourmodify()
/petsc/src/ts/tutorials/
H A Dex22f_mf.F9050 PetscInt im11, i2
59 i2 = 2
68 …PetscCallA(DMDACreate1d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, im11, i2, i2, PETSC_NULL_INTEGER_ARRAY…
106 dof = i2*(xe - xs + 1)
107 gdof = i2*(gxe - gxs + 1)
148 dof = i2*(xe - xs + 1)
149 gdof = i2*(gxe - gxs + 1)
H A Dex22f.F9048 PetscInt im11, i2
53 i2 = 2
62 …PetscCallA(DMDACreate1d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, im11, i2, i2, PETSC_NULL_INTEGER, da, …
/petsc/src/vec/vec/utils/
H A Dvecstash.c335 PetscInt i1, i2, bs = stash->bs; in VecStashScatterGetMesg_Private() local
361 i2 = flg_v[2 * recv_status.MPI_SOURCE + 1]; in VecStashScatterGetMesg_Private()
362 if (i1 != -1 && i2 != -1) { in VecStashScatterGetMesg_Private()
363 *rows = stash->rindices + i2 * stash->rmax; in VecStashScatterGetMesg_Private()
/petsc/src/vec/vec/impls/mpi/
H A Dpdvec.c1045 …PetscInt n2 = nroots, *i2; /* Buffers for received COOs from other ranks, along with a permutat… in VecSetPreallocationCOO_MPI() local
1047 PetscCall(PetscMalloc1(n2, &i2)); in VecSetPreallocationCOO_MPI()
1049 …ithMemTypeBegin(sf2, MPIU_INT, PETSC_MEMTYPE_HOST, i1 + rem, PETSC_MEMTYPE_HOST, i2, MPI_REPLACE)); in VecSetPreallocationCOO_MPI()
1050 PetscCall(PetscSFReduceEnd(sf2, MPIU_INT, i1 + rem, i2, MPI_REPLACE)); in VecSetPreallocationCOO_MPI()
1064 PetscCall(PetscSortIntWithCountArray(n2, i2, perm2)); in VecSetPreallocationCOO_MPI()
1069 if (i2[k] == i2[k - 1]) nnz2--; in VecSetPreallocationCOO_MPI()
1079 if (i2[k] != old) { /* Meet a new entry */ in VecSetPreallocationCOO_MPI()
1081 imap2[p] = i2[k] - rstart; in VecSetPreallocationCOO_MPI()
1083 old = i2[k]; in VecSetPreallocationCOO_MPI()
1091 PetscCall(PetscFree(i2)); in VecSetPreallocationCOO_MPI()

12