Home
last modified time | relevance | path

Searched refs:n1 (Results 1 – 25 of 48) sorted by relevance

12

/petsc/src/ksp/ksp/tutorials/
H A Dbench_kspsolve.c45 PetscInt n = user->n, n2 = n * n, n1 = n - 1; in PreallocateCOO() local
60 if (zstart == zend && (!zstart || zstart == n1)) idx = 0; in PreallocateCOO()
61 if (zstart == zend && (zstart && zstart < n1)) idx = 1; in PreallocateCOO()
64 else if (xstart && xstart < n1 && !ystart) // bottom in PreallocateCOO()
65 user->nnz += 3 * nnz[idx] + (3 * nm2 + n1 - xstart) * nnz[idx + 1] + nm2 * nm2 * nnz[idx + 2]; in PreallocateCOO()
66 else if (xstart == n1 && !ystart) // bottom right in PreallocateCOO()
68 else if (!xstart && ystart && ystart < n1) // left in PreallocateCOO()
69 …user->nnz += 2 * nnz[idx] + (nm2 + 2 * (n1 - ystart)) * nnz[idx + 1] + nm2 * (n1 - ystart) * nnz[i… in PreallocateCOO()
70 else if (xstart && xstart < n1 && ystart && ystart < n1) // center in PreallocateCOO()
71 …user->nnz += 2 * nnz[idx] + (nm2 + (n1 - ystart) + (nm2 - ystart)) * nnz[idx + 1] + (nm2 * (nm2 - … in PreallocateCOO()
[all …]
H A Dex10.c53 PetscInt j, len, start, idx, n1, n2; in CreateSystem() local
94 PetscCall(MatGetLocalSize(A, NULL, &n1)); in CreateSystem()
96 same = (n1 == n2) ? PETSC_TRUE : PETSC_FALSE; in CreateSystem()
101 PetscCall(VecSetSizes(b2, n1, PETSC_DECIDE)); in CreateSystem()
H A Dex27.c153 PetscInt m1, n1; in main() local
163 PetscCall(MatGetLocalSize(A, &m1, &n1)); in main()
164 …n1 == n, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "resulting sizes differ from requested ones: %" PetscIn… in main()
/petsc/src/vec/is/is/utils/
H A Disdiff.c34 PetscInt i, n1, n2, imin, imax, nout, *iout; in ISDifference() local
49 PetscCall(ISGetLocalSize(is1, &n1)); in ISDifference()
52 if (n1) { in ISDifference()
55 for (i = 0; i < n1; i++) { in ISDifference()
64 for (i = 0; i < n1; i++) { in ISDifference()
124 PetscInt n1, n2, n3, p1, p2, *iout; in ISSum() local
136 PetscCall(ISGetLocalSize(is1, &n1)); in ISSum()
149 if (p1 == n1) { /* cleanup for is2 */ in ISSum()
159 n3 += n1 - p1; in ISSum()
171 n3 += n1 - p1; in ISSum()
[all …]
/petsc/src/ksp/pc/tests/
H A Dex5.c176 PetscInt i, n1; in residual() local
181 PetscCall(VecGetSize(bb, &n1)); in residual()
185 n1--; in residual()
187 r[n1] = b[n1] + x[n1 - 1] - 2.0 * x[n1]; in residual()
188 for (i = 1; i < n1; i++) r[i] = b[i] + x[i + 1] + x[i - 1] - 2.0 * x[i]; in residual()
197 PetscInt i, n1; in amult() local
202 PetscCall(VecGetSize(xx, &n1)); in amult()
205 n1--; in amult()
207 y[n1] = -x[n1 - 1] + 2.0 * x[n1]; in amult()
208 for (i = 1; i < n1; i++) y[i] = -x[i + 1] - x[i - 1] + 2.0 * x[i]; in amult()
[all …]
/petsc/src/ksp/ksp/tests/
H A Dex38.c32 PetscInt n1, n2; /* parameters */ in main() local
40 n1 = 64; in main()
42 PetscCall(PetscOptionsGetInt(NULL, NULL, "-n1", &n1, NULL)); in main()
45 h = 1.0 / n1; in main()
68 PetscCall(MatSetSizes(A, PETSC_DECIDE, PETSC_DECIDE, n1 * n2, n1 * n2)); in main()
100 if (i < n1 - 1) { in main()
146 PetscCall(VecSetSizes(b, PETSC_DECIDE, n1 * n2)); in main()
H A Dex39.c23 PetscInt n1, n2, n3; /* parameters */ in main() local
30 n1 = 32; in main()
33 PetscCall(PetscOptionsGetInt(NULL, NULL, "-n1", &n1, NULL)); in main()
37 h = 1.0 / n1; in main()
48 PetscCall(MatSetSizes(A, PETSC_DECIDE, PETSC_DECIDE, n1 * n2 * n3, n1 * n2 * n3)); in main()
73 if (i < n1 - 1) { in main()
106 PetscCall(VecSetSizes(b, PETSC_DECIDE, n1 * n2 * n3)); in main()
/petsc/src/mat/impls/fft/fftw/
H A Dfftw.c508 PetscInt n1, N1; in MatCreateVecsFFTW_FFTW() local
524 PetscCall(PetscIntCast(local_n0, &n1)); in MatCreateVecsFFTW_FFTW()
526 PetscCall(VecCreateMPIWithArray(comm, 1, n1, N1, (const PetscScalar *)data_fin, fin)); in MatCreateVecsFFTW_FFTW()
533 PetscCall(PetscIntCast(local_n1, &n1)); in MatCreateVecsFFTW_FFTW()
535 PetscCall(VecCreateMPIWithArray(comm, 1, n1, N1, (const PetscScalar *)data_fout, fout)); in MatCreateVecsFFTW_FFTW()
543 PetscCall(PetscIntCast(local_n1, &n1)); in MatCreateVecsFFTW_FFTW()
545 PetscCall(VecCreateMPIWithArray(comm, 1, n1, N1, (const PetscScalar *)data_bout, bout)); in MatCreateVecsFFTW_FFTW()
556 PetscCall(PetscIntCast(2 * local_n0 * (dim[1] / 2 + 1), &n1)); in MatCreateVecsFFTW_FFTW()
559 PetscCall(VecCreateMPIWithArray(comm, 1, (PetscInt)n1, N1, (PetscScalar *)data_finr, fin)); in MatCreateVecsFFTW_FFTW()
566 PetscCall(VecCreateMPIWithArray(comm, 1, (PetscInt)n1, N1, (PetscScalar *)data_fout, fout)); in MatCreateVecsFFTW_FFTW()
[all …]
/petsc/src/mat/tests/
H A Dex76.c10 PetscInt n, mbs = 16, bs = 1, nz = 3, prob = 1, i, j, col[3], block, row, Ii, J, n1, lvl; in main() local
75 n1 = (PetscInt)(PetscSqrtReal((PetscReal)n) + 0.001); in main()
76 …PetscCheck(n1 * n1 == n, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "sqrt(n) must be a positive integer… in main()
77 for (i = 0; i < n1; i++) { in main()
78 for (j = 0; j < n1; j++) { in main()
79 Ii = j + n1 * i; in main()
81 J = Ii - n1; in main()
84 if (i < n1 - 1) { in main()
85 J = Ii + n1; in main()
92 if (j < n1 - 1) { in main()
H A Dex77.c10 PetscInt n, mbs = 16, bs = 1, nz = 3, prob = 2, i, j, col[3], row, Ii, J, n1; in main() local
74 n1 = (PetscInt)(PetscSqrtReal((PetscReal)n) + 0.001); in main()
75 …PetscCheck(n1 * n1 == n, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "sqrt(n) must be a positive integer… in main()
76 for (i = 0; i < n1; i++) { in main()
77 for (j = 0; j < n1; j++) { in main()
78 Ii = j + n1 * i; in main()
80 J = Ii - n1; in main()
84 if (i < n1 - 1) { in main()
85 J = Ii + n1; in main()
94 if (j < n1 - 1) { in main()
H A Dex123.c13 PetscInt n1 = 11, n2 = 9; in main() local
72 for (i = 0; i < n1; i++) i1[i] += rstart; in main()
77 for (i = 0; i < n1; i++) j1[i] += cstart; in main()
80 for (i = 0; i < n1; i++) j1[i] += cstart + m; in main()
85 n1 += 2; in main()
89 PetscCall(PetscMalloc2(PetscMax(n1, n2), &it, PetscMax(n1, n2), &jt)); in main()
92 PetscCall(MatSetPreallocationCOO(A, n1, i1, j1)); in main()
94 PetscCall(PetscArraycpy(it, i1, n1)); in main()
95 PetscCall(PetscArraycpy(jt, j1, n1)); in main()
96 PetscCall(MatSetPreallocationCOOLocal(A, n1, it, jt)); in main()
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
64 n1 = (int)PetscSqrtReal((PetscReal)n); in main()
65 for (i = 0; i < n1; i++) { in main()
66 for (j = 0; j < n1; j++) { in main()
67 Ii = j + n1 * i; in main()
69 J = Ii - n1; in main()
72 if (i < n1 - 1) { in main()
73 J = Ii + n1; in main()
80 if (j < n1 - 1) { in main()
H A Dex74.c11 …t n, mbs = 16, bs = 1, nz = 3, prob = 1, i, j, k1, k2, col[3], lf, block, row, Ii, J, n1, inc; in main() local
87 n1 = (PetscInt)(PetscSqrtReal((PetscReal)n) + 0.001); in main()
88 …PetscCheck(n1 * n1 == n, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "sqrt(n) must be a positive integer… in main()
89 for (i = 0; i < n1; i++) { in main()
90 for (j = 0; j < n1; j++) { in main()
91 Ii = j + n1 * i; in main()
93 J = Ii - n1; in main()
97 if (i < n1 - 1) { in main()
98 J = Ii + n1; in main()
107 if (j < n1 - 1) { in main()
H A Dex147.c15 PetscInt i, j, indx[100], n1; in main() local
54 n1 = local_n1 * N0 * 2 * N1 * N2; in main()
/petsc/src/vec/is/tests/
H A Dex8.c8 PetscInt n1, n2, b1, b2; in TestEqual() local
13 PetscCall(ISLocalToGlobalMappingGetSize(m1, &n1)); in TestEqual()
21 flg = (PetscBool)(n1 == n2); in TestEqual()
22 …intf(PETSC_COMM_SELF, "%s: different sizes %" PetscInt_FMT " %" PetscInt_FMT "\n", tname, n1, n2)); in TestEqual()
24 PetscCall(PetscArraycmp(idx1, idx2, n1, &flg)); in TestEqual()
41 PetscInt n, n1, n2, b1, b2; in main() local
97 PetscCall(ISLocalToGlobalMappingGetSize(lg1v, &n1)); in main()
99 n1 /= b1; in main()
103 PetscCall(PetscMalloc1(n1, &idx)); in main()
104 PetscCall(ISLocalToGlobalMappingCreate(PETSC_COMM_SELF, b1, n1, idx, PETSC_OWN_POINTER, &lg1lh)); in main()
/petsc/src/dm/impls/da/
H A Dda2.c201 PetscMPIInt rank, size, n0, n1, n2, n3, n5, n6, n7, n8; in DMSetUp_DA_2D() local
440 n1 = rank - m; in DMSetUp_DA_2D()
442 n0 = n1 - 1; in DMSetUp_DA_2D()
447 n2 = n1 + 1; in DMSetUp_DA_2D()
472 if ((n2 < 0) && (n5 < 0) && (n1 < 0)) n2 = size - m; in DMSetUp_DA_2D()
473 if ((n0 < 0) && (n3 < 0) && (n1 < 0)) n0 = size - 1; in DMSetUp_DA_2D()
476 if (n1 < 0) n1 = rank + m * (n - 1); in DMSetUp_DA_2D()
486 if ((n1 >= 0) && (n0 < 0)) n0 = rank - 1; in DMSetUp_DA_2D()
487 if ((n1 >= 0) && (n2 < 0)) n2 = rank - 2 * m + 1; in DMSetUp_DA_2D()
491 if (n1 < 0) n1 = rank + m * (n - 1); in DMSetUp_DA_2D()
[all …]
H A Dda3.c211 PetscMPIInt n0, n1, n2, n3, n4, n5, n6, n7, n8, n9, n10, n11, n12, n14; in DMSetUp_DA_3D() local
548 n1 = rank - m * n - m; in DMSetUp_DA_3D()
604 n1 = rank + m * (n - 1) - (m * n); in DMSetUp_DA_3D()
628 n1 = size - (m * n) + rank - m; in DMSetUp_DA_3D()
700 n1 = size - m + rank; in DMSetUp_DA_3D()
742 if (ys == 0) n0 = n1 = n2 = n9 = n10 = n11 = n18 = n19 = n20 = -2; in DMSetUp_DA_3D()
748 if (zs == 0) n0 = n1 = n2 = n3 = n4 = n5 = n6 = n7 = n8 = -2; in DMSetUp_DA_3D()
755 dd->neighbors[1] = n1; in DMSetUp_DA_3D()
786 sn1 = n1; in DMSetUp_DA_3D()
805 …n0 = n1 = n2 = n3 = n5 = n6 = n7 = n8 = n9 = n11 = n15 = n17 = n18 = n19 = n20 = n21 = n23 = n24 =… in DMSetUp_DA_3D()
[all …]
/petsc/src/vec/vec/tests/
H A Dex44.c14 PetscInt m, n, i, n1, n2; in main() local
86 n1 = n; in main()
88 PetscCall(PetscSortRemoveDupsInt(&n1, idx)); in main()
91 PetscCall(ISCreateGeneral(PETSC_COMM_WORLD, n1, idx, PETSC_COPY_VALUES, &toISGeneral)); in main()
/petsc/src/dm/dt/interface/
H A Ddtweakform.c477 …onst PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]), PetscInt *n1, void (***f1)(PetscI… in PetscWeakFormGetResidual() argument
481 …PetscCall(PetscWeakFormGetFunction_Private(wf, wf->form[PETSC_WF_F1], label, val, f, part, n1, (vo… in PetscWeakFormGetResidual()
493 …const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]), PetscInt n1, void (**f1)(PetscIn… in PetscWeakFormSetResidual() argument
497 …PetscCall(PetscWeakFormSetFunction_Private(wf, wf->form[PETSC_WF_F1], label, val, f, part, n1, (Pe… in PetscWeakFormSetResidual()
509 …onst PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]), PetscInt *n1, void (***f1)(PetscI… in PetscWeakFormGetBdResidual() argument
513 …PetscCall(PetscWeakFormGetFunction_Private(wf, wf->form[PETSC_WF_BDF1], label, val, f, part, n1, (… in PetscWeakFormGetBdResidual()
525 …const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]), PetscInt n1, void (**f1)(PetscIn… in PetscWeakFormSetBdResidual() argument
529 …PetscCall(PetscWeakFormSetFunction_Private(wf, wf->form[PETSC_WF_BDF1], label, val, f, part, n1, (… in PetscWeakFormSetBdResidual()
543 PetscInt n0, n1, n2, n3; in PetscWeakFormHasJacobian() local
549 PetscCall(PetscHMapFormGetSize(wf->form[PETSC_WF_G1], &n1)); in PetscWeakFormHasJacobian()
[all …]
/petsc/src/ksp/pc/interface/
H A Dprecon.c539 PetscInt m1, M1, m2, M2, n1, N1, n2, N2, m3, M3, n3, N3; in PCMatApplyTranspose_Private() local
552 PetscCall(MatGetLocalSize(Y, &m1, &n1)); in PCMatApplyTranspose_Private()
556 …n1 == n2 && N1 == N2, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Incompatible number of columns between … in PCMatApplyTranspose_Private()
574 for (n1 = 0; n1 < N1; ++n1) { in PCMatApplyTranspose_Private()
575 PetscCall(MatDenseGetColumnVecRead(X, n1, &cx)); in PCMatApplyTranspose_Private()
576 PetscCall(MatDenseGetColumnVecWrite(Y, n1, &cy)); in PCMatApplyTranspose_Private()
579 PetscCall(MatDenseRestoreColumnVecWrite(Y, n1, &cy)); in PCMatApplyTranspose_Private()
580 PetscCall(MatDenseRestoreColumnVecRead(X, n1, &cx)); in PCMatApplyTranspose_Private()
1271 PetscInt m1, n1, m2, n2; in PCSetOperators() local
1280 PetscCall(MatGetLocalSize(Amat, &m1, &n1)); in PCSetOperators()
[all …]
/petsc/src/dm/dt/fe/impls/basic/
H A Dfebasic.c395 PetscInt n0, n1, i; in PetscFEIntegrateResidual_Basic() local
418 …cWeakFormGetResidual(wf, key.label, key.value, key.field, key.part, &n0, &f0_func, &n1, &f1_func)); in PetscFEIntegrateResidual_Basic()
419 if (!n0 && !n1) PetscFunctionReturn(PETSC_SUCCESS); in PetscFEIntegrateResidual_Basic()
462 …for (i = 0; i < n1; ++i) f1_func[i](dE, Nf, NfAux, uOff, uOff_x, u, u_t, u_x, aOff, aOff_x, a, NUL… in PetscFEIntegrateResidual_Basic()
501 PetscInt n0, n1, i; in PetscFEIntegrateBdResidual_Basic() local
524 …eakFormGetBdResidual(wf, key.label, key.value, key.field, key.part, &n0, &f0_func, &n1, &f1_func)); in PetscFEIntegrateBdResidual_Basic()
525 if (!n0 && !n1) PetscFunctionReturn(PETSC_SUCCESS); in PetscFEIntegrateBdResidual_Basic()
579 …for (i = 0; i < n1; ++i) f1_func[i](dE, Nf, NfAux, uOff, uOff_x, u, u_t, u_x, aOff, aOff_x, a, NUL… in PetscFEIntegrateBdResidual_Basic()
586 if (n1) { in PetscFEIntegrateBdResidual_Basic()
615 PetscInt n0, n1, i; in PetscFEIntegrateHybridResidual_Basic() local
[all …]
/petsc/src/vec/vec/impls/mpi/
H A Dpdvec.c915 PetscCount n1 = coo_n, nneg, *perm; in VecSetPreallocationCOO_MPI() local
917 PetscCall(PetscMalloc1(n1, &i1)); in VecSetPreallocationCOO_MPI()
918 PetscCall(PetscMalloc1(n1, &perm)); in VecSetPreallocationCOO_MPI()
919 PetscCall(PetscArraycpy(i1, coo_i, n1)); /* Make a copy since we'll modify it */ in VecSetPreallocationCOO_MPI()
920 for (k = 0; k < n1; k++) perm[k] = k; in VecSetPreallocationCOO_MPI()
926 for (k = 0; k < n1; k++) { in VecSetPreallocationCOO_MPI()
939 PetscCall(PetscSortIntWithCountArray(n1, i1, perm)); in VecSetPreallocationCOO_MPI()
940 for (k = 0; k < n1; k++) { in VecSetPreallocationCOO_MPI()
944 …PetscCall(PetscSortedIntUpperBound(i1, nneg, n1, rend - 1 - PETSC_INT_MAX, &rem)); /* rem is upper… in VecSetPreallocationCOO_MPI()
959 PetscCall(PetscMalloc1(n1 - rem, &Cperm)); in VecSetPreallocationCOO_MPI()
[all …]
/petsc/src/mat/graphops/coarsen/impls/hem/
H A Dhem.c249 PetscCDIntNd *rem, *n1; in PetscCDRemoveAllAt() local
255 if (!(n1 = ail->extra_nodes)) ail->extra_nodes = rem; in PetscCDRemoveAllAt()
257 while (n1->next) n1 = n1->next; in PetscCDRemoveAllAt()
258 n1->next = rem; in PetscCDRemoveAllAt()
267 PetscCDIntNd *n1; in PetscCDCountAt() local
272 n1 = ail->array[a_idx]; in PetscCDCountAt()
273 while (n1) { in PetscCDCountAt()
274 n1 = n1->next; in PetscCDCountAt()
289 PetscCDIntNd *n1 = ail->array[ii]; in PetscCDCount() local
291 while (n1) { in PetscCDCount()
[all …]
/petsc/lib/petsc/bin/maint/
H A Dcheck-fortran-coverage.sh21 petsc_arch=$(find . -maxdepth 1 -type d -name 'arch-*' | head -n1 | sed 's|^\./||')
/petsc/src/mat/impls/lrc/
H A Dlrc.c202 PetscInt k, k1, m, n, m1, n1; in MatLRCSetMats() local
231 PetscCall(MatGetLocalSize(A, &m1, &n1)); in MatLRCSetMats()
233 …cCheck(n == n1, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Local dimensions of V %" PetscInt_FMT " an… in MatLRCSetMats()

12