Lines Matching refs:nzA
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()
5386 PetscCall(PetscMalloc1(nzA + nzB, &idx)); in MatMPIAIJGetLocalMatCondensed()
5393 for (i = 0; i < nzA; i++) idx[ncols++] = start + i; in MatMPIAIJGetLocalMatCondensed()
5688 PetscInt *idx, i, start, ncols, nzA, nzB, *cmap, imark; in MatGetBrowsOfAcols() local
5700 nzA = a->A->cmap->n; in MatGetBrowsOfAcols()
5702 PetscCall(PetscMalloc1(nzA + nzB, &idx)); in MatGetBrowsOfAcols()
5709 for (i = 0; i < nzA; i++) idx[ncols++] = start + i; /* local rows */ in MatGetBrowsOfAcols()