Lines Matching refs:nzB
1717 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()
5386 PetscCall(PetscMalloc1(nzA + nzB, &idx)); in MatMPIAIJGetLocalMatCondensed()
5388 for (i = 0; i < nzB; i++) { in MatMPIAIJGetLocalMatCondensed()
5394 for (i = imark; i < nzB; i++) idx[ncols++] = cmap[i]; in MatMPIAIJGetLocalMatCondensed()
5688 PetscInt *idx, i, start, ncols, nzA, nzB, *cmap, imark; in MatGetBrowsOfAcols() local
5701 nzB = a->B->cmap->n; in MatGetBrowsOfAcols()
5702 PetscCall(PetscMalloc1(nzA + nzB, &idx)); in MatGetBrowsOfAcols()
5704 for (i = 0; i < nzB; i++) { /* row < local row index */ in MatGetBrowsOfAcols()
5710 for (i = imark; i < nzB; i++) idx[ncols++] = cmap[i]; /* row > local row index */ in MatGetBrowsOfAcols()