Lines Matching refs:nz
34 PetscInt k, cstart, cend, l, row, col, nz, spidx, i, j; in MatFDColoringApply_BAIJ() local
95 nz = 0; in MatFDColoringApply_BAIJ()
145 row = bs * Jentry2[nz].row; /* local row index */ in MatFDColoringApply_BAIJ()
146 valaddr = Jentry2[nz++].valaddr; in MatFDColoringApply_BAIJ()
158 row = bs * Jentry[nz].row; /* local row index */ in MatFDColoringApply_BAIJ()
159 col = bs * Jentry[nz].col; /* local column index */ in MatFDColoringApply_BAIJ()
160 valaddr = Jentry[nz++].valaddr; in MatFDColoringApply_BAIJ()
185 PetscInt k, cstart, cend, l, row, col, nz; in MatFDColoringApply_AIJ() local
245 nz = 0; in MatFDColoringApply_AIJ()
302 row = Jentry2[nz].row; /* local row index */ in MatFDColoringApply_AIJ()
307 PetscScalar *tmp = Jentry2[nz].valaddr; in MatFDColoringApply_AIJ()
310 *Jentry2[nz].valaddr = dy[row] * dx; in MatFDColoringApply_AIJ()
312 nz++; in MatFDColoringApply_AIJ()
316 row = Jentry[nz].row; /* local row index */ in MatFDColoringApply_AIJ()
318 PetscScalar *tmp = Jentry[nz].valaddr; in MatFDColoringApply_AIJ()
319 *tmp = dy[row] * vscale_array[Jentry[nz].col]; in MatFDColoringApply_AIJ()
321 *Jentry[nz].valaddr = dy[row] * vscale_array[Jentry[nz].col]; in MatFDColoringApply_AIJ()
323 nz++; in MatFDColoringApply_AIJ()
370 row = Jentry2[nz].row; /* local row index */ in MatFDColoringApply_AIJ()
372 PetscScalar *tmp = Jentry2[nz].valaddr; in MatFDColoringApply_AIJ()
375 *Jentry2[nz].valaddr = y[row] * dx; in MatFDColoringApply_AIJ()
377 nz++; in MatFDColoringApply_AIJ()
381 row = Jentry[nz].row; /* local row index */ in MatFDColoringApply_AIJ()
383 PetscScalar *tmp = Jentry[nz].valaddr; in MatFDColoringApply_AIJ()
384 *tmp = y[row] * vscale_array[Jentry[nz].col]; in MatFDColoringApply_AIJ()
386 *Jentry[nz].valaddr = y[row] * vscale_array[Jentry[nz].col]; in MatFDColoringApply_AIJ()
388 nz++; in MatFDColoringApply_AIJ()
410 PetscInt ctype = c->ctype, *spidxA, *spidxB, nz, bs, bs2, spidx; in MatFDColoringSetUp_MPIXAIJ() local
441 nz = spA->nz + spB->nz; /* total nonzero entries of mat */ in MatFDColoringSetUp_MPIXAIJ()
466 nz = spA->nz + spB->nz; /* total nonzero entries of mat */ in MatFDColoringSetUp_MPIXAIJ()
491 nz = spA->nz + spB->nz; /* total nonzero entries of mat */ in MatFDColoringSetUp_MPIXAIJ()
517 PetscCall(PetscMalloc1(nz, &Jentry)); in MatFDColoringSetUp_MPIXAIJ()
520 PetscCall(PetscMalloc1(nz, &Jentry2)); in MatFDColoringSetUp_MPIXAIJ()
525 nz = 0; in MatFDColoringSetUp_MPIXAIJ()
615 Jentry[nz].row = j; /* local row index */ in MatFDColoringSetUp_MPIXAIJ()
616 Jentry[nz].col = rowhit[j] - 1; /* local column index */ in MatFDColoringSetUp_MPIXAIJ()
617 Jentry[nz].valaddr = valaddrhit[j]; /* address of mat value for this entry */ in MatFDColoringSetUp_MPIXAIJ()
618 nz++; in MatFDColoringSetUp_MPIXAIJ()
624 Jentry2[nz].row = j; /* local row index */ in MatFDColoringSetUp_MPIXAIJ()
625 Jentry2[nz].valaddr = valaddrhit[j]; /* address of mat value for this entry */ in MatFDColoringSetUp_MPIXAIJ()
626 nz++; in MatFDColoringSetUp_MPIXAIJ()
635 PetscCall(MatFDColoringSetUpBlocked_AIJ_Private(mat, c, nz)); in MatFDColoringSetUp_MPIXAIJ()
677 PetscInt nz, brows, bcols; in MatFDColoringCreate_MPIXAIJ() local
686 nz = spA->nz + spB->nz; /* total local nonzero entries of mat */ in MatFDColoringCreate_MPIXAIJ()
687 mem = nz * (sizeof(PetscScalar) + sizeof(PetscInt)) + 3 * m * sizeof(PetscInt); in MatFDColoringCreate_MPIXAIJ()
699 PetscInt nz, brows, bcols; in MatFDColoringCreate_MPIXAIJ() local
708 nz = spA->nz + spB->nz; /* total local nonzero entries of mat */ in MatFDColoringCreate_MPIXAIJ()
709 mem = nz * (sizeof(PetscScalar) + sizeof(PetscInt)) + 3 * m * sizeof(PetscInt); in MatFDColoringCreate_MPIXAIJ()
749 PetscInt row, nrows_k, l, ncolors, nz = 0, bcols, nbcols = 0; in MatFDColoringSetValues() local
766 row = Jentry2[nz].row; /* local row index */ in MatFDColoringSetValues()
767 *Jentry2[nz++].valaddr = y[row]; in MatFDColoringSetValues()