Home
last modified time | relevance | path

Searched refs:AA (Results 1 – 25 of 38) sorted by relevance

12

/petsc/src/snes/tutorials/ !
H A Dex75.c59 PetscScalar a, b, r, _aa, _bb, AA, BB, Rm; in SolKxSolution() local
118 AA = _aa; in SolKxSolution()
152 t92 = t65 * t76 * AA; in SolKxSolution()
161 t127 = t110 * t115 * AA; in SolKxSolution()
163AA * t9 + ((0.2e1 * Rp * (-B * t13 + 0.3e1 * t15 * B - t19 - 0.2e1 * t20 - 0.2e1 * t22 - t24) * AA in SolKxSolution()
174 t12 = t2 * t10 * AA; in SolKxSolution()
226AA + 0.8e1 * B * BB * t32 * Rp) * t38 + (0.2e1 * Rp * (-B * t8 + 0.3e1 * t41 - t9 * B - 0.2e1 * t4… in SolKxSolution()
259 t90 = t63 * t74 * AA; in SolKxSolution()
268 t125 = t108 * t113 * AA; in SolKxSolution()
270AA * t7 + ((-0.2e1 * Rp * (-t11 - t13 + 0.2e1 * t15 + 0.3e1 * t14 * B - B * t19 + 0.2e1 * t21) * A… in SolKxSolution()
[all …]
H A Dex69.c203 PetscReal a, b, r, _aa, _bb, AA, BB, Rm; in SolKxSolution() local
270 AA = _aa; in SolKxSolution()
307 t92 = t65 * t76 * AA; in SolKxSolution()
316 t127 = t110 * t115 * AA; in SolKxSolution()
318AA * t9 + ((0.2e1 * Rp * (0.3e1 * t13 * B - 0.2e1 * t16 - t19 - 0.2e1 * t20 - B * t22 - t24) * AA in SolKxSolution()
329 t12 = t2 * t10 * AA; in SolKxSolution()
379AA + 0.8e1 * B * BB * t32 * Rp) * t38 + (0.2e1 * Rp * (0.3e1 * t40 - 0.2e1 * Rp * t7 - t44 - 0.2e1… in SolKxSolution()
412 t90 = t63 * t74 * AA; in SolKxSolution()
421 t125 = t108 * t113 * AA; in SolKxSolution()
423AA * t7 + ((-0.2e1 * Rp * (-B * t11 - t13 - t15 + 0.2e1 * t17 + 0.3e1 * t16 * B + 0.2e1 * t21) * A… in SolKxSolution()
[all …]
/petsc/src/mat/tutorials/ !
H A Dex8.c17 Mat AA, AB; in MatScaleUserImpl_MPIAIJ() local
20 PetscCall(MatMPIAIJGetSeqAIJ(A, &AA, &AB, NULL)); in MatScaleUserImpl_MPIAIJ()
21 PetscCall(MatScaleUserImpl(AA, aa)); in MatScaleUserImpl_MPIAIJ()
38 Mat AA, AB; in RegisterMatScaleUserImpl() local
39 PetscCall(MatMPIAIJGetSeqAIJ(mat, &AA, &AB, NULL)); in RegisterMatScaleUserImpl()
41 …PetscCall(PetscObjectComposeFunction((PetscObject)AA, "MatScaleUserImpl_C", MatScaleUserImpl_SeqAI… in RegisterMatScaleUserImpl()
59 Mat AA, AB; in DeRegisterMatScaleUserImpl() local
60 PetscCall(MatMPIAIJGetSeqAIJ(mat, &AA, &AB, NULL)); in DeRegisterMatScaleUserImpl()
62 PetscCall(PetscObjectComposeFunction((PetscObject)AA, "MatScaleUserImpl_C", NULL)); in DeRegisterMatScaleUserImpl()
/petsc/src/ksp/ksp/tutorials/ !
H A Dex2f.F90115 Mat A, AA
322 PetscCallA(MatCreate(PETSC_COMM_SELF, AA, ierr))
323 PetscCallA(MatSetSizes(AA, PETSC_DECIDE, PETSC_DECIDE, m, m, ierr))
324 PetscCallA(MatSetFromOptions(AA, ierr))
327 PetscCallA(MatSetValues(AA, ione, [izero], itwo, [izero, ione], val(2:3), INSERT_VALUES, ierr))
330 PetscCallA(MatSetValues(AA, ione, [i], itwo, col, val, INSERT_VALUES, ierr))
332 PetscCallA(MatSetValues(AA, ione, [m - 1], itwo, [m - 2, m - 1], val(1:2), INSERT_VALUES, ierr))
333 PetscCallA(MatAssemblyBegin(AA, MAT_FINAL_ASSEMBLY, ierr))
334 PetscCallA(MatAssemblyEnd(AA, MAT_FINAL_ASSEMBLY, ierr))
342 PetscCallA(MatMult(AA, uu, bb, ierr))
[all …]
H A Dex19.c16 Mat A, Pmat, Aseq, AA; /* linear system matrix */ in main() local
100 PetscCall(MatConvert(nest, MATAIJKOKKOS, MAT_INITIAL_MATRIX, &AA)); in main()
114 PetscCall(MatAXPY(AAseq, 1.0, AA, DIFFERENT_NONZERO_PATTERN)); in main()
115 PetscCall(MatDestroy(&AA)); in main()
117 Pmat = AA; in main()
H A Dex71.c396 Mat AA; in main() local
398 PetscCall(MatConvert(A, MATAIJ, MAT_INITIAL_MATRIX, &AA)); in main()
399 PetscCall(MatViewFromOptions(AA, NULL, "-assembled_view")); in main()
400 PetscCall(MatDestroy(&AA)); in main()
H A Dex49.c75 static PetscErrorCode DMDABCApplySymmetricCompression(DM elas_da, Mat A, Vec f, IS *dofs, Mat *AA, …
984 Mat AA; in solve_elasticity_2d() local
989 PetscCall(DMDABCApplySymmetricCompression(elas_da, A, f, &is, &AA, &ff)); in solve_elasticity_2d()
992 PetscCall(KSPSetOperators(ksp_E, AA, AA)); in solve_elasticity_2d()
1005 PetscCall(MatDestroy(&AA)); in solve_elasticity_2d()
1205 static PetscErrorCode DMDABCApplySymmetricCompression(DM elas_da, Mat A, Vec f, IS *dofs, Mat *AA, … in DMDABCApplySymmetricCompression() argument
1252 PetscCall(MatCreateSubMatrix(A, is, is, MAT_INITIAL_MATRIX, AA)); in DMDABCApplySymmetricCompression()
1254 PetscCall(MatCreateVecs(*AA, NULL, ff)); in DMDABCApplySymmetricCompression()
1276 PetscCall(MatSetNearNullSpace(*AA, unull)); in DMDABCApplySymmetricCompression()
/petsc/src/mat/tests/ !
H A Dex110.c15 Mat AA, AB; in main() local
42 PetscCall(MatMPIAIJGetSeqAIJ(A, &AA, &AB, &garray)); in main()
43 …PetscCall(MatGetRowIJ(AA, 0, PETSC_FALSE, PETSC_FALSE, &nd, (const PetscInt **)&di, (const PetscIn… in main()
44 PetscCall(MatSeqAIJGetArray(AA, &da)); in main()
57 …PetscCall(MatRestoreRowIJ(AA, 0, PETSC_FALSE, PETSC_FALSE, &nd, (const PetscInt **)&di, (const Pet… in main()
58 PetscCall(MatSeqAIJRestoreArray(AA, &da)); in main()
H A Dex89.c59 Mat AA, AB; in main() local
60 PetscCall(MatMPIAIJGetSeqAIJ(A, &AA, &AB, NULL)); in main()
61 PetscCall(MatGetRowIJ(AA, 0, PETSC_FALSE, PETSC_FALSE, &nrows, &ia, &ja, &flg)); in main()
63 PetscCall(MatSeqAIJGetArray(AA, &array)); in main()
65 PetscCall(MatSeqAIJRestoreArray(AA, &array)); in main()
67 PetscCall(MatRestoreRowIJ(AA, 0, PETSC_FALSE, PETSC_FALSE, &nrows, &ia, &ja, &flg)); in main()
H A Dex302k.kokkos.cxx23 Mat AA, AB; in main() local
62 PetscCall(MatMPIAIJGetSeqAIJ(A, &AA, &AB, &garray)); in main()
63 …PetscCall(MatGetRowIJ(AA, 0, PETSC_FALSE, PETSC_FALSE, &nd, (const PetscInt **)&di, (const PetscIn… in main()
64 PetscCall(MatSeqAIJGetArray(AA, &da)); in main()
143 …PetscCall(MatRestoreRowIJ(AA, 0, PETSC_FALSE, PETSC_FALSE, &nd, (const PetscInt **)&di, (const Pet… in main()
144 PetscCall(MatSeqAIJRestoreArray(AA, &da)); in main()
H A Dex111.c107 Mat AA, AB; in main() local
108 PetscCall(MatMPIAIJGetSeqAIJ(A, &AA, &AB, NULL)); in main()
109 PetscCall(MatGetRowIJ(AA, 0, PETSC_FALSE, PETSC_FALSE, &nrows, &ia, &ja, &flg)); in main()
111 PetscCall(MatSeqAIJGetArray(AA, &array)); in main()
113 PetscCall(MatSeqAIJRestoreArray(AA, &array)); in main()
115 PetscCall(MatRestoreRowIJ(AA, 0, PETSC_FALSE, PETSC_FALSE, &nrows, &ia, &ja, &flg)); in main()
H A Dex96.c117 Mat AA, AB; in main() local
118 PetscCall(MatMPIAIJGetSeqAIJ(A, &AA, &AB, NULL)); in main()
119 PetscCall(MatGetRowIJ(AA, 0, PETSC_FALSE, PETSC_FALSE, &nrows, &ia, &ja, &flg)); in main()
121 PetscCall(MatSeqAIJGetArray(AA, &array)); in main()
123 PetscCall(MatSeqAIJRestoreArray(AA, &array)); in main()
125 PetscCall(MatRestoreRowIJ(AA, 0, PETSC_FALSE, PETSC_FALSE, &nrows, &ia, &ja, &flg)); in main()
/petsc/src/ts/tutorials/power_grid/ !
H A Dex7.m8 AA = PetscBinaryRead('binaryoutput','cell',10000); variable
9 l = size(AA); l = l(2);
11 for i=1:l-80-1; A{1,i} = AA{i+79}; end;
/petsc/src/dm/tests/ !
H A Dex27.c22 Mat A, AA; in main() local
123 PetscCall(MatCreateSeqFFTW(PETSC_COMM_SELF, 3, dim, &AA)); in main()
127 PetscCall(MatMult(AA, xx, zz)); in main()
131 PetscCall(MatMult(AA, xx, yy)); in main()
133 PetscCall(MatMultTranspose(AA, yy, zz)); in main()
H A Dex28.c24 Mat A, AA; in main() local
143 PetscCall(MatCreateSeqFFTW(PETSC_COMM_SELF, 3, dim, &AA)); in main()
147 for (ii = 0; ii < 3; ++ii) PetscCall(MatMult(AA, xxsplit[ii], zzsplit[ii])); in main()
151 for (ii = 0; ii < 3; ++ii) PetscCall(MatMult(AA, xxsplit[ii], yysplit[ii])); in main()
153 for (ii = 0; ii < 3; ++ii) PetscCall(MatMult(AA, yysplit[ii], zzsplit[ii])); in main()
/petsc/src/ksp/pc/impls/gamg/ !
H A Dgeo.c286 PetscScalar AA[3][3]; in triangulateAndFormProl() local
311 AA[tt][0] = coords[lid2]; in triangulateAndFormProl()
312 AA[tt][1] = coords[data_stride + lid2]; in triangulateAndFormProl()
313 AA[tt][2] = 1.0; in triangulateAndFormProl()
320 …PetscCallBLAS("LAPACKgesv", LAPACKgesv_(&N, &NRHS, (PetscScalar *)AA, &LDA, IPIV, alpha, &LDB, &IN… in triangulateAndFormProl()
342 AA[tt][0] = coords[lid2]; in triangulateAndFormProl()
343 AA[tt][1] = coords[data_stride + lid2]; in triangulateAndFormProl()
344 AA[tt][2] = 1.0; in triangulateAndFormProl()
349 …PetscCallBLAS("LAPACKgesv", LAPACKgesv_(&N, &NRHS, (PetscScalar *)AA, &LDA, IPIV, alpha, &LDB, &IN… in triangulateAndFormProl()
371 AA[tt][0] = coords[lid2]; in triangulateAndFormProl()
[all …]
/petsc/src/binding/petsc4py/test/ !
H A Dtest_mat_py.py396 AA = self.A.getPythonContext()
397 if not hasattr(AA, 'mult'):
399 AA.multTranspose = AA.mult
432 del AA.multTranspose
465 AA = self.A.getPythonContext()
466 AA.multTranspose = AA.mult
468 del AA.multTranspose
593 AA = self.A.getPythonContext()
594 AA.multTranspose = AA.mult
596 del AA.multTranspose
/petsc/src/ts/utils/dmplexlandau/tutorials/ !
H A Dex1.c136 PetscReal AA, v_bar_ab, vTe, t1, TeDiff, Te, Ti, Tdiff; in FormFunction() local
147 AA = tPerp / tPar - 1; in FormFunction()
148 if (AA < 0) ff = PetscAtanhReal(PetscSqrtReal(-AA)) / PetscSqrtReal(-AA); in FormFunction()
149 else ff = PetscAtanReal(PetscSqrtReal(AA)) / PetscSqrtReal(AA); in FormFunction()
150 t1 = (-3 + (AA + 3) * ff) / PetscSqr(AA); in FormFunction()
/petsc/src/ts/tests/ !
H A Dex12.c425 PetscErrorCode RHSJacobian(TS ts, PetscReal t, Vec global_in, Mat AA, Mat BB, PetscCtx ctx) in RHSJacobian() argument
517 if (BB != AA) { in RHSJacobian()
518 PetscCall(MatAssemblyBegin(AA, MAT_FINAL_ASSEMBLY)); in RHSJacobian()
519 PetscCall(MatAssemblyEnd(AA, MAT_FINAL_ASSEMBLY)); in RHSJacobian()
/petsc/src/ts/tutorials/ !
H A Dex2.c520 PetscErrorCode RHSJacobian(TS ts, PetscReal t, Vec global_in, Mat AA, Mat BB, PetscCtx ctx) in RHSJacobian() argument
612 if (BB != AA) { in RHSJacobian()
613 PetscCall(MatAssemblyBegin(AA, MAT_FINAL_ASSEMBLY)); in RHSJacobian()
614 PetscCall(MatAssemblyEnd(AA, MAT_FINAL_ASSEMBLY)); in RHSJacobian()
H A Dex5.c416 PetscErrorCode RHSMatrixHeat(TS ts, PetscReal t, Vec X, Mat AA, Mat BB, PetscCtx ctx) in RHSMatrixHeat() argument
418 Mat A = AA; /* Jacobian matrix */ in RHSMatrixHeat()
H A Dex6.c425 PetscErrorCode RHSMatrixHeat(TS ts, PetscReal t, Vec X, Mat AA, Mat BB, PetscCtx ctx) in RHSMatrixHeat() argument
427 Mat A = AA; /* Jacobian matrix */ in RHSMatrixHeat()
/petsc/src/dm/dt/interface/ !
H A Ddt.c2974 …LegendreElementLaplacianCreate(PetscInt n, PetscReal nodes[], PetscReal weights[], PetscReal ***AA) in PetscGaussLobattoLegendreElementLaplacianCreate() argument
3043 *AA = A; in PetscGaussLobattoLegendreElementLaplacianCreate()
3062 …egendreElementLaplacianDestroy(PetscInt n, PetscReal nodes[], PetscReal weights[], PetscReal ***AA) in PetscGaussLobattoLegendreElementLaplacianDestroy() argument
3065 PetscCall(PetscFree((*AA)[0])); in PetscGaussLobattoLegendreElementLaplacianDestroy()
3066 PetscCall(PetscFree(*AA)); in PetscGaussLobattoLegendreElementLaplacianDestroy()
3067 *AA = NULL; in PetscGaussLobattoLegendreElementLaplacianDestroy()
3094 …adientCreate(PetscInt n, PetscReal nodes[], PetscReal weights[], PetscReal ***AA, PetscReal ***AAT) in PetscGaussLobattoLegendreElementGradientCreate() argument
3127 *AA = A; in PetscGaussLobattoLegendreElementGradientCreate()
3147 …dientDestroy(PetscInt n, PetscReal nodes[], PetscReal weights[], PetscReal ***AA, PetscReal ***AAT) in PetscGaussLobattoLegendreElementGradientDestroy() argument
3150 PetscCall(PetscFree((*AA)[0])); in PetscGaussLobattoLegendreElementGradientDestroy()
[all …]
/petsc/src/mat/impls/aij/seq/ !
H A Daij.h222 static inline PetscErrorCode MatSeqXAIJFreeAIJ(Mat AA, MatScalar **a, PetscInt **j, PetscInt **i) in MatSeqXAIJFreeAIJ() argument
224 Mat_SeqAIJ *A = (Mat_SeqAIJ *)AA->data; in MatSeqXAIJFreeAIJ()
236 #define MatSeqXAIJReallocateAIJ(Amat, AM, BS2, NROW, ROW, COL, RMAX, AA, AI, AJ, RP, AP, AIMAX, NON… argument
256 PetscCall(PetscArraycpy(new_a, AA, BS2 * (AI[ROW] + NROW))); \
258 …cpy(new_a + BS2 * (AI[ROW] + NROW + CHUNKSIZE), PetscSafePointerPlusOffset(AA, BS2 * (AI[ROW] + NR…
261 AA = new_a; \
267 AP = AA + BS2 * AI[ROW]; \
/petsc/src/ts/tutorials/advection-diffusion-reaction/ !
H A Dex3.c256 PetscErrorCode RHSMatrixHeat(TS ts, PetscReal t, Vec U, Mat AA, Mat BB, PetscCtx ctx) in RHSMatrixHeat() argument
258 Mat A = AA; /* Jacobian matrix */ in RHSMatrixHeat()

12