Lines Matching refs:lens

1090   PetscInt    m, *b_sendj, *garray = a->garray, *lens, *jsendbuf, *a_jsendbuf, *b_jsendbuf;  in MatCreateSubMatrix_MPIAIJ_All()  local
1097 PetscCall(PetscCalloc1(A->rmap->N, &lens)); in MatCreateSubMatrix_MPIAIJ_All()
1098 …for (PetscInt i = A->rmap->rstart; i < A->rmap->rend; i++) lens[i] = ad->i[i - A->rmap->rstart + 1… in MatCreateSubMatrix_MPIAIJ_All()
1101 …PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, lens, A->rmap->N, MPIU_INT, MPI_SUM, PetscObjectComm((Pe… in MatCreateSubMatrix_MPIAIJ_All()
1108 PetscCall(MatSeqAIJSetPreallocation(B, 0, lens)); in MatCreateSubMatrix_MPIAIJ_All()
1116 for (j = A->rmap->range[i]; j < A->rmap->range[i + 1]; j++) nrecv += lens[j]; in MatCreateSubMatrix_MPIAIJ_All()
1150 PetscCall(PetscArraycpy(b->ilen, lens, A->rmap->N)); in MatCreateSubMatrix_MPIAIJ_All()
1153 for (PetscInt i = 1; i <= A->rmap->N; i++) b->i[i] = b->i[i - 1] + lens[i - 1]; in MatCreateSubMatrix_MPIAIJ_All()
1154 PetscCall(PetscFree(lens)); in MatCreateSubMatrix_MPIAIJ_All()
1226 PetscInt *lens, rmax, ncols, *cols, Crow; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local() local
1496 PetscCall(PetscCalloc1(nrow, &lens)); in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1512 if (tcol) lens[j]++; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1515 lens[j] = ncols; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1528 if (tcol) lens[j]++; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1531 lens[j] += ncols; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1588 if (tcol) lens[row]++; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1591 lens[row] += nnz; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1606 PetscCall(MatSeqAIJSetPreallocation(submat, 0, lens)); in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1647 for (PetscMPIInt i = 0; i < nrow; i++) rmax = PetscMax(rmax, lens[i]); in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1908 PetscCall(PetscFree(lens)); in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
2046 PetscInt **lens, is_no, ncols, *cols, mat_i, *mat_j, tmp2, jmax; in MatCreateSubMatrices_MPIAIJ_Local() local
2414 PetscCall(PetscMalloc1(ismax, &lens)); in MatCreateSubMatrices_MPIAIJ_Local()
2416 if (ismax) PetscCall(PetscCalloc1(jcnt, &lens[0])); in MatCreateSubMatrices_MPIAIJ_Local()
2417 …for (PetscInt i = 1; i < ismax; i++) lens[i] = PetscSafePointerPlusOffset(lens[i - 1], nrow[i - 1]… in MatCreateSubMatrices_MPIAIJ_Local()
2425 lens_i = lens[i]; in MatCreateSubMatrices_MPIAIJ_Local()
2481 lens_i = lens[is_no]; in MatCreateSubMatrices_MPIAIJ_Local()
2525 PetscCall(MatSeqAIJSetPreallocation(submats[i], 0, lens[i])); in MatCreateSubMatrices_MPIAIJ_Local()
2599 if (ismax) PetscCall(PetscFree(lens[0])); in MatCreateSubMatrices_MPIAIJ_Local()
2600 PetscCall(PetscFree(lens)); in MatCreateSubMatrices_MPIAIJ_Local()