| /petsc/src/mat/impls/aij/seq/ |
| H A D | matmatmult.c | 150 PetscInt i, j, anzi, brow, bnzj, cnzi, *bj, *aj, *lnk, ndouble = 0, Crmax; in MatMatMultSymbolic_SeqAIJ_SeqAIJ_LLCondensed() local
180 brow = aj[j]; in MatMatMultSymbolic_SeqAIJ_SeqAIJ_LLCondensed()
181 bnzj = bi[brow + 1] - bi[brow]; in MatMatMultSymbolic_SeqAIJ_SeqAIJ_LLCondensed()
182 bj = b->j + bi[brow]; in MatMatMultSymbolic_SeqAIJ_SeqAIJ_LLCondensed()
254 PetscInt i, j, k, anzi, bnzi, cnzi, brow; in MatMatMultNumeric_SeqAIJ_SeqAIJ_Sorted() local
287 brow = aj[j]; in MatMatMultNumeric_SeqAIJ_SeqAIJ_Sorted()
288 bnzi = bi[brow + 1] - bi[brow]; in MatMatMultNumeric_SeqAIJ_SeqAIJ_Sorted()
289 bjj = PetscSafePointerPlusOffset(bj, bi[brow]); in MatMatMultNumeric_SeqAIJ_SeqAIJ_Sorted()
290 baj = PetscSafePointerPlusOffset(ba, bi[brow]); in MatMatMultNumeric_SeqAIJ_SeqAIJ_Sorted()
327 PetscInt i, j, k, anzi, bnzi, cnzi, brow; in MatMatMultNumeric_SeqAIJ_SeqAIJ_Scalable() local
[all …]
|
| /petsc/src/mat/impls/sbaij/mpi/ |
| H A D | sbaijov.c | 19 PetscInt *ai, brow, nz, nis, l, nmax, nstages, max_no, pos; in MatIncreaseOverlap_MPISBAIJ() local
101 for (brow = 0; brow < Mbs; brow++) { in MatIncreaseOverlap_MPISBAIJ()
102 nz = ai[brow + 1] - ai[brow]; in MatIncreaseOverlap_MPISBAIJ()
104 if (!PetscBTLookupSet(table, brow)) nidx[isz++] = brow; in MatIncreaseOverlap_MPISBAIJ()
|
| H A D | mpisbaij.c | 203 brow = row / bs; \
204 rp = aj + ai[brow]; \
205 ap = aa + bs2 * ai[brow]; \
206 rmax = aimax[brow]; \
207 nrow = ailen[brow]; \
229 …MatSeqXAIJReallocateAIJ(A, a->mbs, bs2, nrow, brow, bcol, rmax, aa, ai, aj, rp, ap, aimax, a->none…
238 ailen[brow] = nrow; \
243 brow = row / bs; \
244 rp = bj + bi[brow]; \
245 ap = ba + bs2 * bi[brow]; \
[all …]
|
| /petsc/src/mat/impls/sbaij/seq/ |
| H A D | sbaij.c | 515 PetscInt brow, bcol, ridx, cidx, bs = A->rmap->bs, bs2 = a->bs2; in MatGetValues_SeqSBAIJ() local
521 brow = row / bs; in MatGetValues_SeqSBAIJ()
527 rp = aj + ai[brow]; in MatGetValues_SeqSBAIJ()
528 ap = aa + bs2 * ai[brow]; in MatGetValues_SeqSBAIJ()
529 nrow = ailen[brow]; in MatGetValues_SeqSBAIJ()
742 PetscInt *aj = a->j, nonew = a->nonew, bs = A->rmap->bs, brow, bcol; in MatSetValues_SeqSBAIJ() local
749 brow = row / bs; /* block row number */ in MatSetValues_SeqSBAIJ()
752 rp = aj + ai[brow]; /*ptr to beginning of column value of the row block*/ in MatSetValues_SeqSBAIJ()
753 ap = aa + bs2 * ai[brow]; /*ptr to beginning of element value of the row block*/ in MatSetValues_SeqSBAIJ()
754 rmax = imax[brow]; /* maximum space allocated for this row */ in MatSetValues_SeqSBAIJ()
[all …]
|
| H A D | sbaij2.c | 11 PetscInt brow, i, j, k, l, mbs, n, *nidx, isz, bcol, bcol_max, start, end, *ai, *aj, bs; in MatIncreaseOverlap_SeqSBAIJ() local
36 brow = idx[j] / bs; /* convert the indices into block indices */ in MatIncreaseOverlap_SeqSBAIJ()
37 … PetscCheck(brow < mbs, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "index greater than mat-dim"); in MatIncreaseOverlap_SeqSBAIJ()
38 if (!PetscBTLookupSet(table_out, brow)) { in MatIncreaseOverlap_SeqSBAIJ()
39 nidx[isz++] = brow; in MatIncreaseOverlap_SeqSBAIJ()
40 if (bcol_max < brow) bcol_max = brow; in MatIncreaseOverlap_SeqSBAIJ()
53 for (brow = 0; brow < mbs; brow++) { in MatIncreaseOverlap_SeqSBAIJ()
54 start = ai[brow]; in MatIncreaseOverlap_SeqSBAIJ()
55 end = ai[brow + 1]; in MatIncreaseOverlap_SeqSBAIJ()
56 …if (PetscBTLookup(table_in, brow)) { /* brow is on nidx - row search: collect all bcol in this bro… in MatIncreaseOverlap_SeqSBAIJ()
[all …]
|
| /petsc/src/mat/impls/blockmat/seq/ |
| H A D | blockmat.c | 219 PetscInt *aj = a->j, nonew = a->nonew, bs = A->rmap->bs, brow, bcol; in MatSetValues_BlockMat() local
228 brow = row / bs; in MatSetValues_BlockMat()
231 rp = aj + ai[brow]; in MatSetValues_BlockMat()
232 ap = aa + ai[brow]; in MatSetValues_BlockMat()
233 nrow = ailen[brow]; in MatSetValues_BlockMat()
241 if (A->symmetric == PETSC_BOOL3_TRUE && brow > bcol) continue; in MatSetValues_BlockMat()
262 …MatSeqXAIJReallocateAIJ(A, a->mbs, 1, nrow, brow, bcol, rmax, aa, ai, aj, rp, ap, imax, nonew, Mat… in MatSetValues_BlockMat()
279 ailen[brow] = nrow; in MatSetValues_BlockMat()
|
| /petsc/src/mat/impls/baij/seq/ |
| H A D | baij.c | 1362 PetscInt *aj = a->j, brow, bcol; in matsetvalues4_() local
1369 brow = row / 4; in matsetvalues4_()
1370 rp = aj + ai[brow]; in matsetvalues4_()
1371 ap = aa + 16 * ai[brow]; in matsetvalues4_()
1372 nrow = ailen[brow]; in matsetvalues4_()
1408 ailen[brow] = nrow; in matsetvalues4_()
2059 PetscInt brow, bcol, ridx, cidx, bs = A->rmap->bs, bs2 = a->bs2; in MatGetValues_SeqBAIJ() local
2065 brow = row / bs; in MatGetValues_SeqBAIJ()
2071 rp = PetscSafePointerPlusOffset(aj, ai[brow]); in MatGetValues_SeqBAIJ()
2072 ap = PetscSafePointerPlusOffset(aa, bs2 * ai[brow]); in MatGetValues_SeqBAIJ()
[all …]
|
| H A D | baij2.c | 3738 PetscInt brow = usecprow ? ridx[i] : i; in MatTransposeMatMult_SeqBAIJ_1_Private() local
3744 for (j = 0, bi = b + 1 * brow; j < n; j++) { in MatTransposeMatMult_SeqBAIJ_1_Private()
3776 PetscInt brow = usecprow ? ridx[i] : i; in MatTransposeMatMult_SeqBAIJ_2_Private() local
3782 for (j = 0, bi = b + 2 * brow; j < n; j++) { in MatTransposeMatMult_SeqBAIJ_2_Private()
3819 PetscInt brow = usecprow ? ridx[i] : i; in MatTransposeMatMult_SeqBAIJ_3_Private() local
3825 for (j = 0, bi = b + 3 * brow; j < n; j++) { in MatTransposeMatMult_SeqBAIJ_3_Private()
3864 PetscInt brow = usecprow ? ridx[i] : i; in MatTransposeMatMult_SeqBAIJ_4_Private() local
3870 for (j = 0, bi = b + 4 * brow; j < n; j++) { in MatTransposeMatMult_SeqBAIJ_4_Private()
3911 PetscInt brow = usecprow ? ridx[i] : i; in MatTransposeMatMult_SeqBAIJ_5_Private() local
3917 for (j = 0, bi = b + 5 * brow; j < n; j++) { in MatTransposeMatMult_SeqBAIJ_5_Private()
|
| /petsc/src/ts/impls/implicit/glle/ |
| H A D | glle.c | 378 PetscScalar brow[32], vrow[32]; in TSGLLECompleteStep_Rescale() local
387 for (j = 0; j < s; j++) brow[j] = h * sc->b[i * s + j]; in TSGLLECompleteStep_Rescale()
388 PetscCall(VecMAXPY(X[i], s, brow, Ydot)); in TSGLLECompleteStep_Rescale()
397 PetscScalar brow[32], vrow[32]; in TSGLLECompleteStep_RescaleAndModify() local
410 …brow[j] = h * (PetscPowRealInt(ratio, i) * sc->b[i * s + j] + (PetscPowRealInt(ratio, i) - PetscPo… in TSGLLECompleteStep_RescaleAndModify()
412 PetscCall(VecMAXPY(X[i], s, brow, Ydot)); in TSGLLECompleteStep_RescaleAndModify()
421 for (j = 0; j < s; j++) brow[j] = h * PetscPowRealInt(ratio, p + 1) * sc->phi[0 * s + j]; in TSGLLECompleteStep_RescaleAndModify()
422 PetscCall(VecMAXPY(X[r], s, brow, Ydot)); in TSGLLECompleteStep_RescaleAndModify()
|
| /petsc/src/mat/impls/baij/mpi/ |
| H A D | mpibaij.c | 162 brow = row / bs; \
163 rp = PetscSafePointerPlusOffset(aj, ai[brow]); \
164 ap = PetscSafePointerPlusOffset(aa, bs2 * ai[brow]); \
165 rmax = aimax[brow]; \
166 nrow = ailen[brow]; \
188 …MatSeqXAIJReallocateAIJ(A, a->mbs, bs2, nrow, brow, bcol, rmax, aa, ai, aj, rp, ap, aimax, a->none…
197 ailen[brow] = nrow; \
202 brow = row / bs; \
203 rp = PetscSafePointerPlusOffset(bj, bi[brow]); \
204 ap = PetscSafePointerPlusOffset(ba, bs2 * bi[brow]); \
[all …]
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.c | 7781 for (PetscInt brow = 0, nc1, nc2, ok = 1; brow < nloc * bs; brow += bs) { // block rows in MatCreateGraph_Simple_AIJ() local
7782 PetscCall(MatGetRow(c, brow, &nc2, &cols2, NULL)); in MatCreateGraph_Simple_AIJ()
7783 nnz[brow / bs] = nc2 / bs; in MatCreateGraph_Simple_AIJ()
7785 if (nnz[brow / bs] > nmax) nmax = nnz[brow / bs]; in MatCreateGraph_Simple_AIJ()
7787 PetscCall(MatGetRow(c, brow + ii, &nc1, &cols1, NULL)); in MatCreateGraph_Simple_AIJ()
7795 PetscCall(MatRestoreRow(c, brow + ii, &nc1, &cols1, NULL)); in MatCreateGraph_Simple_AIJ()
7797 PetscCall(MatRestoreRow(c, brow, &nc2, &cols2, NULL)); in MatCreateGraph_Simple_AIJ()
7810 for (PetscInt brow = 0, n, grow; brow < nloc * bs; brow += bs) { // block rows in MatCreateGraph_Simple_AIJ() local
7814 n = ai[brow + 1] - ai[brow]; in MatCreateGraph_Simple_AIJ()
7815 aj = aseq->j + ai[brow]; in MatCreateGraph_Simple_AIJ()
[all …]
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 2690 const PetscInt brow = a->j[j]; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE() local
2691 flops += 2. * (b->i[brow + 1] - b->i[brow]); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 3198 const PetscInt brow = a->j[j]; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE() local
3199 flops += 2. * (b->i[brow + 1] - b->i[brow]); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
|