| /petsc/src/ksp/ksp/tests/ |
| H A D | ex39.c | 23 PetscInt n1, n2, n3; /* parameters */ in main() local
32 n3 = 32; in main()
35 PetscCall(PetscOptionsGetInt(NULL, NULL, "-n3", &n3, 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()
79 J = Ii - n3; in main()
[all …]
|
| /petsc/src/vec/is/is/utils/ |
| H A D | isdiff.c | 124 PetscInt n1, n2, n3, p1, p2, *iout; in ISSum() local
147 n3 = 0; in ISSum()
150 n3 += n2 - p2; in ISSum()
154 n3++; in ISSum()
159 n3 += n1 - p1; in ISSum()
163 n3++; in ISSum()
171 n3 += n1 - p1; in ISSum()
175 n3++; in ISSum()
180 n3 += n2 - p2; in ISSum()
184 n3++; in ISSum()
[all …]
|
| /petsc/src/dm/impls/da/ |
| H A D | da2.c | 201 PetscMPIInt rank, size, n0, n1, n2, n3, n5, n6, n7, n8; in DMSetUp_DA_2D() local
457 n3 = rank - 1; in DMSetUp_DA_2D()
458 n6 = n3 + m; in DMSetUp_DA_2D()
461 n3 = -1; in DMSetUp_DA_2D()
470 if ((n6 < 0) && (n7 < 0) && (n3 < 0)) n6 = m - 1; in DMSetUp_DA_2D()
473 if ((n0 < 0) && (n3 < 0) && (n1 < 0)) n0 = size - 1; in DMSetUp_DA_2D()
478 if ((n3 >= 0) && (n0 < 0)) n0 = size - m + rank - 1; in DMSetUp_DA_2D()
479 if ((n3 >= 0) && (n6 < 0)) n6 = (rank % m) - 1; in DMSetUp_DA_2D()
484 if (n3 < 0) n3 = rank + (m - 1); in DMSetUp_DA_2D()
493 if ((n3 >= 0) && (n0 < 0)) n0 = size - m + rank - 1; in DMSetUp_DA_2D()
[all …]
|
| H A D | da3.c | 211 PetscMPIInt n0, n1, n2, n3, n4, n5, n6, n7, n8, n9, n10, n11, n12, n14; in DMSetUp_DA_3D() local
550 n3 = rank - m * n - 1; in DMSetUp_DA_3D()
580 n3 = rank + m - 1 - (m * n); in DMSetUp_DA_3D()
630 n3 = size - (m * n) + rank - 1; in DMSetUp_DA_3D()
652 n3 = size - m * n + rank + m - 1; in DMSetUp_DA_3D()
736 if (xs == 0) n0 = n3 = n6 = n9 = n12 = n15 = n18 = n21 = n24 = -2; in DMSetUp_DA_3D()
748 if (zs == 0) n0 = n1 = n2 = n3 = n4 = n5 = n6 = n7 = n8 = -2; in DMSetUp_DA_3D()
757 dd->neighbors[3] = n3; in DMSetUp_DA_3D()
788 sn3 = n3; 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/dm/dt/interface/ |
| H A D | dtweakform.c | 543 PetscInt n0, n1, n2, n3; in PetscWeakFormHasJacobian() local
551 PetscCall(PetscHMapFormGetSize(wf->form[PETSC_WF_G3], &n3)); in PetscWeakFormHasJacobian()
552 *hasJac = n0 + n1 + n2 + n3 ? PETSC_TRUE : PETSC_FALSE; in PetscWeakFormHasJacobian()
556 …onst PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]), PetscInt *n3, void (***g3)(PetscI… in PetscWeakFormGetJacobian() argument
564 …PetscCall(PetscWeakFormGetFunction_Private(wf, wf->form[PETSC_WF_G3], label, val, find, part, n3, … in PetscWeakFormGetJacobian()
580 …const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]), PetscInt n3, void (**g3)(PetscIn… in PetscWeakFormSetJacobian() argument
588 …PetscCall(PetscWeakFormSetFunction_Private(wf, wf->form[PETSC_WF_G3], label, val, find, part, n3, … in PetscWeakFormSetJacobian()
606 PetscInt n0, n1, n2, n3; in PetscWeakFormHasJacobianPreconditioner() local
614 PetscCall(PetscHMapFormGetSize(wf->form[PETSC_WF_GP3], &n3)); in PetscWeakFormHasJacobianPreconditioner()
615 *hasJacPre = n0 + n1 + n2 + n3 ? PETSC_TRUE : PETSC_FALSE; in PetscWeakFormHasJacobianPreconditioner()
[all …]
|
| H A D | dtds.c | 1511 PetscInt n0, n1, n2, n3; in PetscDSGetJacobian() local
1517 …etscWeakFormGetJacobian(ds->wf, NULL, 0, f, g, 0, &n0, &tmp0, &n1, &tmp1, &n2, &tmp2, &n3, &tmp3)); in PetscDSGetJacobian()
1649 PetscInt n0, n1, n2, n3; in PetscDSGetJacobianPreconditioner() local
1655 …tJacobianPreconditioner(ds->wf, NULL, 0, f, g, 0, &n0, &tmp0, &n1, &tmp1, &n2, &tmp2, &n3, &tmp3)); in PetscDSGetJacobianPreconditioner()
1761 PetscInt n0, n1, n2, n3; in PetscDSGetDynamicJacobian() local
1767 …kFormGetDynamicJacobian(ds->wf, NULL, 0, f, g, 0, &n0, &tmp0, &n1, &tmp1, &n2, &tmp2, &n3, &tmp3)); in PetscDSGetDynamicJacobian()
2100 PetscInt n0, n1, n2, n3; in PetscDSGetBdJacobian() local
2106 …scWeakFormGetBdJacobian(ds->wf, NULL, 0, f, g, 0, &n0, &tmp0, &n1, &tmp1, &n2, &tmp2, &n3, &tmp3)); in PetscDSGetBdJacobian()
2216 PetscInt n0, n1, n2, n3; in PetscDSGetBdJacobianPreconditioner() local
2222 …dJacobianPreconditioner(ds->wf, NULL, 0, f, g, 0, &n0, &tmp0, &n1, &tmp1, &n2, &tmp2, &n3, &tmp3)); in PetscDSGetBdJacobianPreconditioner()
|
| /petsc/src/dm/dt/fe/impls/basic/ |
| H A D | febasic.c | 749 PetscInt n0, n1, n2, n3; in PetscFEIntegrateJacobian_Basic() local
784 … key.value, fieldI, fieldJ, key.part, &n0, &g0_func, &n1, &g1_func, &n2, &g2_func, &n3, &g3_func)); in PetscFEIntegrateJacobian_Basic()
787 … key.value, fieldI, fieldJ, key.part, &n0, &g0_func, &n1, &g1_func, &n2, &g2_func, &n3, &g3_func)); in PetscFEIntegrateJacobian_Basic()
790 … key.value, fieldI, fieldJ, key.part, &n0, &g0_func, &n1, &g1_func, &n2, &g2_func, &n3, &g3_func)); in PetscFEIntegrateJacobian_Basic()
793 if (!n0 && !n1 && !n2 && !n3) PetscFunctionReturn(PETSC_SUCCESS); in PetscFEIntegrateJacobian_Basic()
796 …akFormArrays(rds, NULL, NULL, n0 ? &g0 : NULL, n1 ? &g1 : NULL, n2 ? &g2 : NULL, n3 ? &g3 : NULL)); in PetscFEIntegrateJacobian_Basic()
867 if (n3) { in PetscFEIntegrateJacobian_Basic()
869 …for (PetscInt i = 0; i < n3; ++i) g3_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 … key.value, fieldI, fieldJ, key.part, &n0, &g0_func, &n1, &g1_func, &n2, &g2_func, &n3, &g3_func)); in PetscFEIntegrateBdJacobian_Basic()
[all …]
|
| /petsc/src/ksp/pc/interface/ |
| H A D | precon.c | 539 PetscInt m1, M1, m2, M2, n1, N1, n2, N2, m3, M3, n3, N3; in PCMatApplyTranspose_Private() local
550 PetscCall(MatGetLocalSize(A, &m3, &n3)); in PCMatApplyTranspose_Private()
557 … PetscInt_FMT ",%" PetscInt_FMT ")x(%" PetscInt_FMT ",%" PetscInt_FMT ")", m2, M2, m3, M3, n3, N3); in PCMatApplyTranspose_Private()
558 …n3 && M1 == N3, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Incompatible layout between block of output v… in PCMatApplyTranspose_Private()
|