Home
last modified time | relevance | path

Searched refs:nzB (Results 1 – 7 of 7) sorted by relevance

/petsc/src/mat/tests/
H A Dex225.c104 PetscInt rstart, rend, nzA, nzB; in main() local
113 PetscCall(MatGetRow(B, i, &nzB, &idxB, &vB)); in main()
114 …PetscCheck(nzA == nzB, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Error MatGetRow %" PetscInt_FMT, nzA - nz… in main()
115 PetscCall(PetscSortIntWithScalarArray(nzB, (PetscInt *)idxB, (PetscScalar *)vB)); in main()
121 PetscCall(MatRestoreRow(B, i, &nzB, &idxB, &vB)); in main()
/petsc/src/mat/impls/baij/mpi/
H A Dbaijov.c644 …PetscInt nzA, nzB, *a_i = a->i, *b_i = b->i, *a_j = a->j, *b_j = b->j, ctmp, imark, *cworkA… local
955 nzB = b_i[row + 1] - b_i[row];
956 ncols = nzA + nzB;
962 for (l = 0; l < nzB; l++) {
968 for (l = imark; l < nzB; l++) cols[nzA + l] = bmap[cworkB[l]];
1026 nzB = b_i[row + 1] - b_i[row];
1036 for (PetscInt k = 0; k < nzB; k++) {
1045 for (PetscInt k = 0; k < nzB; k++) {
1050 lens_i[j] = nzA + nzB;
1260 nzB = b_i[row + 1] - b_i[row];
[all …]
H A Dmpibaij.c1304 PetscInt nztot, nzA, nzB, lrow, brstart = matin->rmap->rstart, brend = matin->rmap->rend; in MatGetRow_MPIBAIJ() local
1339 PetscCall((*mat->B->ops->getrow)(mat->B, lrow, &nzB, pcB, pvB)); in MatGetRow_MPIBAIJ()
1340 nztot = nzA + nzB; in MatGetRow_MPIBAIJ()
1349 for (i = 0; i < nzB; i++) { in MatGetRow_MPIBAIJ()
1355 for (i = imark; i < nzB; i++) v_p[nzA + i] = vworkB[i]; in MatGetRow_MPIBAIJ()
1362 for (i = 0; i < nzB; 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()
1378 PetscCall((*mat->B->ops->restorerow)(mat->B, lrow, &nzB, pcB, pvB)); in MatGetRow_MPIBAIJ()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiov.c1218 PetscInt *ai = a->i, *aj = a->j, *bi = b->i, *bj = b->j, nzA, nzB; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local() local
1447 nzB = bi[row + 1] - bi[row]; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1448 ncols = nzA + nzB; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1456 for (PetscInt l = 0; l < nzB; l++) { in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1460 for (PetscInt l = 0; l < nzB; l++) { in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1710 nzB = bi[row + 1] - bi[row]; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1711 ncols = nzA + nzB; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1720 for (PetscInt l = 0; l < nzB; l++) { in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1724 for (PetscInt l = 0; l < nzB; l++) { in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
2345 PetscInt nzA, nzB, *a_i = a->i, *b_i = b->i, lwrite; in MatCreateSubMatrices_MPIAIJ_Local() local
[all …]
H A Dmpiaij.c1717 PetscInt nztot, nzA, nzB, lrow, rstart = matin->rmap->rstart, rend = matin->rmap->rend; in MatGetRow_MPIAIJ() local
1753 PetscCall((*mat->B->ops->getrow)(mat->B, lrow, &nzB, pcB, pvB)); in MatGetRow_MPIAIJ()
1754 nztot = nzA + nzB; in MatGetRow_MPIAIJ()
1763 for (i = 0; i < nzB; i++) { in MatGetRow_MPIAIJ()
1769 for (i = imark; i < nzB; i++) v_p[nzA + i] = vworkB[i]; in MatGetRow_MPIAIJ()
1776 for (i = 0; i < nzB; i++) { in MatGetRow_MPIAIJ()
1783 for (i = imark; i < nzB; i++) idx_p[nzA + i] = cmap[cworkB[i]]; in MatGetRow_MPIAIJ()
1792 PetscCall((*mat->B->ops->restorerow)(mat->B, lrow, &nzB, pcB, pvB)); in MatGetRow_MPIAIJ()
5365 PetscInt i, start, end, ncols, nzA, nzB, *cmap, imark, *idx; in MatMPIAIJGetLocalMatCondensed() local
5385 nzB = a->B->cmap->n; in MatMPIAIJGetLocalMatCondensed()
[all …]
/petsc/src/mat/impls/sell/mpi/
H A Dmpisell.c1575 PetscInt i, start, end, ncols, nzA, nzB, *cmap, imark, *idx; in MatMPISELLGetLocalMatCondensed() local
1595 nzB = a->B->cmap->n; in MatMPISELLGetLocalMatCondensed()
1596 PetscCall(PetscMalloc1(nzA + nzB, &idx)); in MatMPISELLGetLocalMatCondensed()
1598 for (i = 0; i < nzB; i++) { in MatMPISELLGetLocalMatCondensed()
1604 for (i = imark; i < nzB; i++) idx[ncols++] = cmap[i]; in MatMPISELLGetLocalMatCondensed()
/petsc/src/mat/impls/sbaij/mpi/
H A Dmpisbaij.c1176 PetscInt nztot, nzA, nzB, lrow, brstart = matin->rmap->rstart, brend = matin->rmap->rend; in MatGetRow_MPISBAIJ() local
1213 PetscCall((*mat->B->ops->getrow)(mat->B, lrow, &nzB, pcB, pvB)); in MatGetRow_MPISBAIJ()
1214 nztot = nzA + nzB; in MatGetRow_MPISBAIJ()
1223 for (i = 0; i < nzB; i++) { in MatGetRow_MPISBAIJ()
1229 for (i = imark; i < nzB; i++) v_p[nzA + i] = vworkB[i]; in MatGetRow_MPISBAIJ()
1236 for (i = 0; i < nzB; 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()
1252 PetscCall((*mat->B->ops->restorerow)(mat->B, lrow, &nzB, pcB, pvB)); in MatGetRow_MPISBAIJ()