Home
last modified time | relevance | path

Searched refs:arr (Results 1 – 25 of 36) sorted by relevance

12

/petsc/src/sys/utils/
H A Dsortso.c134 static inline PetscErrorCode PetscGallopSearchLeft_Private(const char *arr, size_t size, CompFunc c… in PetscGallopSearchLeft_Private() argument
141 if ((*cmp)(x, arr + r * size, ctx) >= 0) { in PetscGallopSearchLeft_Private()
145 …if ((*cmp)(x, (arr) + l * size, ctx) < 0 || PetscUnlikely(!(r - l))) PetscFunctionReturn(PETSC_SUC… in PetscGallopSearchLeft_Private()
151 if ((*cmp)(x, (arr) + cur * size, ctx) < 0) break; in PetscGallopSearchLeft_Private()
159 if ((*cmp)(x, (arr) + mid * size, ctx) < 0) { in PetscGallopSearchLeft_Private()
171 static inline PetscErrorCode PetscGallopSearchRight_Private(const char *arr, size_t size, CompFunc … in PetscGallopSearchRight_Private() argument
178 if ((*cmp)(x, arr + l * size, ctx) <= 0) { in PetscGallopSearchRight_Private()
182 …if ((*cmp)(x, (arr) + r * size, ctx) > 0 || PetscUnlikely(!(r - l))) PetscFunctionReturn(PETSC_SUC… in PetscGallopSearchRight_Private()
188 if ((*cmp)(x, (arr) + cur * size, ctx) > 0) break; in PetscGallopSearchRight_Private()
196 if ((*cmp)(x, (arr) + mid * size, ctx) > 0) { in PetscGallopSearchRight_Private()
[all …]
/petsc/src/dm/impls/stag/tests/
H A Dex13.c139 PetscScalar **arr; in Test2_1d() local
147 PetscCall(DMStagVecGetArray(dm, vecLocal, &arr)); in Test2_1d()
153 arr[i][idxLeft + c] = valRef; in Test2_1d()
158 arr[i][idxElement + c] = valRef; in Test2_1d()
162 PetscCall(DMStagVecRestoreArray(dm, vecLocal, &arr)); in Test2_1d()
168 PetscCall(DMStagVecGetArrayRead(dm, vecLocalCheck, &arr)); in Test2_1d()
172 const PetscScalar val = arr[i][idxLeft + c]; in Test2_1d()
178 const PetscScalar val = arr[i][idxElement + c]; in Test2_1d()
184 const PetscScalar val = arr[i][idxElement + c]; in Test2_1d()
189 PetscCall(DMStagVecRestoreArrayRead(dm, vecLocalCheck, &arr)); in Test2_1d()
[all …]
H A Dex19.c34 PetscScalar ***arr; in main() local
36 PetscCall(DMStagVecGetArray(dmCoarse, localCoarse, &arr)); in main()
42 arr[j][i][vy] = (i < startx + nx) ? 10.0 : -1; in main()
43 arr[j][i][vx] = (j < starty + ny) ? 20.0 : -1; in main()
44 arr[j][i][p] = (i < startx + nx) && (j < starty + ny) ? 30.0 : -1; in main()
47 PetscCall(DMStagVecRestoreArray(dmCoarse, localCoarse, &arr)); in main()
55 PetscScalar ***arr; in main() local
57 PetscCall(DMStagVecGetArrayRead(dm, localFine, &arr)); in main()
66 …if (arr[j][i][vy] != expected_vy) PetscCall(PetscPrintf(PETSC_COMM_SELF, "wrong %" PetscInt_FMT " … in main()
67 …if (arr[j][i][vx] != expected_vx) PetscCall(PetscPrintf(PETSC_COMM_SELF, "wrong %" PetscInt_FMT " … in main()
[all …]
H A Dex2.c37 PetscScalar *arr; in Test_3d_4x4x4_3x3x3() local
55 PetscCall(VecGetArray(vecGlobal, &arr)); in Test_3d_4x4x4_3x3x3()
56 for (i = 0; i < n; ++i) arr[i] = (PetscScalar)(i + low); in Test_3d_4x4x4_3x3x3()
57 PetscCall(VecRestoreArray(vecGlobal, &arr)); in Test_3d_4x4x4_3x3x3()
/petsc/src/vec/vec/tests/
H A Dex63.c5 static PetscErrorCode CheckExp(Vec v, PetscInt n, PetscScalar *arr, PetscScalar value) in CheckExp() argument
16 for (PetscInt i = 0; i < n; ++i) arr[i] = PetscExpScalar(value); in CheckExp()
20 const PetscScalar rhs = arr[i]; in CheckExp()
40 PetscScalar *arr; in main() local
50 PetscCall(PetscMalloc1(n, &arr)); in main()
52 PetscCall(CheckExp(v, n, arr, 0.0)); in main()
53 PetscCall(CheckExp(v, n, arr, 1.0)); in main()
54 PetscCall(CheckExp(v, n, arr, -1.0)); in main()
56 PetscCall(PetscFree(arr)); in main()
H A Dex64.c5 static PetscErrorCode CheckLog(Vec v, PetscInt n, PetscScalar *arr, PetscScalar value) in CheckLog() argument
16 for (PetscInt i = 0; i < n; ++i) arr[i] = PetscLogScalar(value); in CheckLog()
20 const PetscScalar rhs = arr[i]; in CheckLog()
40 PetscScalar *arr; in main() local
50 PetscCall(PetscMalloc1(n, &arr)); in main()
52 PetscCall(CheckLog(v, n, arr, 1.0)); in main()
53 PetscCall(CheckLog(v, n, arr, PetscExpScalar(1.0))); in main()
54 PetscCall(CheckLog(v, n, arr, 10.0)); in main()
56 PetscCall(PetscFree(arr)); in main()
/petsc/src/benchmarks/
H A DPetscMalloc.c8 void *arr[1000], *dummy; in main() local
29 for (i = 0; i < 1000; i++) PetscCall(PetscMalloc1(rand1[i], &arr[i])); in main()
34 for (i = 0; i < 1000; i += 2) PetscCall(PetscFree(arr[i])); in main()
37 for (i = 0; i < 1000; i += 2) PetscCall(PetscMalloc1(rand2[i], &arr[i])); in main()
40 for (i = 0; i < 1000; i++) PetscCall(PetscFree(arr[i])); in main()
/petsc/src/sys/utils/ftn-custom/
H A Dzsortsof.c25 PETSC_EXTERN void petsctimsort_(PetscInt *n, void *arr, size_t *size, void (*cmp)(const void *, con… in petsctimsort_() argument
28 *ierr = PetscTimSort(*n, arr, *size, cmp_via_fortran, &fc); in petsctimsort_()
31 PETSC_EXTERN void petsctimsortwitharray_(PetscInt *n, void *arr, size_t *asize, void *barr, size_t … in petsctimsortwitharray_() argument
34 *ierr = PetscTimSortWithArray(*n, arr, *asize, barr, *bsize, cmp_via_fortran, &fc); in petsctimsortwitharray_()
/petsc/src/dm/impls/plex/transform/impls/refine/sbr/
H A Dplexrefsbr.c338 const PetscInt *arr = DMPolytopeTypeGetArrangement(DM_POLYTOPE_TRIANGLE, o); in SBRGetTriangleSplitSingle() local
347 triC1[2] = arr[0 * 2]; in SBRGetTriangleSplitSingle()
348 triC1[7] = arr[1 * 2]; in SBRGetTriangleSplitSingle()
349 triC1[11] = arr[0 * 2]; in SBRGetTriangleSplitSingle()
350 triC1[15] = arr[1 * 2]; in SBRGetTriangleSplitSingle()
351 triC1[22] = arr[1 * 2]; in SBRGetTriangleSplitSingle()
352 triC1[26] = arr[2 * 2]; in SBRGetTriangleSplitSingle()
450 const PetscInt *arr = DMPolytopeTypeGetArrangement(DM_POLYTOPE_TRIANGLE, o); in SBRGetTriangleSplitDouble() local
458 triC2[2] = arr[0 * 2]; in SBRGetTriangleSplitDouble()
459 triC2[3] = arr[0 * 2 + 1] ? 1 : 0; in SBRGetTriangleSplitDouble()
[all …]
/petsc/src/ts/tutorials/
H A Dex20opt_ic.c285 PetscErrorCode Adjoint2(Vec U, PetscScalar arr[], User ctx) in Adjoint2() argument
326 arr[0] = x_ptr[0]; in Adjoint2()
327 arr[1] = x_ptr[1]; in Adjoint2()
335 PetscErrorCode FiniteDiff(Vec U, PetscScalar arr[], User ctx) in FiniteDiff() argument
358 arr[0] = u[0]; in FiniteDiff()
359 arr[1] = u[1]; in FiniteDiff()
370 arr[2] = u[0]; in FiniteDiff()
371 arr[3] = u[1]; in FiniteDiff()
401 PetscScalar *x_ptr, arr[4]; in main() local
484 PetscCall(FiniteDiff(x, arr, &user)); in main()
[all …]
/petsc/src/mat/tests/
H A Dex261.c10 const PetscScalar *arr; in CheckDiagonal() local
32 PetscCall(VecGetArrayRead(diag, &arr)); in CheckDiagonal()
34 const PetscScalar lhs = arr[i]; in CheckDiagonal()
46 PetscCall(VecRestoreArrayRead(diag, &arr)); in CheckDiagonal()
/petsc/src/dm/impls/stag/
H A Dstagutils.c42 void *arr[DMSTAG_MAX_DIM]; in DMStagGetProductCoordinateArrays_Private() local
49 arr[0] = arrX; in DMStagGetProductCoordinateArrays_Private()
50 arr[1] = arrY; in DMStagGetProductCoordinateArrays_Private()
51 arr[2] = arrZ; in DMStagGetProductCoordinateArrays_Private()
71 if (!arr[d]) continue; in DMStagGetProductCoordinateArrays_Private()
89 PetscCall(DMStagVecGetArrayRead(subDM, coord1d_local, arr[d])); in DMStagGetProductCoordinateArrays_Private()
91 PetscCall(DMStagVecGetArray(subDM, coord1d_local, arr[d])); in DMStagGetProductCoordinateArrays_Private()
814 const PetscScalar *arr; in DMStagMigrateVec() local
839 PetscCall(VecGetArrayRead(vecLocal, &arr)); in DMStagMigrateVec()
849 for (; d < b && dTo < bTo; ++d, ++dTo) arrTo[i * epeTo + dTo] = arr[i * epe + d]; in DMStagMigrateVec()
[all …]
H A Dstagstencil.c453 PetscScalar const *arr; in DMStagVecGetValuesStencil() local
462 PetscCall(VecGetArrayRead(vec, &arr)); in DMStagVecGetValuesStencil()
463 for (idx = 0; idx < n; ++idx) val[idx] = arr[ix[idx]]; in DMStagVecGetValuesStencil()
464 PetscCall(VecRestoreArrayRead(vec, &arr)); in DMStagVecGetValuesStencil()
H A Dstag1d.c94 PetscScalar **arr; in DMStagSetUniformCoordinatesExplicit_1d() local
107 PetscCall(DMStagVecGetArray(dmCoord, coordLocal, &arr)); in DMStagSetUniformCoordinatesExplicit_1d()
118 arr[ind][ileft] = min + ((PetscReal)ind + off) * h; in DMStagSetUniformCoordinatesExplicit_1d()
122 arr[ind][ielement] = min + ((PetscReal)ind + off) * h; in DMStagSetUniformCoordinatesExplicit_1d()
125 PetscCall(DMStagVecRestoreArray(dmCoord, coordLocal, &arr)); in DMStagSetUniformCoordinatesExplicit_1d()
H A Dstag2d.c133 PetscScalar ***arr; in DMStagSetUniformCoordinatesExplicit_2d() local
146 PetscCall(DMStagVecGetArray(dmCoord, coordLocal, &arr)); in DMStagSetUniformCoordinatesExplicit_2d()
164 …for (c = 0; c < 2; ++c) arr[ind[1]][ind[0]][idownleft + c] = min[c] + ((PetscReal)ind[c] + offs[c]… in DMStagSetUniformCoordinatesExplicit_2d()
168 …for (c = 0; c < 2; ++c) arr[ind[1]][ind[0]][idown + c] = min[c] + ((PetscReal)ind[c] + offs[c]) * … in DMStagSetUniformCoordinatesExplicit_2d()
172 …for (c = 0; c < 2; ++c) arr[ind[1]][ind[0]][ileft + c] = min[c] + ((PetscReal)ind[c] + offs[c]) * … in DMStagSetUniformCoordinatesExplicit_2d()
176 …for (c = 0; c < 2; ++c) arr[ind[1]][ind[0]][ielement + c] = min[c] + ((PetscReal)ind[c] + offs[c])… in DMStagSetUniformCoordinatesExplicit_2d()
180 PetscCall(DMStagVecRestoreArray(dmCoord, coordLocal, &arr)); in DMStagSetUniformCoordinatesExplicit_2d()
/petsc/src/vec/vec/utils/
H A Dvecstash.c405 PetscScalar *arr; in VecStashSortCompress_Private() local
406 PetscCall(PetscMalloc2(stash->n, &perm, stash->n * bs, &arr)); in VecStashSortCompress_Private()
411 PetscCall(PetscArraycpy(arr, stash->array + perm[0] * bs, bs)); in VecStashSortCompress_Private()
417 for (k = 0; k < bs; k++) arr[j * bs + k] += stash->array[perm[i] * bs + k]; in VecStashSortCompress_Private()
420 for (k = 0; k < bs; k++) arr[j * bs + k] = stash->array[perm[i] * bs + k]; in VecStashSortCompress_Private()
428 for (k = 0; k < bs; k++) arr[j * bs + k] = stash->array[perm[i] * bs + k]; in VecStashSortCompress_Private()
432 PetscCall(PetscArraycpy(stash->array, arr, stash->n * bs)); in VecStashSortCompress_Private()
433 PetscCall(PetscFree2(perm, arr)); in VecStashSortCompress_Private()
H A Dvecio.c122 PetscScalar *x, *arr; in VecLoad_HDF5() local
140 PetscCall(VecGetArray(xin, &arr)); in VecLoad_HDF5()
141 PetscCall(PetscArraycpy(arr, x, xin->map->n)); in VecLoad_HDF5()
143 PetscCall(VecRestoreArray(xin, &arr)); in VecLoad_HDF5()
/petsc/src/mat/impls/kaij/
H A Dkaij.c467 PetscCall(PetscFree5(b->sor.w, b->sor.y, b->sor.work, b->sor.t, b->sor.arr)); in MatDestroy_SeqKAIJ()
862 PetscScalar *x, *work, *workt, *w, *y, *arr, *t, *arrt; in MatSOR_SeqKAIJ() local
882 …>sor.w, bs, &kaij->sor.y, m * bs, &kaij->sor.work, m * bs, &kaij->sor.t, m * bs2, &kaij->sor.arr)); in MatSOR_SeqKAIJ()
889 arr = kaij->sor.arr; in MatSOR_SeqKAIJ()
951 arrt = arr; in MatSOR_SeqKAIJ()
957 PetscKernel_w_gets_w_minus_Ar_times_v(bs, bs * nz, w, arr, work); in MatSOR_SeqKAIJ()
990 arrt = arr; in MatSOR_SeqKAIJ()
997 PetscKernel_w_gets_w_minus_Ar_times_v(bs, bs * nz, w, arr, work); in MatSOR_SeqKAIJ()
1004 PetscKernel_w_gets_w_minus_Ar_times_v(bs, bs * nz, w, arr, work); in MatSOR_SeqKAIJ()
1016 arrt = arr; in MatSOR_SeqKAIJ()
[all …]
H A Dkaij.h14 PetscScalar *w, *work, *t, *arr, *y; \
/petsc/src/vec/is/utils/hdf5/
H A Dhdf5io.c225 …HDF5ReadArray_Private(PetscViewer viewer, HDF5ReadCtx h, hid_t datatype, hid_t memspace, void *arr) in PetscViewerHDF5ReadArray_Private() argument
230 PetscCallHDF5(H5Dread, (h->dataset, datatype, memspace, h->dataspace, hdf5->dxpl_id, arr)); in PetscViewerHDF5ReadArray_Private()
241 void *arr; in PetscViewerHDF5Load_Internal() local
283 PetscCall(PetscMalloc(map->n * unitsize, &arr)); in PetscViewerHDF5Load_Internal()
285 PetscCall(PetscViewerHDF5ReadArray_Private(viewer, h, datatype, memspace, arr)); in PetscViewerHDF5Load_Internal()
286 *newarr = arr; in PetscViewerHDF5Load_Internal()
/petsc/src/dm/impls/plex/tests/
H A Dex18.c922 const PetscInt *arr, *cone; in DMPlexExpandedConesToFaces_Private() local
930 PetscCall(ISGetIndices(is, &arr)); in DMPlexExpandedConesToFaces_Private()
935 cone = &arr[o]; in DMPlexExpandedConesToFaces_Private()
957 PetscCall(ISRestoreIndices(is, &arr)); in DMPlexExpandedConesToFaces_Private()
1261 const PetscInt *arr; in PointSFGetOutwardInterfacePoints() local
1264 PetscCall(PetscSFGetGraph(sf, NULL, &n, &arr, NULL)); in PointSFGetOutwardInterfacePoints()
1265 PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)sf), n, arr, PETSC_USE_POINTER, is)); in PointSFGetOutwardInterfacePoints()
1273 PetscInt *arr; in PointSFGetInwardInterfacePoints() local
1279 PetscCall(PetscSFComputeMultiRootOriginalNumbering(sf, rootdegree, &n, &arr)); in PointSFGetInwardInterfacePoints()
1280 PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)sf), n, arr, PETSC_OWN_POINTER, is)); in PointSFGetInwardInterfacePoints()
/petsc/src/dm/interface/
H A Ddmcoordinates.c671 const PetscScalar *arr; in DMGetCoordinatesLocalTuple() local
685 PetscCall(VecGetArrayRead(coords, &arr)); in DMGetCoordinatesLocalTuple()
686 …PetscCall(PetscSectionExtractDofsFromArray(cs, MPIU_SCALAR, arr, p, &newcs, pCoord ? ((void **)&ne… in DMGetCoordinatesLocalTuple()
687 PetscCall(VecRestoreArrayRead(coords, &arr)); in DMGetCoordinatesLocalTuple()
/petsc/src/dm/impls/plex/
H A Dplexorient.c21 const PetscInt *arr, *cone, *ornt, *support; in DMPlexOrientPoint() local
28 arr = DMPolytopeTypeGetArrangement(ct, o); in DMPlexOrientPoint()
29 if (!arr) PetscFunctionReturn(PETSC_SUCCESS); in DMPlexOrientPoint()
41 newcone[c] = cone[arr[c * 2 + 0]]; in DMPlexOrientPoint()
42 newornt[c] = DMPolytopeTypeComposeOrientation(ft, arr[c * 2 + 1], ornt[arr[c * 2 + 0]]); in DMPlexOrientPoint()
H A Dplex.c3323 PetscInt *arr = NULL, *newarr = NULL; in DMPlexGetConeRecursive() local
3339 arr = (PetscInt *)arr0; /* this is ok because first generation of arr is not modified */ in DMPlexGetConeRecursive()
3345 if (arr[i] >= start && arr[i] < end) { in DMPlexGetConeRecursive()
3346 PetscCall(DMPlexGetConeSize(dm, arr[i], &cn)); in DMPlexGetConeRecursive()
3359 PetscCall(DMPlexGetCone(dm, arr[i], &cone)); in DMPlexGetConeRecursive()
3362 newarr[co] = arr[i]; in DMPlexGetConeRecursive()
3366 arr = newarr; in DMPlexGetConeRecursive()
3968 …vate(DM dm, PetscInt p, PetscBool useCone, PetscInt *size, const PetscInt *arr[], const PetscInt *… in DMPlexGetTransitiveClosure_Hot_Private() argument
3976 PetscCall(DMPlexGetOrientedCone(dm, p, arr, ornt)); in DMPlexGetTransitiveClosure_Hot_Private()
3979 PetscCall(DMPlexGetSupport(dm, p, arr)); in DMPlexGetTransitiveClosure_Hot_Private()
[all …]
/petsc/src/mat/impls/aij/seq/mkl_pardiso/
H A Dmkl_pardiso.c290 const PetscScalar *arr; in MatFactorSetSchurIS_MKL_PARDISO() local
307 PetscCall(MatDenseGetArrayRead(F->schur, &arr)); in MatFactorSetSchurIS_MKL_PARDISO()
308 mpardiso->schur = (PetscScalar *)arr; in MatFactorSetSchurIS_MKL_PARDISO()
310 PetscCall(MatDenseRestoreArrayRead(F->schur, &arr)); in MatFactorSetSchurIS_MKL_PARDISO()

12