| /petsc/src/mat/tests/ |
| H A D | ex225.c | 104 PetscInt rstart, rend, nzA, nzB; in main() local
112 PetscCall(MatGetRow(A, i, &nzA, &idxA, &vA)); in main()
114 …PetscCheck(nzA == nzB, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Error MatGetRow %" PetscInt_FMT, nzA - nz… in main()
116 PetscCall(PetscArraycmp(idxA, idxB, nzA, &flg)); in main()
118 PetscCall(PetscArraycmp(vA, vB, nzA, &flg)); in main()
120 PetscCall(MatRestoreRow(A, i, &nzA, &idxA, &vA)); in main()
|
| /petsc/src/mat/impls/baij/mpi/ |
| H A D | baijov.c | 644 …PetscInt nzA, nzB, *a_i = a->i, *b_i = b->i, *a_j = a->j, *b_j = b->j, ctmp, imark, *cworkA… local
954 nzA = a_i[row + 1] - a_i[row];
956 ncols = nzA + nzB;
967 for (l = 0; l < nzA; l++) cols[imark + l] = cstart + cworkA[l];
968 for (l = imark; l < nzB; l++) cols[nzA + l] = bmap[cworkB[l]];
1025 nzA = a_i[row + 1] - a_i[row];
1032 for (PetscInt k = 0; k < nzA; k++) {
1042 for (PetscInt k = 0; k < nzA; k++) {
1050 lens_i[j] = nzA + nzB;
1259 nzA = a_i[row + 1] - a_i[row];
[all …]
|
| H A D | mpibaij.c | 1304 PetscInt nztot, nzA, nzB, lrow, brstart = matin->rmap->rstart, brend = matin->rmap->rend; in MatGetRow_MPIBAIJ() local
1338 PetscCall((*mat->A->ops->getrow)(mat->A, lrow, &nzA, pcA, pvA)); in MatGetRow_MPIBAIJ()
1340 nztot = nzA + nzB; in MatGetRow_MPIBAIJ()
1354 for (i = 0; i < nzA; i++) v_p[imark + i] = vworkA[i]; in MatGetRow_MPIBAIJ()
1355 for (i = imark; i < nzB; i++) v_p[nzA + i] = vworkB[i]; in MatGetRow_MPIBAIJ()
1368 for (i = 0; i < nzA; i++) idx_p[imark + i] = cstart * bs + cworkA[i]; in MatGetRow_MPIBAIJ()
1369 for (i = imark; i < nzB; i++) idx_p[nzA + i] = cmap[cworkB[i] / bs] * bs + cworkB[i] % bs; in MatGetRow_MPIBAIJ()
1377 PetscCall((*mat->A->ops->restorerow)(mat->A, lrow, &nzA, pcA, pvA)); in MatGetRow_MPIBAIJ()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiov.c | 1218 PetscInt *ai = a->i, *aj = a->j, *bi = b->i, *bj = b->j, nzA, nzB; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local() local
1446 nzA = ai[row + 1] - ai[row]; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1448 ncols = nzA + nzB; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1459 for (PetscInt l = 0; l < nzA; l++) cols[lwrite++] = cstart + cworkA[l]; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1709 nzA = ai[row + 1] - ai[row]; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1711 ncols = nzA + nzB; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1723 for (PetscInt l = 0; l < nzA; l++) vals[lwrite++] = vworkA[l]; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
2345 PetscInt nzA, nzB, *a_i = a->i, *b_i = b->i, lwrite; in MatCreateSubMatrices_MPIAIJ_Local() local
2359 nzA = a_i[row + 1] - a_i[row]; in MatCreateSubMatrices_MPIAIJ_Local()
2361 ncols = nzA + nzB; in MatCreateSubMatrices_MPIAIJ_Local()
[all …]
|
| H A D | mpiaij.c | 1717 PetscInt nztot, nzA, nzB, lrow, rstart = matin->rmap->rstart, rend = matin->rmap->rend; in MatGetRow_MPIAIJ() local
1752 PetscCall((*mat->A->ops->getrow)(mat->A, lrow, &nzA, pcA, pvA)); in MatGetRow_MPIAIJ()
1754 nztot = nzA + nzB; in MatGetRow_MPIAIJ()
1768 for (i = 0; i < nzA; i++) v_p[imark + i] = vworkA[i]; in MatGetRow_MPIAIJ()
1769 for (i = imark; i < nzB; i++) v_p[nzA + i] = vworkB[i]; in MatGetRow_MPIAIJ()
1782 for (i = 0; i < nzA; i++) idx_p[imark + i] = cstart + cworkA[i]; in MatGetRow_MPIAIJ()
1783 for (i = imark; i < nzB; i++) idx_p[nzA + i] = cmap[cworkB[i]]; in MatGetRow_MPIAIJ()
1791 PetscCall((*mat->A->ops->restorerow)(mat->A, lrow, &nzA, pcA, pvA)); in MatGetRow_MPIAIJ()
5365 PetscInt i, start, end, ncols, nzA, nzB, *cmap, imark, *idx; in MatMPIAIJGetLocalMatCondensed() local
5384 nzA = a->A->cmap->n; in MatMPIAIJGetLocalMatCondensed()
[all …]
|
| /petsc/src/mat/impls/sbaij/mpi/ |
| H A D | mpisbaij.c | 1176 PetscInt nztot, nzA, nzB, lrow, brstart = matin->rmap->rstart, brend = matin->rmap->rend; in MatGetRow_MPISBAIJ() local
1212 PetscCall((*mat->A->ops->getrow)(mat->A, lrow, &nzA, pcA, pvA)); in MatGetRow_MPISBAIJ()
1214 nztot = nzA + nzB; in MatGetRow_MPISBAIJ()
1228 for (i = 0; i < nzA; i++) v_p[imark + i] = vworkA[i]; in MatGetRow_MPISBAIJ()
1229 for (i = imark; i < nzB; i++) v_p[nzA + i] = vworkB[i]; in MatGetRow_MPISBAIJ()
1242 for (i = 0; i < nzA; i++) idx_p[imark + i] = cstart * bs + cworkA[i]; in MatGetRow_MPISBAIJ()
1243 for (i = imark; i < nzB; i++) idx_p[nzA + i] = cmap[cworkB[i] / bs] * bs + cworkB[i] % bs; in MatGetRow_MPISBAIJ()
1251 PetscCall((*mat->A->ops->restorerow)(mat->A, lrow, &nzA, pcA, pvA)); in MatGetRow_MPISBAIJ()
|
| /petsc/src/mat/impls/sell/mpi/ |
| H A D | mpisell.c | 1575 PetscInt i, start, end, ncols, nzA, nzB, *cmap, imark, *idx; in MatMPISELLGetLocalMatCondensed() local
1594 nzA = a->A->cmap->n; in MatMPISELLGetLocalMatCondensed()
1596 PetscCall(PetscMalloc1(nzA + nzB, &idx)); in MatMPISELLGetLocalMatCondensed()
1603 for (i = 0; i < nzA; i++) idx[ncols++] = start + i; in MatMPISELLGetLocalMatCondensed()
|