Home
last modified time | relevance | path

Searched refs:lens (Results 1 – 17 of 17) sorted by relevance

/petsc/src/snes/tests/
H A Dex5.c28 PetscInt its, n = 5, nb, maxit, maxf, *lens; in main() local
67 PetscCall(PetscMalloc1(nb, &lens)); in main()
69 lens[3 * i + 0] = 1; in main()
70 lens[3 * i + 1] = 2; in main()
71 lens[3 * i + 2] = 2; in main()
73 PetscCall(MatSetVariableBlockSizes(J, nb, lens)); in main()
74 PetscCall(PetscFree(lens)); in main()
/petsc/src/vec/is/ao/impls/mapping/
H A Daomapping.c245 PetscMPIInt size, rank, *lens, *disp, nnapp; in AOCreateMapping() local
262 PetscCall(PetscMalloc2(size, &lens, size, &disp)); in AOCreateMapping()
264 PetscCallMPI(MPI_Allgather(&nnapp, 1, MPI_INT, lens, 1, MPI_INT, comm)); in AOCreateMapping()
268 N += lens[i]; in AOCreateMapping()
285 PetscCallMPI(MPI_Allgatherv((void *)myapp, nnapp, MPIU_INT, allapp, lens, disp, MPIU_INT, comm)); in AOCreateMapping()
286 …PetscCallMPI(MPI_Allgatherv((void *)petsc, nnapp, MPIU_INT, allpetsc, lens, disp, MPIU_INT, comm)); in AOCreateMapping()
287 PetscCall(PetscFree2(lens, disp)); in AOCreateMapping()
/petsc/src/mat/graphops/order/
H A Dsorder.c51 PetscInt n, *permr, *lens, i; in MatGetOrdering_RowLength() local
59 PetscCall(PetscMalloc2(n, &lens, n, &permr)); in MatGetOrdering_RowLength()
61 lens[i] = ia[i + 1] - ia[i]; in MatGetOrdering_RowLength()
66 PetscCall(PetscSortIntWithPermutation(n, lens, permr)); in MatGetOrdering_RowLength()
70 PetscCall(PetscFree2(lens, permr)); in MatGetOrdering_RowLength()
/petsc/src/sys/utils/
H A Dstr.c38 int n, i, j, *lens, cnt = 0; in PetscStrToArray() local
62 lens = (int *)malloc(((*argc) + 1) * sizeof(int)); in PetscStrToArray()
63 if (!lens) return PETSC_ERR_MEM; in PetscStrToArray()
64 for (i = 0; i < *argc; i++) lens[i] = 0; in PetscStrToArray()
75 lens[*argc]++; in PetscStrToArray()
81 (*args)[i] = (char *)malloc((lens[i] + 1) * sizeof(char)); in PetscStrToArray()
83 free(lens); in PetscStrToArray()
89 free(lens); in PetscStrToArray()
/petsc/src/vec/is/ao/impls/basic/
H A Daobasic.c157 PetscMPIInt size, rank, count, *lens, *disp; in AOCreate_Basic() local
179 PetscCall(PetscMalloc2(size, &lens, size, &disp)); in AOCreate_Basic()
180 PetscCallMPI(MPI_Allgather(&count, 1, MPI_INT, lens, 1, MPI_INT, comm)); in AOCreate_Basic()
184 N += lens[i]; in AOCreate_Basic()
203 PetscCallMPI(MPI_Allgatherv(petsc, count, MPIU_INT, allpetsc, lens, disp, MPIU_INT, comm)); in AOCreate_Basic()
204 PetscCallMPI(MPI_Allgatherv((void *)myapp, count, MPIU_INT, allapp, lens, disp, MPIU_INT, comm)); in AOCreate_Basic()
205 PetscCall(PetscFree2(lens, disp)); in AOCreate_Basic()
/petsc/src/ksp/pc/impls/bjacobi/
H A Dbjacobi.c271 static PetscErrorCode PCBJacobiSetTotalBlocks_BJacobi(PC pc, PetscInt blocks, const PetscInt *lens) in PCBJacobiSetTotalBlocks_BJacobi() argument
278 if (!lens) jac->g_lens = NULL; in PCBJacobiSetTotalBlocks_BJacobi()
281 PetscCall(PetscArraycpy(jac->g_lens, lens, blocks)); in PCBJacobiSetTotalBlocks_BJacobi()
286 static PetscErrorCode PCBJacobiGetTotalBlocks_BJacobi(PC pc, PetscInt *blocks, const PetscInt *lens… in PCBJacobiGetTotalBlocks_BJacobi() argument
292 if (lens) *lens = jac->g_lens; in PCBJacobiGetTotalBlocks_BJacobi()
296 static PetscErrorCode PCBJacobiSetLocalBlocks_BJacobi(PC pc, PetscInt blocks, const PetscInt lens[]) in PCBJacobiSetLocalBlocks_BJacobi() argument
304 if (!lens) jac->l_lens = NULL; in PCBJacobiSetLocalBlocks_BJacobi()
307 PetscCall(PetscArraycpy(jac->l_lens, lens, blocks)); in PCBJacobiSetLocalBlocks_BJacobi()
312 static PetscErrorCode PCBJacobiGetLocalBlocks_BJacobi(PC pc, PetscInt *blocks, const PetscInt *lens… in PCBJacobiGetLocalBlocks_BJacobi() argument
318 if (lens) *lens = jac->l_lens; in PCBJacobiGetLocalBlocks_BJacobi()
[all …]
/petsc/src/dm/impls/plex/transform/impls/refine/sbr/
H A Dplexrefsbr.c244 PetscReal lens[3]; in DMPlexTransformSetUp_SBR() local
251 PetscCall(SBRGetEdgeLen_Private(tr, cone[i], &lens[i])); in DMPlexTransformSetUp_SBR()
254 …else if (vals[0] && vals[1]) PetscCall(DMLabelSetValue(trType, p, lens[0] > lens[1] ? RT_TRIANGLE_… in DMPlexTransformSetUp_SBR()
255 …else if (vals[1] && vals[2]) PetscCall(DMLabelSetValue(trType, p, lens[1] > lens[2] ? RT_TRIANGLE_… in DMPlexTransformSetUp_SBR()
256 …else if (vals[2] && vals[0]) PetscCall(DMLabelSetValue(trType, p, lens[2] > lens[0] ? RT_TRIANGLE_… in DMPlexTransformSetUp_SBR()
/petsc/src/vec/is/ao/impls/memscalable/
H A Daomemscalable.c389 PetscInt napp, n_local, N, i, start, *petsc, *lens, *disp; in AOCreate_MemoryScalable() local
407 PetscCall(PetscMalloc2(size, &lens, size, &disp)); in AOCreate_MemoryScalable()
409 PetscCallMPI(MPI_Allgather(&napp, 1, MPIU_INT, lens, 1, MPIU_INT, comm)); in AOCreate_MemoryScalable()
414 N += lens[i]; in AOCreate_MemoryScalable()
457 PetscCall(PetscFree2(lens, disp)); in AOCreate_MemoryScalable()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiov.c1090 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
[all …]
/petsc/src/vec/vec/impls/mpi/
H A Dpdvec.c278 PetscMPIInt i, size, rank, n, N, *lens = NULL, *disp = NULL; in VecView_MPI_Draw_LG() local
302 PetscCall(PetscMalloc2(size, &lens, size, &disp)); in VecView_MPI_Draw_LG()
303 …; i < size; i++) PetscCall(PetscMPIIntCast(xin->map->range[i + 1] - xin->map->range[i], &lens[i])); in VecView_MPI_Draw_LG()
306 …PetscCallMPI(MPI_Gatherv(values, n, MPIU_REAL, yy, lens, disp, MPIU_REAL, 0, PetscObjectComm((Pets… in VecView_MPI_Draw_LG()
307 PetscCall(PetscFree2(lens, disp)); in VecView_MPI_Draw_LG()
383 PetscMPIInt rank, size, *lens; in VecView_MPI_Matlab() local
394 PetscCall(PetscMalloc1(size, &lens)); in VecView_MPI_Matlab()
395 for (i = 0; i < size; i++) lens[i] = xin->map->range[i + 1] - xin->map->range[i]; in VecView_MPI_Matlab()
397 …PetscCallMPI(MPI_Gatherv((void *)xarray, xin->map->n, MPIU_SCALAR, xx, lens, xin->map->range, MPIU… in VecView_MPI_Matlab()
398 PetscCall(PetscFree(lens)); in VecView_MPI_Matlab()
/petsc/src/mat/impls/baij/mpi/
H A Dbaijov.c625 PetscInt **lens, is_no, ncols, *cols, mat_i, *mat_j, tmp2, jmax; local
1003 PetscCall(PetscMalloc1(ismax, &lens));
1004 if (ismax) PetscCall(PetscCalloc1(jcnt, &lens[0]));
1005 …for (PetscInt i = 1; i < ismax; i++) lens[i] = PetscSafePointerPlusOffset(lens[i - 1], nrow[i - 1]…
1013 lens_i = lens[i];
1098 lens_i = lens[is_no];
1144 PetscCall(MatSeqBAIJSetPreallocation(submats[i], bs_tmp, 0, lens[i]));
1145 …PetscCall(MatSeqSBAIJSetPreallocation(submats[i], bs_tmp, 0, lens[i])); /* this subroutine is used…
1220 if (ismax) PetscCall(PetscFree(lens[0]));
1221 PetscCall(PetscFree(lens));
H A Dmpibaij.c2095 PetscInt m, *garray = a->garray, *lens, *jsendbuf, *a_jsendbuf, *b_jsendbuf; in MatGetSeqNonzeroStructure_MPIBAIJ() local
2102 PetscCall(PetscMalloc1(A->rmap->N / bs, &lens)); in MatGetSeqNonzeroStructure_MPIBAIJ()
2103 …for (i = A->rmap->rstart / bs; i < A->rmap->rend / bs; i++) lens[i] = ad->i[i - A->rmap->rstart / … in MatGetSeqNonzeroStructure_MPIBAIJ()
2110 …PetscCallMPI(MPI_Allgatherv(MPI_IN_PLACE, 0, MPI_DATATYPE_NULL, lens, recvcounts, displs, MPIU_INT… in MatGetSeqNonzeroStructure_MPIBAIJ()
2115 PetscCall(MatSeqAIJSetPreallocation(B, 0, lens)); in MatGetSeqNonzeroStructure_MPIBAIJ()
2146 for (j = A->rmap->range[i] / bs; j < A->rmap->range[i + 1] / bs; j++) recvcounts[i] += lens[j]; in MatGetSeqNonzeroStructure_MPIBAIJ()
2153 PetscCall(PetscArraycpy(b->ilen, lens, A->rmap->N / bs)); in MatGetSeqNonzeroStructure_MPIBAIJ()
2156 for (i = 1; i <= A->rmap->N / bs; i++) b->i[i] = b->i[i - 1] + lens[i - 1]; in MatGetSeqNonzeroStructure_MPIBAIJ()
2157 PetscCall(PetscFree(lens)); in MatGetSeqNonzeroStructure_MPIBAIJ()
/petsc/src/mat/impls/blockmat/seq/
H A Dblockmat.c287 …PetscInt i, j, m, n, bs = 1, ncols, *lens, currentcol, mbs, **ii, *ilens, nextcol, *llen… in MatLoad_BlockMat() local
310 PetscCall(PetscMalloc3(mbs, &lens, bs, &ii, bs, &ilens)); in MatLoad_BlockMat()
311 PetscCall(PetscArrayzero(lens, mbs)); in MatLoad_BlockMat()
334 if (!flg || (nextcol >= i)) lens[i]++; in MatLoad_BlockMat()
340 PetscCall(MatBlockMatSetPreallocation(newmat, bs, 0, lens)); in MatLoad_BlockMat()
381 amat->ilen[i] = lens[i]; in MatLoad_BlockMat()
384 PetscCall(PetscFree3(lens, ii, ilens)); in MatLoad_BlockMat()
/petsc/lib/petsc/bin/maint/petsclinter/petsclinter/classes/docs/
H A D_doc_section_base.py888 lens = list(map(len, group_args))
889 max_arg_len = max(lens, default=0)
890 longest_arg = group_args[lens.index(max_arg_len)] if lens else 'NO ARGS'
/petsc/src/mat/impls/sbaij/seq/
H A Dsbaij2.c109 PetscInt *smap, i, k, kstart, kend, oldcols = a->nbs, *lens; in MatCreateSubMatrix_SeqSBAIJ_Private() local
126 PetscCall(PetscMalloc1(1 + nrows, &lens)); in MatCreateSubMatrix_SeqSBAIJ_Private()
132 lens[i] = 0; in MatCreateSubMatrix_SeqSBAIJ_Private()
134 if (ssmap[aj[k]]) lens[i]++; in MatCreateSubMatrix_SeqSBAIJ_Private()
143 PetscCall(PetscArraycmp(c->ilen, lens, c->mbs, &flag)); in MatCreateSubMatrix_SeqSBAIJ_Private()
150 PetscCall(PetscArraycmp(d->ilen, lens, d->mbs, &flag)); in MatCreateSubMatrix_SeqSBAIJ_Private()
160 PetscCall(MatSeqSBAIJSetPreallocation(C, bs, 0, lens)); in MatCreateSubMatrix_SeqSBAIJ_Private()
163 PetscCall(MatSeqBAIJSetPreallocation(C, bs, 0, lens)); in MatCreateSubMatrix_SeqSBAIJ_Private()
218 PetscCall(PetscFree(lens)); in MatCreateSubMatrix_SeqSBAIJ_Private()
/petsc/src/mat/impls/aij/seq/
H A Daij.c2391 PetscInt *smap, i, k, kstart, kend, oldcols = A->cmap->n, *lens; in MatCreateSubMatrix_SeqAIJ() local
2415 PetscCall(PetscMalloc2(nrows, &lens, nrows, &starts)); in MatCreateSubMatrix_SeqAIJ()
2432 lens[i] = sum; in MatCreateSubMatrix_SeqAIJ()
2449 PetscCall(MatSeqAIJSetPreallocation_SeqAIJ(C, 0, lens)); in MatCreateSubMatrix_SeqAIJ()
2460 lensi = lens[i]; in MatCreateSubMatrix_SeqAIJ()
2471 PetscCall(PetscFree2(lens, starts)); in MatCreateSubMatrix_SeqAIJ()
2475 PetscCall(PetscMalloc1(1 + nrows, &lens)); in MatCreateSubMatrix_SeqAIJ()
2485 lens[i] = 0; in MatCreateSubMatrix_SeqAIJ()
2487 if (smap[aj[k]]) lens[i]++; in MatCreateSubMatrix_SeqAIJ()
2496 PetscCall(PetscArraycmp(c->ilen, lens, (*B)->rmap->n, &equal)); in MatCreateSubMatrix_SeqAIJ()
[all …]
/petsc/src/mat/impls/baij/seq/
H A Dbaij2.c73 PetscInt *smap, i, k, kstart, kend, oldcols = a->nbs, *lens; in MatCreateSubMatrix_SeqBAIJ_Private() local
90 PetscCall(PetscMalloc1(1 + nrows, &lens)); in MatCreateSubMatrix_SeqBAIJ_Private()
96 lens[i] = 0; in MatCreateSubMatrix_SeqBAIJ_Private()
98 if (ssmap[aj[k]]) lens[i]++; in MatCreateSubMatrix_SeqBAIJ_Private()
106 PetscCall(PetscArraycmp(c->ilen, lens, c->mbs, &flag)); in MatCreateSubMatrix_SeqBAIJ_Private()
114 PetscCall(MatSeqBAIJSetPreallocation(C, bs, 0, lens)); in MatCreateSubMatrix_SeqBAIJ_Private()
152 PetscCall(PetscFree(lens)); in MatCreateSubMatrix_SeqBAIJ_Private()