| /petsc/src/mat/tests/ |
| H A D | ex137.c | 10 PetscScalar aa[7]; in main() local
31 aa[0] = 0; in main()
32 aa[1] = 1; in main()
33 aa[2] = 2; in main()
34 aa[3] = 3; in main()
35 aa[4] = 4; in main()
36 aa[5] = 5; in main()
37 aa[6] = 6; in main()
54 aa[0] = 8; in main()
55 aa[1] = 9; in main()
[all …]
|
| H A D | ex36f.F90 | 25 PetscScalar, pointer :: aa(:, :)
38 PetscCall(MatDenseGetArray(A, aa, ierr))
41 PetscCall(FillUpMatrix(m, n, aa))
43 PetscCall(MatDenseRestoreArray(A, aa, ierr))
67 PetscScalar aa(m, n)
70 PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, m, n, aa, A, ierr))
73 PetscCall(FillUpMatrix(m, n, aa))
|
| H A D | ex79f.F90 | 17 PetscScalar, pointer :: aa(:)
39 PetscCallA(MatSeqAIJGetArray(Ad, aa, ierr))
43 write (7 + rank, *) ' ', j, ja(j) + rstart, aa(j)
47 PetscCallA(MatSeqAIJRestoreArray(Ad, aa, ierr))
52 PetscCallA(MatSeqAIJGetArray(Ao, aa, ierr))
56 write (7 + rank, *) ' ', j, icol(ja(j)) + 1, aa(j)
61 PetscCallA(MatSeqAIJRestoreArray(Ao, aa, ierr))
|
| H A D | ex196f90.F90 | 14 PetscScalar, pointer :: aa(:)
29 PetscCallA(MatSeqAIJGetArray(A, aa, ierr))
30 print *, aa(3)
|
| /petsc/src/mat/impls/baij/seq/ |
| H A D | dgefa2.c | 18 MatScalar *aa, *ax, *ay, work[4], stmp; in PetscKernel_A_gets_inverse_A_2() local
35 aa = &a[k4]; in PetscKernel_A_gets_inverse_A_2()
36 max = PetscAbsScalar(aa[0]); in PetscKernel_A_gets_inverse_A_2()
39 tmp = PetscAbsScalar(aa[ll]); in PetscKernel_A_gets_inverse_A_2()
68 aa = &a[1 + k4]; in PetscKernel_A_gets_inverse_A_2()
69 for (ll = 0; ll < i__2; ll++) aa[ll] *= stmp; in PetscKernel_A_gets_inverse_A_2()
101 aa = &a[k3 + 1]; in PetscKernel_A_gets_inverse_A_2()
102 for (ll = 0; ll < i__2; ll++) aa[ll] *= stmp; in PetscKernel_A_gets_inverse_A_2()
105 ax = aa; in PetscKernel_A_gets_inverse_A_2()
119 aa = a + k3; in PetscKernel_A_gets_inverse_A_2()
[all …]
|
| H A D | dgefa3.c | 18 MatScalar *aa, *ax, *ay, work[9], stmp; in PetscKernel_A_gets_inverse_A_3() local
35 aa = &a[k4]; in PetscKernel_A_gets_inverse_A_3()
36 max = PetscAbsScalar(aa[0]); in PetscKernel_A_gets_inverse_A_3()
39 tmp = PetscAbsScalar(aa[ll]); in PetscKernel_A_gets_inverse_A_3()
69 aa = &a[1 + k4]; in PetscKernel_A_gets_inverse_A_3()
70 for (ll = 0; ll < i__2; ll++) aa[ll] *= stmp; in PetscKernel_A_gets_inverse_A_3()
102 aa = &a[k3 + 1]; in PetscKernel_A_gets_inverse_A_3()
103 for (ll = 0; ll < i__2; ll++) aa[ll] *= stmp; in PetscKernel_A_gets_inverse_A_3()
106 ax = aa; in PetscKernel_A_gets_inverse_A_3()
121 aa = a + k3; in PetscKernel_A_gets_inverse_A_3()
[all …]
|
| H A D | dgefa5.c | 18 MatScalar *aa, *ax, *ay, stmp; in PetscKernel_A_gets_inverse_A_5() local
35 aa = &a[k4]; in PetscKernel_A_gets_inverse_A_5()
36 max = PetscAbsScalar(aa[0]); in PetscKernel_A_gets_inverse_A_5()
39 tmp = PetscAbsScalar(aa[ll]); in PetscKernel_A_gets_inverse_A_5()
69 aa = &a[1 + k4]; in PetscKernel_A_gets_inverse_A_5()
70 for (ll = 0; ll < i__2; ll++) aa[ll] *= stmp; in PetscKernel_A_gets_inverse_A_5()
102 aa = &a[k3 + 1]; in PetscKernel_A_gets_inverse_A_5()
103 for (ll = 0; ll < i__2; ll++) aa[ll] *= stmp; in PetscKernel_A_gets_inverse_A_5()
106 ax = aa; in PetscKernel_A_gets_inverse_A_5()
121 aa = a + k3; in PetscKernel_A_gets_inverse_A_5()
[all …]
|
| H A D | dgefa4.c | 18 MatScalar *aa, *ax, *ay, work[16], stmp; in PetscKernel_A_gets_inverse_A_4() local
35 aa = &a[k4]; in PetscKernel_A_gets_inverse_A_4()
36 max = PetscAbsScalar(aa[0]); in PetscKernel_A_gets_inverse_A_4()
39 tmp = PetscAbsScalar(aa[ll]); in PetscKernel_A_gets_inverse_A_4()
69 aa = &a[1 + k4]; in PetscKernel_A_gets_inverse_A_4()
70 for (ll = 0; ll < i__2; ll++) aa[ll] *= stmp; in PetscKernel_A_gets_inverse_A_4()
102 aa = &a[k3 + 1]; in PetscKernel_A_gets_inverse_A_4()
103 for (ll = 0; ll < i__2; ll++) aa[ll] *= stmp; in PetscKernel_A_gets_inverse_A_4()
106 ax = aa; in PetscKernel_A_gets_inverse_A_4()
121 aa = a + k3; in PetscKernel_A_gets_inverse_A_4()
[all …]
|
| H A D | dgefa6.c | 18 MatScalar *aa, *ax, *ay, work[36], stmp; in PetscKernel_A_gets_inverse_A_6() local
35 aa = &a[k4]; in PetscKernel_A_gets_inverse_A_6()
36 max = PetscAbsScalar(aa[0]); in PetscKernel_A_gets_inverse_A_6()
39 tmp = PetscAbsScalar(aa[ll]); in PetscKernel_A_gets_inverse_A_6()
69 aa = &a[1 + k4]; in PetscKernel_A_gets_inverse_A_6()
70 for (ll = 0; ll < i__2; ll++) aa[ll] *= stmp; in PetscKernel_A_gets_inverse_A_6()
102 aa = &a[k3 + 1]; in PetscKernel_A_gets_inverse_A_6()
103 for (ll = 0; ll < i__2; ll++) aa[ll] *= stmp; in PetscKernel_A_gets_inverse_A_6()
106 ax = aa; in PetscKernel_A_gets_inverse_A_6()
121 aa = a + k3; in PetscKernel_A_gets_inverse_A_6()
[all …]
|
| H A D | dgefa7.c | 18 MatScalar *aa, *ax, *ay, work[49], stmp; in PetscKernel_A_gets_inverse_A_7() local
35 aa = &a[k4]; in PetscKernel_A_gets_inverse_A_7()
36 max = PetscAbsScalar(aa[0]); in PetscKernel_A_gets_inverse_A_7()
39 tmp = PetscAbsScalar(aa[ll]); in PetscKernel_A_gets_inverse_A_7()
69 aa = &a[1 + k4]; in PetscKernel_A_gets_inverse_A_7()
70 for (ll = 0; ll < i__2; ll++) aa[ll] *= stmp; in PetscKernel_A_gets_inverse_A_7()
102 aa = &a[k3 + 1]; in PetscKernel_A_gets_inverse_A_7()
103 for (ll = 0; ll < i__2; ll++) aa[ll] *= stmp; in PetscKernel_A_gets_inverse_A_7()
106 ax = aa; in PetscKernel_A_gets_inverse_A_7()
121 aa = a + k3; in PetscKernel_A_gets_inverse_A_7()
[all …]
|
| H A D | dgedi.c | 20 MatScalar *aa, *ax, *ay, tmp; in PetscLINPACKgedi() local
36 aa = &a[1 + kn]; in PetscLINPACKgedi()
37 for (ll = 0; ll < i__2; ll++) aa[ll] *= t; in PetscLINPACKgedi()
40 ax = aa; in PetscLINPACKgedi()
58 aa = a + kn; in PetscLINPACKgedi()
60 work[i] = aa[i]; in PetscLINPACKgedi()
61 aa[i] = 0.; in PetscLINPACKgedi()
|
| H A D | dgefa.c | 18 MatScalar t, *ax, *ay, *aa; in PetscLINPACKgefa() local
38 aa = &a[knp1]; in PetscLINPACKgefa()
39 max = PetscAbsScalar(aa[0]); in PetscLINPACKgefa()
42 tmp = PetscAbsScalar(aa[ll]); in PetscLINPACKgefa()
67 aa = &a[1 + knp1]; in PetscLINPACKgefa()
68 for (ll = 0; ll < i__2; ll++) aa[ll] *= t; in PetscLINPACKgefa()
71 ax = aa; in PetscLINPACKgefa()
|
| H A D | baijsolvnat1.c | 12 const MatScalar *aa = a->a, *v; in MatSolve_SeqBAIJ_1_NaturalOrdering_inplace() local
26 v = aa + ai[i]; in MatSolve_SeqBAIJ_1_NaturalOrdering_inplace()
41 v = aa + diag[i] + 1; in MatSolve_SeqBAIJ_1_NaturalOrdering_inplace()
52 v = aa + diag[i]; in MatSolve_SeqBAIJ_1_NaturalOrdering_inplace()
67 const MatScalar *aa = a->a, *v; in MatForwardSolve_SeqBAIJ_1_NaturalOrdering() local
78 v = aa; in MatForwardSolve_SeqBAIJ_1_NaturalOrdering()
100 const MatScalar *aa = a->a, *v; in MatBackwardSolve_SeqBAIJ_1_NaturalOrdering() local
111 v = aa + adiag[i + 1] + 1; in MatBackwardSolve_SeqBAIJ_1_NaturalOrdering()
131 const MatScalar *aa = a->a, *v; in MatSolve_SeqBAIJ_1_NaturalOrdering() local
142 v = aa; in MatSolve_SeqBAIJ_1_NaturalOrdering()
[all …]
|
| H A D | baijsolvtrannat1.c | 10 const MatScalar *aa = a->a, *v; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering() local
23 v = aa + adiag[i + 1] + 1; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering()
34 v = aa + ai[i]; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering()
56 const MatScalar *aa = a->a, *v; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering_inplace() local
65 v = aa + diag[i]; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering_inplace()
75 v = aa + diag[i] - 1; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering_inplace()
|
| H A D | baijsolvnat2.c | 16 const MatScalar *aa = a->a, *v; in MatSolve_SeqBAIJ_2_NaturalOrdering_inplace() local
30 v = aa + 4 * ai[i]; in MatSolve_SeqBAIJ_2_NaturalOrdering_inplace()
49 v = aa + 4 * diag[i] + 4; in MatSolve_SeqBAIJ_2_NaturalOrdering_inplace()
63 v = aa + 4 * diag[i]; in MatSolve_SeqBAIJ_2_NaturalOrdering_inplace()
79 const MatScalar *aa = a->a, *v; in MatSolve_SeqBAIJ_2_NaturalOrdering() local
91 v = aa + 4 * ai[i]; in MatSolve_SeqBAIJ_2_NaturalOrdering()
113 v = aa + 4 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_2_NaturalOrdering()
145 const MatScalar *aa = a->a, *v; in MatForwardSolve_SeqBAIJ_2_NaturalOrdering() local
157 v = aa + 4 * ai[i]; in MatForwardSolve_SeqBAIJ_2_NaturalOrdering()
188 const MatScalar *aa = a->a, *v; in MatBackwardSolve_SeqBAIJ_2_NaturalOrdering() local
[all …]
|
| H A D | baijsolvtrann.c | 11 const MatScalar *aa = a->a, *v; in MatSolveTranspose_SeqBAIJ_N_inplace() local
35 PetscKernel_w_gets_transA_times_v(bs, ls, aa + bs2 * a->diag[i], t + i * bs); in MatSolveTranspose_SeqBAIJ_N_inplace()
36 v = aa + bs2 * (a->diag[i] + 1); in MatSolveTranspose_SeqBAIJ_N_inplace()
47 v = aa + bs2 * ai[i]; in MatSolveTranspose_SeqBAIJ_N_inplace()
77 const MatScalar *aa = a->a, *v; in MatSolveTranspose_SeqBAIJ_N() local
101 PetscKernel_w_gets_transA_times_v(bs, ls, aa + bs2 * diag[i], t + i * bs); in MatSolveTranspose_SeqBAIJ_N()
102 v = aa + bs2 * (diag[i] - 1); in MatSolveTranspose_SeqBAIJ_N()
113 v = aa + bs2 * ai[i]; in MatSolveTranspose_SeqBAIJ_N()
|
| /petsc/src/mat/impls/sbaij/seq/ |
| H A D | sbaijfact2.c | 17 const MatScalar *aa = a->a, *v, *diag; in MatSolve_SeqSBAIJ_N_inplace() local
36 v = aa + bs2 * ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
50 diag = aa + k * bs2; /* ptr to inv(Dk) */ in MatSolve_SeqSBAIJ_N_inplace()
56 v = aa + bs2 * ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
92 …N_NaturalOrdering(const PetscInt *ai, const PetscInt *aj, const MatScalar *aa, PetscInt mbs, Petsc… in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering() argument
103 v = aa + bs2 * ai[k]; in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering()
117 diag = aa + k * bs2; /* ptr to inv(Dk) */ in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering()
124 …N_NaturalOrdering(const PetscInt *ai, const PetscInt *aj, const MatScalar *aa, PetscInt mbs, Petsc… in MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering() argument
133 v = aa + bs2 * ai[k]; in MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering()
154 const MatScalar *aa = a->a; in MatSolve_SeqSBAIJ_N_NaturalOrdering_inplace() local
[all …]
|
| H A D | sbaijfact3.c | 11 MatScalar *ba = b->a, *aa, *ap, *dk, *uik; in MatCholeskyFactorNumeric_SeqSBAIJ_3() local
31 aa = a->a; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
35 PetscCall(PetscMalloc1(9 * ai[mbs], &aa)); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
36 PetscCall(PetscArraycpy(aa, a->a, 9 * ai[mbs])); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
49 dk[k1] = aa[k * 9 + k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
50 aa[k * 9 + k1] = aa[j * 9 + k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
51 aa[j * 9 + k1] = dk[k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
57 ap = aa + j * 9; /* ptr to the beginning of j-th block of aa */ in MatCholeskyFactorNumeric_SeqSBAIJ_3()
74 ap = aa + jmin * 9; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
189 if (a->permute) PetscCall(PetscFree(aa)); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
|
| /petsc/src/mat/impls/aij/seq/ftn-kernels/ |
| H A D | fsolve.F90 | 12 PetscScalar, intent(in) :: aa(0:*), b(0:*)
24 x(i) = b(i) - sum(aa(jstart:jend)*x(aj(jstart:jend)))
32 x(i) = x(i) - sum(aa(jstart:jend)*x(aj(jstart:jend)))*aa(adiag(i))
|
| /petsc/src/dm/impls/da/ftn-custom/ |
| H A D | zda1f90.c | 69 PetscScalar *aa; in dmdavecgetarray1_() local
92 *ierr = VecGetArray(*v, &aa); in dmdavecgetarray1_()
94 *ierr = F90Array1dCreate(aa, MPIU_SCALAR, gxs, gxm, a PETSC_F90_2PTR_PARAM(ptrd)); in dmdavecgetarray1_()
110 PetscScalar *aa; in dmdavecgetarray2_() local
139 *ierr = VecGetArray(*v, &aa); in dmdavecgetarray2_()
141 *ierr = F90Array2dCreate(aa, MPIU_SCALAR, gxs, gxm, gys, gym, a PETSC_F90_2PTR_PARAM(ptrd)); in dmdavecgetarray2_()
157 PetscScalar *aa; in dmdavecgetarray3_() local
188 *ierr = VecGetArray(*v, &aa); in dmdavecgetarray3_()
190 …*ierr = F90Array3dCreate(aa, MPIU_SCALAR, gxs, gxm, gys, gym, gzs, gzm, a PETSC_F90_2PTR_PARAM(ptr… in dmdavecgetarray3_()
206 PetscScalar *aa; in dmdavecgetarray4_() local
[all …]
|
| /petsc/src/sys/utils/ |
| H A D | server.c | 370 PetscScalar *aa; in petscshmgetallocatearrayscalar_() local
372 *ierr = PetscShmgetAllocateArray(*len, sizeof(PetscScalar), (void **)&aa); in petscshmgetallocatearrayscalar_()
374 *ierr = F90Array1dCreate(aa, MPIU_SCALAR, *start, *len, a PETSC_F90_2PTR_PARAM(ptrd)); in petscshmgetallocatearrayscalar_()
379 PetscScalar *aa; in petscshmgetdeallocatearrayscalar_() local
381 *ierr = F90Array1dAccess(a, MPIU_SCALAR, (void **)&aa PETSC_F90_2PTR_PARAM(ptrd)); in petscshmgetdeallocatearrayscalar_()
383 *ierr = PetscShmgetDeallocateArray((void **)&aa); in petscshmgetdeallocatearrayscalar_()
390 PetscInt *aa; in petscshmgetallocatearrayint_() local
392 *ierr = PetscShmgetAllocateArray(*len, sizeof(PetscInt), (void **)&aa); in petscshmgetallocatearrayint_()
394 *ierr = F90Array1dCreate(aa, MPIU_INT, *start, *len, a PETSC_F90_2PTR_PARAM(ptrd)); in petscshmgetallocatearrayint_()
399 PetscInt *aa; in petscshmgetdeallocatearrayint_() local
[all …]
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | aijfact.c | 228 const MatScalar *aa, *v; in MatLUFactorNumeric_SeqAIJ() local
237 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatLUFactorNumeric_SeqAIJ()
247 d = aa[ddiag[i]]; in MatLUFactorNumeric_SeqAIJ()
249 v = aa + ai[i]; in MatLUFactorNumeric_SeqAIJ()
282 v = aa + ai[r[i]]; in MatLUFactorNumeric_SeqAIJ()
354 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatLUFactorNumeric_SeqAIJ()
407 const MatScalar *v, *aa; in MatLUFactorNumeric_SeqAIJ_inplace() local
416 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatLUFactorNumeric_SeqAIJ_inplace()
426 d = aa[ddiag[i]]; in MatLUFactorNumeric_SeqAIJ_inplace()
428 v = aa + ai[i]; in MatLUFactorNumeric_SeqAIJ_inplace()
[all …]
|
| H A D | aij.c | 88 const MatScalar *aa; in MatFindZeroDiagonals_SeqAIJ_Private() local
94 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatFindZeroDiagonals_SeqAIJ_Private()
97 if ((diag[i] >= ii[i + 1]) || (jj[diag[i]] != i) || (aa[diag[i]] == 0.0)) cnt++; in MatFindZeroDiagonals_SeqAIJ_Private()
102 if ((diag[i] >= ii[i + 1]) || (jj[diag[i]] != i) || (aa[diag[i]] == 0.0)) rows[cnt++] = i; in MatFindZeroDiagonals_SeqAIJ_Private()
106 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatFindZeroDiagonals_SeqAIJ_Private()
124 const MatScalar *aa; in MatFindNonzeroRows_SeqAIJ() local
130 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatFindNonzeroRows_SeqAIJ()
140 if (aa[j] != 0.0) goto ok1; in MatFindNonzeroRows_SeqAIJ()
146 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatFindNonzeroRows_SeqAIJ()
155 if (aa[j] != 0.0) { in MatFindNonzeroRows_SeqAIJ()
[all …]
|
| /petsc/lib/petsc/bin/maint/ |
| H A D | doublelinecheck.py | 7 aa = a.split(':') variable
8 f = aa[0]
9 num = int(aa[1])
|
| /petsc/src/mat/impls/aij/seq/aijperm/ |
| H A D | aijperm.c | 281 const MatScalar *aa; in MatMult_SeqAIJPERM() local
310 #pragma disjoint(*x, *y, *aa) in MatMult_SeqAIJPERM()
317 aa = a->a; /* Nonzero elements stored row-by-row. */ in MatMult_SeqAIJPERM()
339 y[iold] = aa[ipos] * x[aj[ipos]]; in MatMult_SeqAIJPERM()
382 vec_vals = _mm512_loadu_pd(&aa[ipos]); in MatMult_SeqAIJPERM()
390 vec_vals = _mm512_loadu_pd(&aa[ipos]); in MatMult_SeqAIJPERM()
394 yp[i] += aa[ipos] * x[aj[ipos]]; in MatMult_SeqAIJPERM()
395 yp[i] += aa[ipos + 1] * x[aj[ipos + 1]]; in MatMult_SeqAIJPERM()
397 yp[i] += aa[ipos] * x[aj[ipos]]; in MatMult_SeqAIJPERM()
403 yp[i] += aa[ipos] * x[aj[ipos]]; in MatMult_SeqAIJPERM()
[all …]
|