Home
last modified time | relevance | path

Searched refs:ai (Results 1 – 25 of 109) sorted by relevance

12345

/petsc/src/tao/util/
H A Dtao_util.c252 PetscReal ai, bi, ci, di, ei; in MatDFischer() local
294 ai = fischnorm(1.0, PetscRealPart(t2[i])); in MatDFischer()
296 da[i] = -1.0 / ai - 1.0; in MatDFischer()
297 db[i] = -t2[i] / ai - 1.0; in MatDFischer()
300 ai = fischnorm(bi, PetscRealPart(f[i])); in MatDFischer()
301 ai = PetscMax(PETSC_MACHINE_EPSILON, ai); in MatDFischer()
303 da[i] = bi / ai - 1.0; in MatDFischer()
304 db[i] = -f[i] / ai - 1.0; in MatDFischer()
308 ai = fischnorm(1.0, PetscRealPart(t2[i])); in MatDFischer()
310 da[i] = 1.0 / ai - 1.0; in MatDFischer()
[all …]
/petsc/src/mat/impls/sbaij/seq/
H A Dsro.c41 PetscInt *ai,*aj,*r; in MatReorderingSeqSBAIJ()
54 PetscCall(PetscMalloc2(mbs+1,&ai, 2*a->i[mbs],&aj)); in MatReorderingSeqSBAIJ()
56 ai = a->inew; aj = a->jnew; in MatReorderingSeqSBAIJ()
58 PetscCall(PetscArraycpy(ai,a->i,mbs+1)); in MatReorderingSeqSBAIJ()
69 PetscCall(PetscMalloc1(ai[mbs],&r)); in MatReorderingSeqSBAIJ()
71 for (i=0; i<ai[mbs]; i++) r[i] = 0; in MatReorderingSeqSBAIJ()
75 nz = ai[i+1] - ai[i]; in MatReorderingSeqSBAIJ()
76 j = ai[i]; in MatReorderingSeqSBAIJ()
98 ai[i+1] = ai[i] + nzr[i]; in MatReorderingSeqSBAIJ()
99 nzr[i] = ai[i+1]; in MatReorderingSeqSBAIJ()
[all …]
H A Dsbaijfact2.c13 PetscInt mbs = a->mbs, *ai = a->i, *aj = a->j; in MatSolve_SeqSBAIJ_N_inplace() local
36 v = aa + bs2 * ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
39 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
40 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
56 v = aa + bs2 * ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
58 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
59 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
92 static PetscErrorCode MatForwardSolve_SeqSBAIJ_N_NaturalOrdering(const PetscInt *ai, const PetscInt… in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering() argument
103 v = aa + bs2 * ai[k]; in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering()
106 nz = ai[k + 1] - ai[k]; in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering()
[all …]
H A Drelax.h65 const PetscInt *ai = a->i; in MatMult_SeqSBAIJ_1_ushort() local
83 nz = ai[i + 1] - ai[i]; /* length of i_th row of A */ in MatMult_SeqSBAIJ_1_ushort()
133 const PetscInt *ai = a->i; in MatSOR_SeqSBAIJ_ushort() local
168 d = fshift + aa[ai[i]]; in MatSOR_SeqSBAIJ_ushort()
169 nz = ai[i + 1] - ai[i] - 1; in MatSOR_SeqSBAIJ_ushort()
170 vj = aj + ai[i] + 1; in MatSOR_SeqSBAIJ_ushort()
171 v = aa + ai[i] + 1; in MatSOR_SeqSBAIJ_ushort()
190 nz = ai[i + 1] - ai[i] - 1; in MatSOR_SeqSBAIJ_ushort()
202 v = aa + ai[m] - 1; in MatSOR_SeqSBAIJ_ushort()
203 vj = aj + ai[m] - 1; in MatSOR_SeqSBAIJ_ushort()
[all …]
H A Daijsbaij.c10 …PetscInt *ai = a->i, *aj = a->j, m = A->rmap->N, n = A->cmap->n, i, j, k, *bi, *bj, *rowlength… in MatConvert_SeqSBAIJ_SeqAIJ() local
22 nz = ai[i + 1] - ai[i]; in MatConvert_SeqSBAIJ_SeqAIJ()
69 nz = ai[i + 1] - ai[i]; in MatConvert_SeqSBAIJ_SeqAIJ()
126 const PetscInt *ai = Aa->i, *aj = Aa->j; in MatConvert_SeqAIJ_SeqSBAIJ_Preallocate() local
137 if (adiag[i] == ai[i + 1]) { in MatConvert_SeqAIJ_SeqSBAIJ_Preallocate()
139 for (PetscInt j = ai[i]; j < ai[i + 1]; j++) (*nnz)[i] += (aj[j] > i); in MatConvert_SeqAIJ_SeqSBAIJ_Preallocate()
140 } else (*nnz)[i] = ai[i + 1] - adiag[i]; in MatConvert_SeqAIJ_SeqSBAIJ_Preallocate()
151 for (PetscInt k = ai[i]; k < ai[i + 1]; k++) { in MatConvert_SeqAIJ_SeqSBAIJ_Preallocate()
168 …PetscInt *ai = a->i, *aj, m = A->rmap->N, n = A->cmap->N, i, j, *bi, *bj, *rowlengths, bs = … in MatConvert_SeqAIJ_SeqSBAIJ() local
184 if (adiag[i] == ai[i + 1]) { /* missing diagonal */ in MatConvert_SeqAIJ_SeqSBAIJ()
[all …]
H A Dsbaijfact.c42 const PetscInt *rip, *ai, *aj; in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR() local
57 ai = a->i; in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
64 ai = a->inew; in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
70 umax = (PetscInt)(f * ai[mbs] + 1); in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
88 jmin = ai[rip[k]] + 1; /* exclude diag[k] */ in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
89 jmax = ai[rip[k] + 1]; in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
165 if (ai[mbs] != 0) { in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
166 PetscReal af = ((PetscReal)iu[mbs]) / ((PetscReal)ai[mbs]); in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
206 if (ai[mbs] != 0) { in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
207 F->info.fill_ratio_needed = ((PetscReal)iu[mbs]) / ((PetscReal)ai[mbs]); in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
[all …]
H A Dsbaijfact3.c9 const PetscInt *ai, *aj, *perm_ptr, mbs = a->mbs, *bi = b->i, *bj = b->j; in MatCholeskyFactorNumeric_SeqSBAIJ_3() local
29 ai = a->i; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
33 ai = a->inew; 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()
37 PetscCall(PetscMalloc1(ai[mbs], &a2anew)); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
38 PetscCall(PetscArraycpy(a2anew, a->a2anew, ai[mbs])); in MatCholeskyFactorNumeric_SeqSBAIJ_3()
41 jmin = ai[i]; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
42 jmax = ai[i + 1]; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
71 jmin = ai[perm_ptr[k]]; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
[all …]
/petsc/src/binding/petsc4py/test/
H A Dtest_mat_aij.py82 ai, aj, av = self._set_values()
84 self._chk_aij(self.A, ai, aj)
87 ai, aj, av = self._set_values_ijv()
89 self._chk_aij(self.A, ai, aj)
92 _, ai, _, _ = self._get_aijv()
93 d_nnz = N.diff(ai)
99 ai, aj, av = self._set_values()
101 self._chk_aij(self.A, ai, aj)
104 ai, aj, av = self._set_values_ijv()
106 self._chk_aij(self.A, ai, aj)
[all …]
/petsc/src/mat/impls/aij/seq/
H A Dsymtranspose.c16 PetscInt *ati, *atj, *atfill, *ai = a->i, *aj = a->j; in MatTransposeSymbolic_SeqAIJ() local
21 PetscCall(PetscMalloc1(ai[am], &atj)); in MatTransposeSymbolic_SeqAIJ()
25 for (i = 0; i < ai[am]; i++) ati[aj[i] + 1] += 1; in MatTransposeSymbolic_SeqAIJ()
35 anzj = ai[i + 1] - ai[i]; in MatTransposeSymbolic_SeqAIJ()
61 PetscInt *ati, *atj, *atfill, *ai = a->i, *aj = a->j; in MatTranspose_SeqAIJ() local
81 PetscCall(PetscMalloc1(ai[am], &atj)); in MatTranspose_SeqAIJ()
84 for (i = 0; i < ai[am]; i++) ati[aj[i] + 1] += 1; in MatTranspose_SeqAIJ()
87 PetscCall(PetscMalloc1(ai[am], &ata)); in MatTranspose_SeqAIJ()
103 anzj = ai[i + 1] - ai[i]; in MatTranspose_SeqAIJ()
112 anzj = ai[i + 1] - ai[i]; in MatTranspose_SeqAIJ()
[all …]
H A Dij.c31 PetscErrorCode MatToSymmetricIJ_SeqAIJ(PetscInt m, PetscInt *ai, PetscInt *aj, PetscBool lower_tria… in MatToSymmetricIJ_SeqAIJ() argument
44 nz = ai[row + 1] - ai[row]; in MatToSymmetricIJ_SeqAIJ()
45 j = aj + ai[row] + shiftin; in MatToSymmetricIJ_SeqAIJ()
72 nz = ai[row + 1] - ai[row]; in MatToSymmetricIJ_SeqAIJ()
73 j = aj + ai[row] + shiftin; in MatToSymmetricIJ_SeqAIJ()
H A Daijfact.c58 const PetscInt *r, *ic, *ai = a->i, *aj = a->j, *ajtmp; in MatLUFactorSymbolic_SeqAIJ() local
91 PetscCall(PetscFreeSpaceGet(PetscRealIntMultTruncate(f, ai[n] + 1), &free_space)); in MatLUFactorSymbolic_SeqAIJ()
97 nnz = ai[r[i] + 1] - ai[r[i]]; in MatLUFactorSymbolic_SeqAIJ()
98 ajtmp = aj + ai[r[i]]; in MatLUFactorSymbolic_SeqAIJ()
175 if (ai[n]) { in MatLUFactorSymbolic_SeqAIJ()
176 B->info.fill_ratio_needed = ((PetscReal)(bdiag[0] + 1)) / ((PetscReal)ai[n]); in MatLUFactorSymbolic_SeqAIJ()
181 if (ai[n] != 0) { in MatLUFactorSymbolic_SeqAIJ()
224 const PetscInt n = A->rmap->n, *ai = a->i, *aj = a->j, *bi = b->i, *bj = b->j, *bdiag = b->diag; in MatLUFactorNumeric_SeqAIJ() local
249 v = aa + ai[i]; in MatLUFactorNumeric_SeqAIJ()
250 nz = ai[i + 1] - ai[i]; in MatLUFactorNumeric_SeqAIJ()
[all …]
/petsc/src/mat/impls/baij/seq/
H A Daijbaij.c9 PetscInt bs = A->rmap->bs, *ai = a->i, *aj = a->j, n = A->rmap->N / bs, i, j, k; in MatConvert_SeqBAIJ_SeqAIJ() local
16 for (i = 0; i < n; i++) maxlen = PetscMax(maxlen, ai[i + 1] - ai[i]); in MatConvert_SeqBAIJ_SeqAIJ()
20 maxlen = PetscMax(maxlen, ai[i + 1] - ai[i]); in MatConvert_SeqBAIJ_SeqAIJ()
21 for (j = 0; j < bs; j++) rowlengths[i * bs + j] = bs * (ai[i + 1] - ai[i]); in MatConvert_SeqBAIJ_SeqAIJ()
38 ncols = ai[i + 1] - ai[i]; in MatConvert_SeqBAIJ_SeqAIJ()
63 const PetscInt *ai = Aa->i, *aj = Aa->j; in MatConvert_SeqAIJ_SeqBAIJ_Preallocate() local
70 for (PetscInt i = 0; i < m; i++) (*nnz)[i] = ai[i + 1] - ai[i]; in MatConvert_SeqAIJ_SeqBAIJ_Preallocate()
81 for (k = ai[i]; k < ai[i + 1]; k++) { in MatConvert_SeqAIJ_SeqBAIJ_Preallocate()
H A Dbaijsolvtrannat1.c6 const PetscInt *adiag = a->diag, *ai = a->i, *aj = a->j, *vi; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering() local
34 v = aa + ai[i]; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering()
35 vi = aj + ai[i]; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering()
36 nz = ai[i + 1] - ai[i]; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering()
55 const PetscInt *diag = a->diag, n = a->mbs, *vi, *ai = a->i, *aj = a->j; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering_inplace() local
69 nz = ai[i + 1] - diag[i] - 1; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering_inplace()
77 nz = diag[i] - ai[i]; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering_inplace()
H A Dbaijsolvnat2.c15 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *diag = a->diag; in MatSolve_SeqBAIJ_2_NaturalOrdering_inplace() local
30 v = aa + 4 * ai[i]; in MatSolve_SeqBAIJ_2_NaturalOrdering_inplace()
31 vi = aj + ai[i]; in MatSolve_SeqBAIJ_2_NaturalOrdering_inplace()
32 nz = diag[i] - ai[i]; in MatSolve_SeqBAIJ_2_NaturalOrdering_inplace()
51 nz = ai[i + 1] - diag[i] - 1; in MatSolve_SeqBAIJ_2_NaturalOrdering_inplace()
77 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *adiag = a->diag; in MatSolve_SeqBAIJ_2_NaturalOrdering() local
91 v = aa + 4 * ai[i]; in MatSolve_SeqBAIJ_2_NaturalOrdering()
92 vi = aj + ai[i]; in MatSolve_SeqBAIJ_2_NaturalOrdering()
93 nz = ai[i + 1] - ai[i]; in MatSolve_SeqBAIJ_2_NaturalOrdering()
143 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j; in MatForwardSolve_SeqBAIJ_2_NaturalOrdering() local
[all …]
H A Dbaijsolvnat1.c11 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *diag = a->diag; in MatSolve_SeqBAIJ_1_NaturalOrdering_inplace() local
26 v = aa + ai[i]; in MatSolve_SeqBAIJ_1_NaturalOrdering_inplace()
27 vi = aj + ai[i]; in MatSolve_SeqBAIJ_1_NaturalOrdering_inplace()
28 nz = diag[i] - ai[i]; in MatSolve_SeqBAIJ_1_NaturalOrdering_inplace()
43 nz = ai[i + 1] - diag[i] - 1; in MatSolve_SeqBAIJ_1_NaturalOrdering_inplace()
64 const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *vi; in MatForwardSolve_SeqBAIJ_1_NaturalOrdering() local
81 nz = ai[i + 1] - ai[i]; in MatForwardSolve_SeqBAIJ_1_NaturalOrdering()
128 const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *adiag = a->diag, *vi; in MatSolve_SeqBAIJ_1_NaturalOrdering() local
145 nz = ai[i + 1] - ai[i]; in MatSolve_SeqBAIJ_1_NaturalOrdering()
H A Dbaijfact.c13 const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *bi = b->i, *bj = b->j, bs2 = a->bs2; in MatLUFactorNumeric_SeqBAIJ_2() local
42 nz = ai[r[i] + 1] - ai[r[i]]; in MatLUFactorNumeric_SeqBAIJ_2()
43 ajtmp = aj + ai[r[i]]; in MatLUFactorNumeric_SeqBAIJ_2()
44 v = aa + bs2 * ai[r[i]]; in MatLUFactorNumeric_SeqBAIJ_2()
122 const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *bi = b->i, *bj = b->j, bs2 = a->bs2; in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering() local
150 nz = ai[i + 1] - ai[i]; in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
151 ajtmp = aj + ai[i]; in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
152 v = aa + bs2 * ai[i]; in MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering()
231 PetscInt idx, *ai = a->i, *aj = a->j, *pj; in MatILUFactorNumeric_SeqBAIJ_2_inplace() local
258 nz = ai[idx + 1] - ai[idx]; in MatILUFactorNumeric_SeqBAIJ_2_inplace()
[all …]
H A Dbaijsolvtrann.c8 const PetscInt *r, *c, *rout, *cout, *ai = a->i, *aj = a->j, *vi; in MatSolveTranspose_SeqBAIJ_N_inplace() local
38 nz = ai[i + 1] - a->diag[i] - 1; in MatSolveTranspose_SeqBAIJ_N_inplace()
47 v = aa + bs2 * ai[i]; in MatSolveTranspose_SeqBAIJ_N_inplace()
48 vi = aj + ai[i]; in MatSolveTranspose_SeqBAIJ_N_inplace()
49 nz = a->diag[i] - ai[i]; in MatSolveTranspose_SeqBAIJ_N_inplace()
74 const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *vi, *diag = a->diag; in MatSolveTranspose_SeqBAIJ_N() local
113 v = aa + bs2 * ai[i]; in MatSolveTranspose_SeqBAIJ_N()
114 vi = aj + ai[i]; in MatSolveTranspose_SeqBAIJ_N()
115 nz = ai[i + 1] - ai[i]; in MatSolveTranspose_SeqBAIJ_N()
H A Dbaijsolvtran1.c8 const PetscInt *rout, *cout, *r, *c, *adiag = a->diag, *ai = a->i, *aj = a->j, *vi; in MatSolveTranspose_SeqBAIJ_1() local
41 v = aa + ai[i]; in MatSolveTranspose_SeqBAIJ_1()
42 vi = aj + ai[i]; in MatSolveTranspose_SeqBAIJ_1()
43 nz = ai[i + 1] - ai[i]; in MatSolveTranspose_SeqBAIJ_1()
65 const PetscInt *diag = a->diag, n = a->mbs, *vi, *ai = a->i, *aj = a->j; in MatSolveTranspose_SeqBAIJ_1_inplace() local
90 nz = ai[i + 1] - diag[i] - 1; in MatSolveTranspose_SeqBAIJ_1_inplace()
98 nz = diag[i] - ai[i]; in MatSolveTranspose_SeqBAIJ_1_inplace()
H A Dbaijsolvtrannat2.c7 const PetscInt *diag = a->diag, *vi, n = a->mbs, *ai = a->i, *aj = a->j; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering_inplace() local
27 nz = ai[i + 1] - diag[i] - 1; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering_inplace()
42 nz = diag[i] - ai[i]; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering_inplace()
61 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *diag = a->diag; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering() local
96 v = aa + bs2 * ai[i]; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering()
97 vi = aj + ai[i]; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering()
98 nz = ai[i + 1] - ai[i]; in MatSolveTranspose_SeqBAIJ_2_NaturalOrdering()
H A Dbaijsolvnat3.c11 const PetscInt n = a->mbs, *ai = a->i, *aj = a->j; in MatSolve_SeqBAIJ_3_NaturalOrdering_inplace() local
28 v = aa + 9 * ai[i]; in MatSolve_SeqBAIJ_3_NaturalOrdering_inplace()
29 vi = aj + ai[i]; in MatSolve_SeqBAIJ_3_NaturalOrdering_inplace()
30 nz = diag[i] - ai[i]; in MatSolve_SeqBAIJ_3_NaturalOrdering_inplace()
53 nz = ai[i + 1] - diag[i] - 1; in MatSolve_SeqBAIJ_3_NaturalOrdering_inplace()
83 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *adiag = a->diag; in MatSolve_SeqBAIJ_3_NaturalOrdering() local
100 v = aa + bs2 * ai[i]; in MatSolve_SeqBAIJ_3_NaturalOrdering()
101 vi = aj + ai[i]; in MatSolve_SeqBAIJ_3_NaturalOrdering()
102 nz = ai[i + 1] - ai[i]; in MatSolve_SeqBAIJ_3_NaturalOrdering()
160 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j; in MatForwardSolve_SeqBAIJ_3_NaturalOrdering() local
[all …]
H A Dbaijfact2.c20 …const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *bi = b->i, *bj = b->j, *ajtmp, *bjtmp, *bdia… in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering() local
49 nz = ai[i + 1] - ai[i]; in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
50 ajtmp = aj + ai[i]; in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
51 v = aa + bs2 * ai[i]; in MatLUFactorNumeric_SeqBAIJ_15_NaturalOrdering()
120 PetscInt i, j, k, n = a->mbs, *ai = a->i, *aj = a->j, *bi = b->i, *bj = b->j; in MatLUFactorNumeric_SeqBAIJ_N() local
151 nz = ai[r[i] + 1] - ai[r[i]]; in MatLUFactorNumeric_SeqBAIJ_N()
152 ajtmp = aj + ai[r[i]]; in MatLUFactorNumeric_SeqBAIJ_N()
153 v = aa + bs2 * ai[r[i]]; in MatLUFactorNumeric_SeqBAIJ_N()
255 const PetscInt n = a->mbs, *ai = a->i, *aj, *adiag, bs2 = a->bs2; in MatILUFactorSymbolic_SeqBAIJ_ilu0() local
264 PetscCall(PetscShmgetAllocateArray(bs2 * ai[n], sizeof(PetscScalar), (void **)&b->a)); in MatILUFactorSymbolic_SeqBAIJ_ilu0()
[all …]
H A Dbaijsolvnat4.c12 const PetscInt *ai = a->i, *aj = a->j; in MatSolve_SeqBAIJ_4_NaturalOrdering_inplace() local
25 fortransolvebaij4_(&n, x, ai, aj, diag, aa, b, w); in MatSolve_SeqBAIJ_4_NaturalOrdering_inplace()
28 fortransolvebaij4unroll_(&n, x, ai, aj, diag, aa, b); in MatSolve_SeqBAIJ_4_NaturalOrdering_inplace()
43 v = aa + 16 * ai[i]; in MatSolve_SeqBAIJ_4_NaturalOrdering_inplace()
44 vi = aj + ai[i]; in MatSolve_SeqBAIJ_4_NaturalOrdering_inplace()
45 nz = diag[i] - ai[i]; in MatSolve_SeqBAIJ_4_NaturalOrdering_inplace()
74 nz = ai[i + 1] - diag[i] - 1; in MatSolve_SeqBAIJ_4_NaturalOrdering_inplace()
110 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *adiag = a->diag; in MatSolve_SeqBAIJ_4_NaturalOrdering() local
128 v = aa + bs2 * ai[i]; in MatSolve_SeqBAIJ_4_NaturalOrdering()
129 vi = aj + ai[i]; in MatSolve_SeqBAIJ_4_NaturalOrdering()
[all …]
H A Dbaijsolv.c9 const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *vi; in MatSolve_SeqBAIJ_N_inplace() local
30 v = aa + bs2 * ai[i]; in MatSolve_SeqBAIJ_N_inplace()
31 vi = aj + ai[i]; in MatSolve_SeqBAIJ_N_inplace()
32 nz = a->diag[i] - ai[i]; in MatSolve_SeqBAIJ_N_inplace()
45 nz = ai[i + 1] - a->diag[i] - 1; in MatSolve_SeqBAIJ_N_inplace()
67 const PetscInt *r, *c, *ai = a->i, *aj = a->j; in MatSolve_SeqBAIJ_7_inplace() local
95 v = aa + 49 * ai[i]; in MatSolve_SeqBAIJ_7_inplace()
96 vi = aj + ai[i]; in MatSolve_SeqBAIJ_7_inplace()
97 nz = diag[i] - ai[i]; in MatSolve_SeqBAIJ_7_inplace()
137 nz = ai[i + 1] - diag[i] - 1; in MatSolve_SeqBAIJ_7_inplace()
[all …]
/petsc/src/mat/impls/aij/seq/aijmkl/
H A Daijmkl.c118 PetscInt *aj, *ai; in MatSeqAIJMKL_create_mkl_handle()
144 ai = a->i; /* ai[k] is the position in aa and aj where row k starts. */ in MatSeqAIJMKL_create_mkl_handle()
148 …->csrA, SPARSE_INDEX_BASE_ZERO, (MKL_INT)m, (MKL_INT)n, (MKL_INT *)ai, (MKL_INT *)(ai + 1), (MKL_I… in MatSeqAIJMKL_create_mkl_handle()
167 PetscInt *aj, *ai, *unused; in MatSeqAIJMKL_setup_structure_from_mkl_handle() local
174 …_x_export_csr, csrA, &indexing, (MKL_INT *)&m, (MKL_INT *)&n, (MKL_INT **)&ai, (MKL_INT **)&unused… in MatSeqAIJMKL_setup_structure_from_mkl_handle()
177 aj = ai = NULL; in MatSeqAIJMKL_setup_structure_from_mkl_handle()
188 PetscCall(MatSeqAIJSetPreallocationCSR(A, ai, aj, NULL)); in MatSeqAIJMKL_setup_structure_from_mkl_handle()
227 PetscInt *ai, *aj, *unused; in MatSeqAIJMKL_update_from_mkl_handle() local
237 …ijmkl->csrA, &indexing, (MKL_INT *)&nrows, (MKL_INT *)&ncols, (MKL_INT **)&ai, (MKL_INT **)&unused… in MatSeqAIJMKL_update_from_mkl_handle()
242 nz = ai[i + 1] - ai[i]; in MatSeqAIJMKL_update_from_mkl_handle()
[all …]
/petsc/src/mat/utils/
H A Dcompressedrow.c30 …ssedRow(Mat A, PetscInt nrows, Mat_CompressedRow *compressedrow, PetscInt *ai, PetscInt mbs, Petsc… in MatCheckCompressedRow() argument
57 nz = ai[i + 1] - ai[i]; in MatCheckCompressedRow()
59 cpi[row + 1] = ai[i + 1]; /* compressed row pointer */ in MatCheckCompressedRow()

12345