Home
last modified time | relevance | path

Searched refs:adiag (Results 1 – 25 of 37) sorted by relevance

12

/petsc/src/mat/impls/aij/seq/ftn-kernels/
H A Dfsolve.F9013 PetscInt, intent(in) :: n, ai(0:*), aj(0:*), adiag(0:*)
23 jend = adiag(i) - 1
30 jstart = adiag(i) + 1
32 x(i) = x(i) - sum(aa(jstart:jend)*x(aj(jstart:jend)))*aa(adiag(i))
/petsc/src/mat/impls/baij/seq/
H A Dbaijsolvnat1.c97 const PetscInt n = a->mbs, *aj = a->j, *adiag = a->diag, *vi; in MatBackwardSolve_SeqBAIJ_1_NaturalOrdering() local
111 v = aa + adiag[i + 1] + 1; in MatBackwardSolve_SeqBAIJ_1_NaturalOrdering()
112 vi = aj + adiag[i + 1] + 1; in MatBackwardSolve_SeqBAIJ_1_NaturalOrdering()
113 nz = adiag[i] - adiag[i + 1] - 1; in MatBackwardSolve_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
155 v = aa + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_1_NaturalOrdering()
156 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_1_NaturalOrdering()
157 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_1_NaturalOrdering()
H A Dbaijsolvnat2.c77 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *adiag = a->diag; in MatSolve_SeqBAIJ_2_NaturalOrdering() local
113 v = aa + 4 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_2_NaturalOrdering()
114 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_2_NaturalOrdering()
115 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_2_NaturalOrdering()
186 const PetscInt n = a->mbs, *vi, *aj = a->j, *adiag = a->diag; in MatBackwardSolve_SeqBAIJ_2_NaturalOrdering() local
198 v = aa + 4 * (adiag[i + 1] + 1); in MatBackwardSolve_SeqBAIJ_2_NaturalOrdering()
199 vi = aj + adiag[i + 1] + 1; in MatBackwardSolve_SeqBAIJ_2_NaturalOrdering()
200 nz = adiag[i] - adiag[i + 1] - 1; in MatBackwardSolve_SeqBAIJ_2_NaturalOrdering()
H A Dbaijsolvtrannat1.c6 const PetscInt *adiag = a->diag, *ai = a->i, *aj = a->j, *vi; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering() local
23 v = aa + adiag[i + 1] + 1; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering()
24 vi = aj + adiag[i + 1] + 1; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering()
25 nz = adiag[i] - adiag[i + 1] - 1; in MatSolveTranspose_SeqBAIJ_1_NaturalOrdering()
H A Dbaijsolvnat14.c10 …const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *adiag = a->diag, *vi, bs = A->rmap->bs, bs… in MatSolve_SeqBAIJ_14_NaturalOrdering() local
65 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_14_NaturalOrdering()
66 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_14_NaturalOrdering()
67 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_14_NaturalOrdering()
136 …const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *adiag = a->diag, *vi, bs = A->rmap->bs, bs… in MatSolve_SeqBAIJ_13_NaturalOrdering() local
189 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_13_NaturalOrdering()
190 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_13_NaturalOrdering()
191 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_13_NaturalOrdering()
257 …const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *adiag = a->diag, *vi, bs = A->rmap->bs, bs… in MatSolve_SeqBAIJ_12_NaturalOrdering() local
308 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_12_NaturalOrdering()
[all …]
H A Dbaijsolvnat3.c83 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *adiag = a->diag; in MatSolve_SeqBAIJ_3_NaturalOrdering() local
126 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_3_NaturalOrdering()
127 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_3_NaturalOrdering()
128 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_3_NaturalOrdering()
210 const PetscInt n = a->mbs, *vi, *aj = a->j, *adiag = a->diag; in MatBackwardSolve_SeqBAIJ_3_NaturalOrdering() local
224 v = aa + bs2 * (adiag[i + 1] + 1); in MatBackwardSolve_SeqBAIJ_3_NaturalOrdering()
225 vi = aj + adiag[i + 1] + 1; in MatBackwardSolve_SeqBAIJ_3_NaturalOrdering()
226 nz = adiag[i] - adiag[i + 1] - 1; in MatBackwardSolve_SeqBAIJ_3_NaturalOrdering()
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
30 v = aa + adiag[i + 1] + 1; in MatSolveTranspose_SeqBAIJ_1()
31 vi = aj + adiag[i + 1] + 1; in MatSolveTranspose_SeqBAIJ_1()
32 nz = adiag[i] - adiag[i + 1] - 1; in MatSolveTranspose_SeqBAIJ_1()
H A Dbaijsolvnat11.c10 …const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *adiag = a->diag, *vi, bs = A->rmap->bs, bs… in MatSolve_SeqBAIJ_11_NaturalOrdering() local
59 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_11_NaturalOrdering()
60 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_11_NaturalOrdering()
61 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_11_NaturalOrdering()
H A Dbaijfact9.c880 const PetscInt *ai = a->i, *aj = a->j, *adiag = a->diag, *vi; in MatSolve_SeqBAIJ_9_NaturalOrdering() local
990 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_9_NaturalOrdering()
991 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_9_NaturalOrdering()
992 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_9_NaturalOrdering()
1092 a0 = _mm256_loadu_pd(&(aa + bs2 * adiag[i])[0]); in MatSolve_SeqBAIJ_9_NaturalOrdering()
1094 a1 = _mm256_loadu_pd(&(aa + bs2 * adiag[i])[4]); in MatSolve_SeqBAIJ_9_NaturalOrdering()
1096 a2 = _mm256_loadu_pd(&(aa + bs2 * adiag[i])[8]); in MatSolve_SeqBAIJ_9_NaturalOrdering()
1101 a3 = _mm256_loadu_pd(&(aa + bs2 * adiag[i])[9]); in MatSolve_SeqBAIJ_9_NaturalOrdering()
1103 a4 = _mm256_loadu_pd(&(aa + bs2 * adiag[i])[13]); in MatSolve_SeqBAIJ_9_NaturalOrdering()
1105 a5 = _mm256_loadu_pd(&(aa + bs2 * adiag[i])[17]); in MatSolve_SeqBAIJ_9_NaturalOrdering()
[all …]
H A Dbaijsolv.c187 const PetscInt *r, *c, *ai = a->i, *aj = a->j, *adiag = a->diag; in MatSolve_SeqBAIJ_7() local
255 v = aa + 49 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_7()
256 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_7()
257 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_7()
417 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *adiag = a->diag; in MatSolve_SeqBAIJ_6() local
478 v = aa + 36 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_6()
479 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_6()
480 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_6()
627 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *adiag = a->diag; in MatSolve_SeqBAIJ_5() local
683 v = aa + 25 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_5()
[all …]
H A Dbaijsolvnat4.c110 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *adiag = a->diag; in MatSolve_SeqBAIJ_4_NaturalOrdering() local
158 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_4_NaturalOrdering()
159 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_4_NaturalOrdering()
160 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_4_NaturalOrdering()
H A Dbaijsolvnat5.c95 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *adiag = a->diag; in MatSolve_SeqBAIJ_5_NaturalOrdering() local
144 v = aa + 25 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_5_NaturalOrdering()
145 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_5_NaturalOrdering()
146 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_5_NaturalOrdering()
H A Dbaijsolvnat6.c104 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *adiag = a->diag; in MatSolve_SeqBAIJ_6_NaturalOrdering() local
161 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_6_NaturalOrdering()
162 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_6_NaturalOrdering()
163 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_6_NaturalOrdering()
H A Dbaijsolvnat15.c9 …const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *adiag = a->diag, *vi, bs = A->rmap->bs, bs… in MatSolve_SeqBAIJ_15_NaturalOrdering_ver2() local
113 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_15_NaturalOrdering_ver2()
114 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_15_NaturalOrdering_ver2()
115 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_15_NaturalOrdering_ver2()
199 …const PetscInt n = a->mbs, *ai = a->i, *aj = a->j, *adiag = a->diag, *vi, bs = A->rmap->bs, bs… in MatSolve_SeqBAIJ_15_NaturalOrdering_ver1() local
256 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_15_NaturalOrdering_ver1()
257 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_15_NaturalOrdering_ver1()
258 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_15_NaturalOrdering_ver1()
H A Dbaijsolvnat7.c113 const PetscInt n = a->mbs, *vi, *ai = a->i, *aj = a->j, *adiag = a->diag; in MatSolve_SeqBAIJ_7_NaturalOrdering() local
175 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_7_NaturalOrdering()
176 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_7_NaturalOrdering()
177 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_7_NaturalOrdering()
H A Dbaijfact.c1031 const PetscInt *adiag; in MatICCFactorSymbolic_SeqBAIJ() local
1034 PetscCall(MatGetDiagonalMarkers_SeqBAIJ(A, &adiag, &diagDense)); in MatICCFactorSymbolic_SeqBAIJ()
1055 …for (i = 0; i < am; i++) ui[i] = ai[i + 1] - adiag[i]; /* ui: rowlengths - changes when !perm_iden… in MatICCFactorSymbolic_SeqBAIJ()
1062 aj = a->j + adiag[i]; in MatICCFactorSymbolic_SeqBAIJ()
1426 const PetscInt *ai = a->i, *aj = a->j, *adiag = a->diag, *vi; in MatSolve_SeqBAIJ_N_NaturalOrdering() local
1457 v = aa + bs2 * (adiag[i + 1] + 1); in MatSolve_SeqBAIJ_N_NaturalOrdering()
1458 vi = aj + adiag[i + 1] + 1; in MatSolve_SeqBAIJ_N_NaturalOrdering()
1459 nz = adiag[i] - adiag[i + 1] - 1; in MatSolve_SeqBAIJ_N_NaturalOrdering()
1465 PetscKernel_w_gets_A_times_v(bs, ls, aa + bs2 * adiag[i], t + i * bs); /* *inv(diagonal[i]) */ in MatSolve_SeqBAIJ_N_NaturalOrdering()
1479 const PetscInt *r, *c, *rout, *cout, *ai = a->i, *aj = a->j, *adiag = a->diag, *vi; in MatSolve_SeqBAIJ_N() local
[all …]
H A Dbaijfact2.c255 const PetscInt n = a->mbs, *ai = a->i, *aj, *adiag, bs2 = a->bs2; in MatILUFactorSymbolic_SeqBAIJ_ilu0() local
259 PetscCall(MatGetDiagonalMarkers_SeqBAIJ(A, &adiag, NULL)); in MatILUFactorSymbolic_SeqBAIJ_ilu0()
283 nz = adiag[i] - ai[i]; in MatILUFactorSymbolic_SeqBAIJ_ilu0()
296 nz = ai[i + 1] - adiag[i] - 1; in MatILUFactorSymbolic_SeqBAIJ_ilu0()
298 aj = a->j + adiag[i] + 1; in MatILUFactorSymbolic_SeqBAIJ_ilu0()
/petsc/src/mat/impls/baij/seq/ftn-kernels/
H A Dfsolvebaij.F9017 PetscInt, intent(in) :: ai(0:*), aj(0:*), adiag(0:*)
28 PETSC_AssertAlignx(16, adiag(1))
37 jend = adiag(i) - 1
57 jstart = adiag(i) + 1
69 ax = 16*adiag(i)
87 PetscInt, intent(in) :: ai(0:*), aj(0:*), adiag(0:*)
99 PETSC_AssertAlignx(16, adiag(1))
111 jend = adiag(i) - 1
140 jstart = adiag(i) + 1
162 ax = 16*adiag(i)
/petsc/src/mat/impls/aij/seq/
H A Daijfact.c751 const PetscInt *adiag; in MatSolve_SeqAIJ_inplace() local
756 PetscCall(MatGetDiagonalMarkers_SeqAIJ(A, &adiag, NULL)); in MatSolve_SeqAIJ_inplace()
773 nz = adiag[i] - ai[i]; in MatSolve_SeqAIJ_inplace()
781 v = aa + adiag[i] + 1; in MatSolve_SeqAIJ_inplace()
782 vi = aj + adiag[i] + 1; in MatSolve_SeqAIJ_inplace()
783 nz = ai[i + 1] - adiag[i] - 1; in MatSolve_SeqAIJ_inplace()
786 x[*c--] = tmp[i] = sum * aa[adiag[i]]; in MatSolve_SeqAIJ_inplace()
807 const PetscInt *adiag; in MatMatSolve_SeqAIJ_inplace() local
817 PetscCall(MatGetDiagonalMarkers_SeqAIJ(A, &adiag, NULL)); in MatMatSolve_SeqAIJ_inplace()
834 nz = adiag[i] - ai[i]; in MatMatSolve_SeqAIJ_inplace()
[all …]
/petsc/src/mat/impls/sbaij/seq/
H A Daijsbaij.c132 const PetscInt *adiag; in MatConvert_SeqAIJ_SeqSBAIJ_Preallocate() local
134 PetscCall(MatGetDiagonalMarkers_SeqAIJ(A, &adiag, NULL)); in MatConvert_SeqAIJ_SeqSBAIJ_Preallocate()
137 if (adiag[i] == ai[i + 1]) { in MatConvert_SeqAIJ_SeqSBAIJ_Preallocate()
140 } else (*nnz)[i] = ai[i + 1] - adiag[i]; in MatConvert_SeqAIJ_SeqSBAIJ_Preallocate()
171 const PetscInt *adiag; in MatConvert_SeqAIJ_SeqSBAIJ() local
180 PetscCall(MatGetDiagonalMarkers_SeqAIJ(A, &adiag, NULL)); in MatConvert_SeqAIJ_SeqSBAIJ()
184 if (adiag[i] == ai[i + 1]) { /* missing diagonal */ in MatConvert_SeqAIJ_SeqSBAIJ()
188 rowlengths[i] = (ai[i + 1] - adiag[i]); in MatConvert_SeqAIJ_SeqSBAIJ()
207 aj = a->j + adiag[i]; in MatConvert_SeqAIJ_SeqSBAIJ()
208 av = a->a + adiag[i]; in MatConvert_SeqAIJ_SeqSBAIJ()
[all …]
/petsc/src/mat/impls/aij/seq/cholmod/
H A Daijcholmod.c9 const PetscInt *ai = aij->i, *aj = aij->j, *adiag; in MatWrapCholmod_seqaij() local
14 PetscCall(MatGetDiagonalMarkers_SeqAIJ(A, &adiag, NULL)); in MatWrapCholmod_seqaij()
15 for (i = 0, nz = 0; i < m; i++) nz += ai[i + 1] - adiag[i]; in MatWrapCholmod_seqaij()
24 for (j = adiag[i]; j < ai[i + 1]; j++, k++) { in MatWrapCholmod_seqaij()
/petsc/src/mat/impls/sbaij/mpi/
H A Dmpiaijsbaij.c63 const PetscInt *adiag; in MatConvert_MPIBAIJ_MPISBAIJ() local
71 PetscCall(MatGetDiagonalMarkers_SeqBAIJ(mpimat->A, &adiag, NULL)); in MatConvert_MPIBAIJ_MPISBAIJ()
73 d_nnz[i] = Aa->i[i + 1] - adiag[i]; in MatConvert_MPIBAIJ_MPISBAIJ()
/petsc/src/mat/impls/aij/seq/bas/
H A Dbasfactor.c11 const PetscInt *rip, *riip, *adiag; in MatICCFactorSymbolic_SeqAIJ_Bas() local
20 PetscCall(MatGetDiagonalMarkers_SeqAIJ(A, &adiag, &diagDense)); in MatICCFactorSymbolic_SeqAIJ_Bas()
30 for (i = 0; i < am; i++) ui[i + 1] = ui[i] + ai[i + 1] - adiag[i]; in MatICCFactorSymbolic_SeqAIJ_Bas()
34 aj = a->j + adiag[i]; in MatICCFactorSymbolic_SeqAIJ_Bas()
/petsc/src/mat/impls/aij/seq/mkl_pardiso/
H A Dmkl_pardiso.c182 const PetscInt *adiag; in MatMKLPardiso_Convert_seqaij() local
184 PetscCall(MatGetDiagonalMarkers_SeqAIJ(A, &adiag, NULL)); in MatMKLPardiso_Convert_seqaij()
186 for (i = 0; i < m; i++) nz += aa->i[i + 1] - adiag[i]; in MatMKLPardiso_Convert_seqaij()
194 PetscInt *aj = aa->j + adiag[i]; in MatMKLPardiso_Convert_seqaij()
195 PetscScalar *av = aav + adiag[i]; in MatMKLPardiso_Convert_seqaij()
196 PetscInt rl = aa->i[i + 1] - adiag[i], j; in MatMKLPardiso_Convert_seqaij()
216 const PetscInt *adiag; in MatMKLPardiso_Convert_seqaij() local
218 PetscCall(MatGetDiagonalMarkers_SeqAIJ(A, &adiag, NULL)); in MatMKLPardiso_Convert_seqaij()
221 PetscScalar *av = aav + adiag[i]; in MatMKLPardiso_Convert_seqaij()
222 PetscInt rl = aa->i[i + 1] - adiag[i], j; in MatMKLPardiso_Convert_seqaij()
/petsc/src/mat/impls/blockmat/seq/
H A Dblockmat.c40 const PetscInt *adiag; in MatSOR_BlockMat_Symmetric() local
42 PetscCall(MatGetDiagonalMarkers_BlockMat(A, &adiag, NULL)); in MatSOR_BlockMat_Symmetric()
47 PetscCall(MatCholeskyFactorSymbolic(a->diags[i], a->a[adiag[i]], row, &info)); in MatSOR_BlockMat_Symmetric()
48 PetscCall(MatCholeskyFactorNumeric(a->diags[i], a->a[adiag[i]], &info)); in MatSOR_BlockMat_Symmetric()
141 const PetscInt *adiag; in MatSOR_BlockMat() local
143 PetscCall(MatGetDiagonalMarkers_BlockMat(A, &adiag, NULL)); in MatSOR_BlockMat()
147 PetscCall(MatGetOrdering(a->a[adiag[i]], MATORDERINGND, &row, &col)); in MatSOR_BlockMat()
148 PetscCall(MatLUFactorSymbolic(a->diags[i], a->a[adiag[i]], row, col, &info)); in MatSOR_BlockMat()
149 PetscCall(MatLUFactorNumeric(a->diags[i], a->a[adiag[i]], &info)); in MatSOR_BlockMat()

12