Lines Matching refs:j2
6081 static PetscErrorCode MatMergeEntries_Internal(Mat mat, const PetscInt j1[], const PetscInt j2[], c… in MatMergeEntries_Internal() argument
6096 if (j1[b1] == j2[b2]) { /* Same column index and hence same nonzero */ in MatMergeEntries_Internal()
6105 } else if (j1[b1] < j2[b2]) { in MatMergeEntries_Internal()
6112 j[t] = j2[b2]; in MatMergeEntries_Internal()
6128 j[t] = j2[b2]; in MatMergeEntries_Internal()
6487 …PetscInt n2 = nroots, *i2, *j2; /* Buffers for received COOs from other ranks, along with a per… in MatSetPreallocationCOO_MPIAIJ() local
6489 PetscCall(PetscMalloc3(n2, &i2, n2, &j2, n2, &perm2)); in MatSetPreallocationCOO_MPIAIJ()
6496 …eWithMemTypeBegin(sf2, MPIU_INT, PETSC_MEMTYPE_HOST, j1prem, PETSC_MEMTYPE_HOST, j2, MPI_REPLACE)); in MatSetPreallocationCOO_MPIAIJ()
6497 PetscCall(PetscSFReduceEnd(sf2, MPIU_INT, j1prem, j2, MPI_REPLACE)); in MatSetPreallocationCOO_MPIAIJ()
6504 PetscCall(PetscSortIntWithIntCountArrayPair(n2, i2, j2, perm2)); in MatSetPreallocationCOO_MPIAIJ()
6531 if (j2[k] < cstart || j2[k] >= cend) continue; in MatSetPreallocationCOO_MPIAIJ()
6533 if ((j2[k] - cstart) == rindex) PetscCall(PetscBTSet(hasdiag, rindex)); in MatSetPreallocationCOO_MPIAIJ()
6534 minj[rindex] = PetscMin(minj[rindex], j2[k]); in MatSetPreallocationCOO_MPIAIJ()
6544 if (j2[k] < cstart || j2[k] >= cend || !PetscBTLookup(hasdiag, rindex)) continue; in MatSetPreallocationCOO_MPIAIJ()
6545 if (j2[k] == minj[rindex]) j2[k] = i2[k] + (cstart - rstart); in MatSetPreallocationCOO_MPIAIJ()
6546 else if ((j2[k] - cstart) == rindex) j2[k] = minj[rindex]; in MatSetPreallocationCOO_MPIAIJ()
6563 …PetscCall(MatSplitEntries_Internal(mat, n2, i2, j2, perm2, rowBegin2, rowMid2, rowEnd2, &Atot2, &A… in MatSetPreallocationCOO_MPIAIJ()
6580 …PetscCall(MatMergeEntries_Internal(mat, j1, j2, rowBegin1, rowMid1, rowBegin2, rowMid2, Ajmap1, Aj… in MatSetPreallocationCOO_MPIAIJ()
6581 …PetscCall(MatMergeEntries_Internal(mat, j1, j2, rowMid1, rowEnd1, rowMid2, rowEnd2, Bjmap1, Bjmap2… in MatSetPreallocationCOO_MPIAIJ()
6602 PetscCall(PetscFree3(i2, j2, perm2)); in MatSetPreallocationCOO_MPIAIJ()