Lines Matching refs:nz

26     PetscInt  nz, brows, bcols;  in MatFDColoringCreate_SeqXAIJ()  local
29 nz = spA->nz; in MatFDColoringCreate_SeqXAIJ()
32 nz = spA->nz; in MatFDColoringCreate_SeqXAIJ()
36 mem = nz * (sizeof(PetscScalar) + sizeof(PetscInt)) + 3 * m * sizeof(PetscInt); in MatFDColoringCreate_SeqXAIJ()
62 PetscErrorCode MatFDColoringSetUpBlocked_AIJ_Private(Mat mat, MatFDColoring c, PetscInt nz) in MatFDColoringSetUpBlocked_AIJ_Private() argument
80 PetscCall(PetscMalloc1(nz, &Jentry_new)); in MatFDColoringSetUpBlocked_AIJ_Private()
99 nz = color_start[j]; in MatFDColoringSetUpBlocked_AIJ_Private()
101 if (Jentry[nz].row >= row_end) { in MatFDColoringSetUpBlocked_AIJ_Private()
102 color_start[j] = nz; in MatFDColoringSetUpBlocked_AIJ_Private()
105 Jentry_new[nz_new].row = Jentry[nz].row + j * mbs; /* index in dy-array */ in MatFDColoringSetUpBlocked_AIJ_Private()
106 Jentry_new[nz_new].col = Jentry[nz].col; in MatFDColoringSetUpBlocked_AIJ_Private()
107 Jentry_new[nz_new].valaddr = Jentry[nz].valaddr; in MatFDColoringSetUpBlocked_AIJ_Private()
109 nz++; in MatFDColoringSetUpBlocked_AIJ_Private()
125 PetscCall(PetscMalloc1(nz, &Jentry2_new)); in MatFDColoringSetUpBlocked_AIJ_Private()
144 nz = color_start[j]; in MatFDColoringSetUpBlocked_AIJ_Private()
146 if (Jentry2[nz].row >= row_end) { in MatFDColoringSetUpBlocked_AIJ_Private()
147 color_start[j] = nz; in MatFDColoringSetUpBlocked_AIJ_Private()
150 Jentry2_new[nz_new].row = Jentry2[nz].row + j * mbs; /* index in dy-array */ in MatFDColoringSetUpBlocked_AIJ_Private()
151 Jentry2_new[nz_new].valaddr = Jentry2[nz].valaddr; in MatFDColoringSetUpBlocked_AIJ_Private()
153 nz++; in MatFDColoringSetUpBlocked_AIJ_Private()
178 …rows, mbs = c->m, j, k, m, ncols, col, nis = iscoloring->n, *rowhit, bs, bs2, *spidx, nz, tmp, nnz; in MatFDColoringSetUp_SeqXAIJ() local
196 nz = spA->nz; in MatFDColoringSetUp_SeqXAIJ()
201 nz = spA->nz; in MatFDColoringSetUp_SeqXAIJ()
207 nz = spA->nz; in MatFDColoringSetUp_SeqXAIJ()
214 PetscCall(PetscMalloc1(nz, &Jentry)); in MatFDColoringSetUp_SeqXAIJ()
217 PetscCall(PetscMalloc1(nz, &Jentry2)); in MatFDColoringSetUp_SeqXAIJ()
280 PetscCheck(nnz == nz, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Incorrect coloring of matrix"); in MatFDColoringSetUp_SeqXAIJ()
282 PetscCall(MatFDColoringSetUpBlocked_AIJ_Private(mat, c, nz)); in MatFDColoringSetUp_SeqXAIJ()