Home
last modified time | relevance | path

Searched refs:n2 (Results 1 – 25 of 36) sorted by relevance

12

/petsc/src/ksp/ksp/tests/
H A Dex39.c23 PetscInt n1, n2, n3; /* parameters */ in main() local
31 n2 = 32; in main()
34 PetscCall(PetscOptionsGetInt(NULL, NULL, "-n2", &n2, NULL)); in main()
48 PetscCall(MatSetSizes(A, PETSC_DECIDE, PETSC_DECIDE, n1 * n2 * n3, n1 * n2 * n3)); in main()
65 i = Ii / (n2 * n3); in main()
66 j = (Ii - i * n2 * n3) / n3; in main()
67 k = Ii - i * n2 * n3 - j * n3; in main()
69 J = Ii - n2 * n3; in main()
74 J = Ii + n2 * n3; in main()
83 if (j < n2 - 1) { in main()
[all …]
H A Dex38.c32 PetscInt n1, n2; /* parameters */ in main() local
41 n2 = 64; in main()
43 PetscCall(PetscOptionsGetInt(NULL, NULL, "-n2", &n2, NULL)); in main()
68 PetscCall(MatSetSizes(A, PETSC_DECIDE, PETSC_DECIDE, n1 * n2, n1 * n2)); in main()
93 i = Ii / n2; in main()
94 j = Ii - i * n2; in main()
96 J = Ii - n2; in main()
101 J = Ii + n2; in main()
110 if (j < n2 - 1) { in main()
146 PetscCall(VecSetSizes(b, PETSC_DECIDE, n1 * n2)); in main()
/petsc/src/vec/is/sf/tests/
H A Dex23.c9 PetscInt n, n2, N = 12; in main() local
35 n2 = 2 * n; in main()
36 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_HOST, n2, &xh)); in main()
37 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_HOST, n2, &yh)); in main()
38 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_DEVICE, n2, &xd)); in main()
39 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_DEVICE, n2, &yd)); in main()
45 PetscCall(PetscDeviceMemcpy(dctx, xd, xh, sizeof(PetscInt) * n2)); in main()
46 PetscCall(PetscDeviceMemcpy(dctx, yd, yh, sizeof(PetscInt) * n2)); in main()
56 PetscCall(PetscDeviceMemcpy(dctx, xh, xd, sizeof(PetscInt) * n2)); in main()
58 PetscCall(PetscIntView(n2, xh, PETSC_VIEWER_STDOUT_WORLD)); in main()
/petsc/src/vec/is/is/utils/
H A Disdiff.c34 PetscInt i, n1, n2, imin, imax, nout, *iout; in ISDifference() local
71 PetscCall(ISGetLocalSize(is2, &n2)); in ISDifference()
72 for (i = 0; i < n2; i++) { in ISDifference()
124 PetscInt n1, n2, n3, p1, p2, *iout; in ISSum() local
137 PetscCall(ISGetLocalSize(is2, &n2)); in ISSum()
138 if (!n2) { in ISSum()
150 n3 += n2 - p2; in ISSum()
153 while (p2 < n2 && i2[p2] < i1[p1]) { in ISSum()
157 if (p2 == n2) { in ISSum()
169 if (p2 == n2) { in ISSum()
[all …]
/petsc/src/mat/tests/
H A Dex123.c13 PetscInt n1 = 11, n2 = 9; in main() local
74 for (i = 0; i < n2; i++) i2[i] += rstart; in main()
78 for (i = 0; i < n2; i++) j2[i] += cstart; in main()
81 for (i = 0; i < n2; i++) j2[i] += cstart + m; in main()
86 n2 += 2; in main()
89 PetscCall(PetscMalloc2(PetscMax(n1, n2), &it, PetscMax(n1, n2), &jt)); in main()
126 PetscCall(PetscArraycpy(it, i2, n2)); in main()
127 PetscCall(PetscArraycpy(jt, j2, n2)); in main()
129 PetscCall(MatSetPreallocationCOO(A, n2, it, jt)); in main()
131 PetscCall(MatSetPreallocationCOOLocal(A, n2, it, jt)); in main()
[all …]
H A Dex47.c12 PetscInt i, j, row, m, n, ncols1, ncols2, ct, m2, n2; in main() local
48 PetscCall(MatGetSize(B, &m2, &n2)); in main()
/petsc/src/ksp/ksp/tutorials/
H A Dbench_kspsolve.c45 PetscInt n = user->n, n2 = n * n, n1 = n - 1; in PreallocateCOO() local
52 zstart = user->Istart / n2; in PreallocateCOO()
55 zend = (user->Iend - 1) / n2; in PreallocateCOO()
118 PetscInt Ii, x, y, z, n = user->n, n2 = n * n, n1 = n - 1; in FillCOO() local
134 z = Ii / n2; in FillCOO()
137 coo_j[count] = Ii - 1 - n - n2; in FillCOO()
143 coo_j[count] = Ii - 1 + n - n2; in FillCOO()
149 coo_j[count] = Ii + 1 - n - n2; in FillCOO()
155 coo_j[count] = Ii + 1 + n - n2; in FillCOO()
161 coo_j[count] = Ii - 1 - n + n2; in FillCOO()
[all …]
H A Dex10.c53 PetscInt j, len, start, idx, n1, n2; in CreateSystem() local
95 PetscCall(VecGetLocalSize(b, &n2)); in CreateSystem()
96 same = (n1 == n2) ? PETSC_TRUE : PETSC_FALSE; in CreateSystem()
/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
447 n2 = n1 + 1; in DMSetUp_DA_2D()
452 n2 = -1; in DMSetUp_DA_2D()
472 if ((n2 < 0) && (n5 < 0) && (n1 < 0)) n2 = size - m; in DMSetUp_DA_2D()
480 if ((n5 >= 0) && (n2 < 0)) n2 = size - m + rank + 1; in DMSetUp_DA_2D()
487 if ((n1 >= 0) && (n2 < 0)) n2 = rank - 2 * m + 1; in DMSetUp_DA_2D()
495 if ((n5 >= 0) && (n2 < 0)) n2 = size - m + rank + 1; in DMSetUp_DA_2D()
501 if ((n1 >= 0) && (n2 < 0)) n2 = rank - 2 * m + 1; in DMSetUp_DA_2D()
510 dd->neighbors[2] = n2; in DMSetUp_DA_2D()
521 sn2 = n2; 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
549 n2 = rank - m * n - m + 1; in DMSetUp_DA_3D()
591 n2 = rank - 2 * m + 1 - (m * n); in DMSetUp_DA_3D()
605 n2 = rank + m * (n - 1) + 1 - (m * n); in DMSetUp_DA_3D()
629 n2 = size - (m * n) + rank - m + 1; in DMSetUp_DA_3D()
675 n2 = size - (m * n - rank) - (m - 1) - m; in DMSetUp_DA_3D()
687 n2 = rank + m * (n - 1) - (m - 1) - m * n; in DMSetUp_DA_3D()
701 n2 = size - m + rank + 1; in DMSetUp_DA_3D()
727 if ((xe == M) && (ys == 0) && (zs == 0)) n2 = size - m; in DMSetUp_DA_3D()
737 if (xe == M) n2 = n5 = n8 = n11 = n14 = n17 = n20 = n23 = n26 = -2; in DMSetUp_DA_3D()
[all …]
/petsc/src/vec/is/tests/
H A Dex8.c8 PetscInt n1, n2, b1, b2; in TestEqual() local
16 PetscCall(ISLocalToGlobalMappingGetSize(m2, &n2)); 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()
41 PetscInt n, n1, n2, b1, b2; in main() local
100 PetscCall(ISLocalToGlobalMappingGetSize(lg2v, &n2)); in main()
102 n2 /= b2; in main()
105 PetscCall(PetscMalloc1(n2, &idx)); in main()
106 PetscCall(ISLocalToGlobalMappingCreate(PETSC_COMM_SELF, b2, n2, idx, PETSC_OWN_POINTER, &lg2lh)); in main()
/petsc/src/vec/vec/tests/
H A Dex44.c14 PetscInt m, n, i, n1, n2; in main() local
87 n2 = n; in main()
89 PetscCall(PetscSortRemoveDupsInt(&n2, idy)); in main()
92 PetscCall(ISCreateGeneral(PETSC_COMM_WORLD, n2, idy, PETSC_COPY_VALUES, &fromISGeneral)); in main()
H A Dex30f.F9012 PetscInt nlocal2, n2, eight
31 n2 = 2*n
33 …PetscCallA(VecCreateFromOptions(PETSC_COMM_WORLD, PETSC_NULL_CHARACTER, i1, nlocal2, n2, v1, ierr))
/petsc/src/dm/dt/interface/
H A Ddtweakform.c543 PetscInt n0, n1, n2, n3; in PetscWeakFormHasJacobian() local
550 PetscCall(PetscHMapFormGetSize(wf->form[PETSC_WF_G2], &n2)); in PetscWeakFormHasJacobian()
552 *hasJac = n0 + n1 + n2 + n3 ? PETSC_TRUE : PETSC_FALSE; in PetscWeakFormHasJacobian()
556 …onst PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]), PetscInt *n2, void (***g2)(PetscI… in PetscWeakFormGetJacobian() argument
563 …PetscCall(PetscWeakFormGetFunction_Private(wf, wf->form[PETSC_WF_G2], label, val, find, part, n2, … in PetscWeakFormGetJacobian()
580 …const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]), PetscInt n2, void (**g2)(PetscIn… in PetscWeakFormSetJacobian() argument
587 …PetscCall(PetscWeakFormSetFunction_Private(wf, wf->form[PETSC_WF_G2], label, val, find, part, n2, … in PetscWeakFormSetJacobian()
606 PetscInt n0, n1, n2, n3; in PetscWeakFormHasJacobianPreconditioner() local
613 PetscCall(PetscHMapFormGetSize(wf->form[PETSC_WF_GP2], &n2)); in PetscWeakFormHasJacobianPreconditioner()
615 *hasJacPre = n0 + n1 + n2 + n3 ? PETSC_TRUE : PETSC_FALSE; in PetscWeakFormHasJacobianPreconditioner()
[all …]
/petsc/src/mat/graphops/coarsen/impls/hem/
H A Dhem.c140 PetscCDIntNd *n, *n2; in PetscCDAppendID() local
145 if (!(n2 = ail->array[a_idx])) ail->array[a_idx] = n; in PetscCDAppendID()
148 if (!n2->next) { in PetscCDAppendID()
149 n2->next = n; in PetscCDAppendID()
153 n2 = n2->next; in PetscCDAppendID()
154 } while (n2); in PetscCDAppendID()
155 PetscCheck(n2, PETSC_COMM_SELF, PETSC_ERR_PLIB, "n2 should be non-null"); in PetscCDAppendID()
164 PetscCDIntNd *n2; in PetscCDAppendNode() local
168 if (!(n2 = ail->array[a_idx])) ail->array[a_idx] = a_n; in PetscCDAppendNode()
171 if (!n2->next) { in PetscCDAppendNode()
[all …]
/petsc/src/vec/is/sf/impls/basic/neighbor/
H A Dsfneighbor.c195 PetscMPIInt m, n, m2, n2; in PetscSFSetUp_Neighbor() local
209 n2 = n; in PetscSFSetUp_Neighbor()
213 n2 = n ? n : 1; in PetscSFSetUp_Neighbor()
217 …displs, m2, &dat->rootcounts, m2, &dat->rootweights, n2, &dat->leafdispls, n2, &dat->leafcounts, n… in PetscSFSetUp_Neighbor()
/petsc/src/binding/petsc4py/test/
H A Dtest_vec.py51 n2 = self.vec.norm(PETSc.NormType.NORM_2)
54 self.assertAlmostEqual(n2, sqrt(self.vec.getSize()))
55 self.assertAlmostEqual(n2, self.vec.norm())
62 self.assertAlmostEqual(nn2, n2)
71 n2 = self.vec.normalize()
72 self.assertAlmostEqual(n2, sqrt(self.vec.getSize()))
H A Dtest_dmda.py238 for n1, n2 in zip(self.da.getSizes(), cda.getSizes()):
239 self.assertTrue(abs(n1 - n2) <= 1)
252 for n1, n2 in zip(self.da.getSizes(), rda.getSizes()):
253 self.assertTrue(abs(n1 - n2) <= 1)
/petsc/src/ksp/ksp/interface/
H A Ditfunc.c1204 PetscInt n1, N1, n2, N2, Bbn = PETSC_DECIDE; in KSPMatSolve_Private() local
1223 PetscCall(MatGetLocalSize(B, NULL, &n2)); in KSPMatSolve_Private()
1227 …n2 && N1 == N2, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Incompatible number of columns between block … in KSPMatSolve_Private()
1256 for (n2 = 0; n2 < N2; n2 += Bbn) { in KSPMatSolve_Private()
1257 … PetscCall(MatDenseGetSubMatrix(B, PETSC_DECIDE, PETSC_DECIDE, n2, PetscMin(n2 + Bbn, N2), &vB)); in KSPMatSolve_Private()
1258 … PetscCall(MatDenseGetSubMatrix(X, PETSC_DECIDE, PETSC_DECIDE, n2, PetscMin(n2 + Bbn, N2), &vX)); in KSPMatSolve_Private()
1260 …cCall(KSPViewFinalMatResidual_Internal(ksp, vB, vX, ksp->viewerFinalRes, ksp->formatFinalRes, n2)); in KSPMatSolve_Private()
1289 for (n2 = 0; n2 < N2; ++n2) { in KSPMatSolve_Private()
1290 PetscCall(MatDenseGetColumnVecRead(B, n2, &cb)); in KSPMatSolve_Private()
1291 PetscCall(MatDenseGetColumnVecWrite(X, n2, &cx)); in KSPMatSolve_Private()
[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
551 PetscCall(MatGetLocalSize(X, &m2, &n2)); in PCMatApplyTranspose_Private()
556 …n2 && N1 == N2, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Incompatible number of columns between block … in PCMatApplyTranspose_Private()
1271 PetscInt m1, n1, m2, n2; in PCSetOperators() local
1281 PetscCall(MatGetLocalSize(pc->mat, &m2, &n2)); in PCSetOperators()
1282 …n2, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Cannot change local size of Amat after use old sizes %" P… in PCSetOperators()
1284 PetscCall(MatGetLocalSize(pc->pmat, &m2, &n2)); in PCSetOperators()
1285 …n2, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Cannot change local size of Pmat after use old sizes %" P… in PCSetOperators()
/petsc/src/dm/dt/fe/impls/basic/
H A Dfebasic.c749 PetscInt n0, n1, n2, n3; in PetscFEIntegrateJacobian_Basic() local
784 …label, key.value, fieldI, fieldJ, key.part, &n0, &g0_func, &n1, &g1_func, &n2, &g2_func, &n3, &g3_… in PetscFEIntegrateJacobian_Basic()
787 …label, key.value, fieldI, fieldJ, key.part, &n0, &g0_func, &n1, &g1_func, &n2, &g2_func, &n3, &g3_… in PetscFEIntegrateJacobian_Basic()
790 …label, key.value, fieldI, fieldJ, key.part, &n0, &g0_func, &n1, &g1_func, &n2, &g2_func, &n3, &g3_… in PetscFEIntegrateJacobian_Basic()
793 if (!n0 && !n1 && !n2 && !n3) PetscFunctionReturn(PETSC_SUCCESS); in PetscFEIntegrateJacobian_Basic()
796 …PetscCall(PetscDSGetWeakFormArrays(rds, NULL, NULL, n0 ? &g0 : NULL, n1 ? &g1 : NULL, n2 ? &g2 : N… in PetscFEIntegrateJacobian_Basic()
862 if (n2) { in PetscFEIntegrateJacobian_Basic()
864 …for (PetscInt i = 0; i < n2; ++i) g2_func[i](dE, Nf, NfAux, uOff, uOff_x, u, u_t, u_x, aOff, aOff_… in PetscFEIntegrateJacobian_Basic()
898 PetscInt n0, n1, n2, n3, i; in PetscFEIntegrateBdJacobian_Basic() local
932 …label, key.value, fieldI, fieldJ, key.part, &n0, &g0_func, &n1, &g1_func, &n2, &g2_func, &n3, &g3_… in PetscFEIntegrateBdJacobian_Basic()
[all …]
/petsc/src/vec/is/section/interface/
H A Dsection.c252 PetscInt pStart, pEnd, nfields, ncdof, nfcdof, p, f, n1, n2; in PetscSectionCompare() local
271 PetscCall(PetscSectionGetChart(s2, &n1, &n2)); in PetscSectionCompare()
272 if (pStart != n1 || pEnd != n2) goto not_congruent; in PetscSectionCompare()
283 PetscCall(PetscSectionGetOffset(s2, p, &n2)); in PetscSectionCompare()
284 if (n1 != n2) goto not_congruent; in PetscSectionCompare()
287 PetscCall(PetscSectionGetDof(s2, p, &n2)); in PetscSectionCompare()
288 if (n1 != n2) goto not_congruent; in PetscSectionCompare()
291 PetscCall(PetscSectionGetConstraintDof(s2, p, &n2)); in PetscSectionCompare()
292 if (ncdof != n2) goto not_congruent; in PetscSectionCompare()
301 PetscCall(PetscSectionGetNumFields(s2, &n2)); in PetscSectionCompare()
[all …]
/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()
1048 PetscCall(PetscMalloc1(n2, &perm2)); in VecSetPreallocationCOO_MPI()
1061 PetscCount sendlen = n1 - rem, recvlen = n2; in VecSetPreallocationCOO_MPI()
1063 for (k = 0; k < n2; k++) perm2[k] = k; in VecSetPreallocationCOO_MPI()
1064 PetscCall(PetscSortIntWithCountArray(n2, i2, perm2)); in VecSetPreallocationCOO_MPI()
1067 nnz2 = n2; in VecSetPreallocationCOO_MPI()
1068 for (k = 1; k < n2; k++) { in VecSetPreallocationCOO_MPI()
1078 for (k = 0; k < n2; k++) { in VecSetPreallocationCOO_MPI()
/petsc/src/mat/utils/
H A Daxpy.c99 PetscInt m1, m2, n1, n2; in MatAXPY() local
110 PetscCall(MatGetLocalSize(Y, &m2, &n2)); in MatAXPY()
112 …n2, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Non conforming matrix add: local sizes X %" PetscInt_FMT … in MatAXPY()
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx270 PetscInt n2, *garray2; in ReduceTwoSetsOfGlobalIndices() local
286 n2 = tot; in ReduceTwoSetsOfGlobalIndices()
287 PetscCall(PetscMalloc1(n2, &garray2)); in ReduceTwoSetsOfGlobalIndices()
297 PetscCall(PetscSortInt(n2, garray2)); in ReduceTwoSetsOfGlobalIndices()
310 *n2_ = n2; in ReduceTwoSetsOfGlobalIndices()
595 PetscInt n2, *garray2; in MatMPIAIJKokkosReduceBegin() local
597 PetscCall(ReduceTwoSetsOfGlobalIndices(n1, garray1, Fonz, Foj, &n2, &garray2, map)); in MatMPIAIJKokkosReduceBegin()
602 mm->n = n2; in MatMPIAIJKokkosReduceBegin()
618 …PetscCallCXX(mm->Fo = KokkosCsrMatrix("Fo", Fm, n2, Fonz, Foa_d, Foi_d, Foj_d)); // Fo's column si… in MatMPIAIJKokkosReduceBegin()
906 PetscInt n2, *garray2; in MatMPIAIJKokkosBcastBegin() local
[all …]

12