Home
last modified time | relevance | path

Searched refs:zm (Results 1 – 25 of 33) sorted by relevance

12

/petsc/src/dm/impls/da/
H A Ddagetarray.c45 PetscInt xs, ys, zs, xm, ym, zm, gxs, gys, gzs, gxm, gym, gzm, N, dim, dof; in DMDAVecGetArray() local
51 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); in DMDAVecGetArray()
57 if (N == xm * ym * zm * dof) { in DMDAVecGetArray()
60 gzm = zm; in DMDAVecGetArray()
64 …ible with DMDA local sizes %" PetscInt_FMT " %" PetscInt_FMT, N, xm * ym * zm * dof, gxm * gym * g… in DMDAVecGetArray()
94 PetscInt xs, ys, zs, xm, ym, zm, gxs, gys, gzs, gxm, gym, gzm, N, dim, dof; in DMDAVecRestoreArray() local
100 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); in DMDAVecRestoreArray()
106 if (N == xm * ym * zm * dof) { in DMDAVecRestoreArray()
109 gzm = zm; in DMDAVecRestoreArray()
113 …ible with DMDA local sizes %" PetscInt_FMT " %" PetscInt_FMT, N, xm * ym * zm * dof, gxm * gym * g… in DMDAVecRestoreArray()
[all …]
H A Dgrvtk.c80 rloc[5] = info.zm; in DMDAVTKWriteAll_VTS()
88 PetscInt xs = -1, xm = -1, ys = -1, ym = -1, zs = -1, zm = -1, nnodes = 0; in DMDAVTKWriteAll_VTS() local
96 zm = grloc[r][5]; in DMDAVTKWriteAll_VTS()
97 nnodes = xm * ym * zm; in DMDAVTKWriteAll_VTS()
100 …T " %" PetscInt_FMT " %" PetscInt_FMT "\">\n", xs, xs + xm - 1, ys, ys + ym - 1, zs, zs + zm - 1)); in DMDAVTKWriteAll_VTS()
151 PetscInt xs = -1, xm = -1, ys = -1, ym = -1, zs = -1, zm = -1, nnodes = 0; in DMDAVTKWriteAll_VTS() local
159 zm = grloc[r][5]; in DMDAVTKWriteAll_VTS()
160 nnodes = xm * ym * zm; in DMDAVTKWriteAll_VTS()
162 nnodes = info.xm * info.ym * info.zm; in DMDAVTKWriteAll_VTS()
179 for (k = 0; k < zm; k++) { in DMDAVTKWriteAll_VTS()
[all …]
H A Ddadd.c43 PetscInt xm = 1, ym = 1, zm = 1; in DMDACreatePatchIS() local
179 zm = pe - ps; in DMDACreatePatchIS()
180 base = ms * ym * zm + ns * M * zm + ps * M * N; in DMDACreatePatchIS()
213 zm = pe - ps; in DMDACreatePatchIS()
214 base = ms * ym * zm + ns * M * zm + ps * M * N; in DMDACreatePatchIS()
273 PetscInt xm = 1, ym = 1, zm = 1; in DMDASubDomainDA_Private() local
291 P = info.zm; in DMDASubDomainDA_Private()
340 zm = P / p + ((P % p) > k); in DMDASubDomainDA_Private()
345 zsize = zm; in DMDASubDomainDA_Private()
373 if (info.bz == DM_BOUNDARY_PERIODIC || (zs + zm != info.mz)) zsize += zol; in DMDASubDomainDA_Private()
[all …]
H A Ddalocal.c410 PetscInt j, i, xs, ys, xm, ym, zs, zm; in DMDAGetArray() local
433 zm = dd->Ze - dd->Zs; in DMDAGetArray()
450 zm = dd->ze - dd->zs; in DMDAGetArray()
476 …PetscCall(PetscMalloc((zm + 1) * sizeof(void **) + (ym * zm + 1) * sizeof(void *) + xm * ym * zm *… in DMDAGetArray()
478 ptr = (void ***)(iarray_start + xm * ym * zm * sizeof(PetscScalar) - zs * sizeof(void *)); in DMDAGetArray()
479 bptr = (void **)(iarray_start + xm * ym * zm * sizeof(PetscScalar) + zm * sizeof(void **)); in DMDAGetArray()
480 for (i = zs; i < zs + zm; i++) ptr[i] = bptr + ((i - zs) * ym - ys); in DMDAGetArray()
481 for (i = zs; i < zs + zm; i++) { in DMDAGetArray()
H A Ddasub.c203 PetscInt flag, *owners, xs, xm, ys, ym, zs, zm; in DMDAGetProcessorSubset() local
210 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); in DMDAGetProcessorSubset()
215 if (gp >= zs && gp < zs + zm) flag = 1; in DMDAGetProcessorSubset()
276 PetscInt xs, xm, ys, ym, zs, zm, firstPoint; in DMDAGetProcessorSubsets() local
281 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); in DMDAGetProcessorSubsets()
/petsc/src/dm/impls/da/ftn-custom/
H A Dzda1f90.c68 PetscInt xs, ys, zs, xm, ym, zm, gxs, gys, gzs, gxm, gym, gzm, N, dim, dof; in dmdavecgetarray1_() local
71 *ierr = DMDAGetCorners(*da, &xs, &ys, &zs, &xm, &ym, &zm); in dmdavecgetarray1_()
81 if (N == xm * ym * zm * dof) { in dmdavecgetarray1_()
84 gzm = zm; in dmdavecgetarray1_()
109 PetscInt xs, ys, zs, xm, ym, zm, gxs, gys, gzs, gxm, gym, gzm, N, dim, dof; in dmdavecgetarray2_() local
112 *ierr = DMDAGetCorners(*da, &xs, &ys, &zs, &xm, &ym, &zm); in dmdavecgetarray2_()
122 if (N == xm * ym * zm * dof) { in dmdavecgetarray2_()
125 gzm = zm; in dmdavecgetarray2_()
156 PetscInt xs, ys, zs, xm, ym, zm, gxs, gys, gzs, gxm, gym, gzm, N, dim, dof; in dmdavecgetarray3_() local
159 *ierr = DMDAGetCorners(*da, &xs, &ys, &zs, &xm, &ym, &zm); in dmdavecgetarray3_()
[all …]
/petsc/src/dm/tests/
H A Dex2k.kokkos.cxx36 PetscInt xm, ym, zm, xs, ys, zs, gxm, gym, gzm, gxs, gys, gzs; in Update1() local
39 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); in Update1()
43 for (k = zs; k < zs + zm; k++) { in Update1()
59 PetscInt xm, ym, zm, xs, ys, zs, gxm, gym, gzm, gxs, gys, gzs; in Update2() local
62 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); in Update2()
68 for (k = zs; k < zs + zm; k++) { in Update2()
84 PetscInt xm, ym, zm, xs, ys, zs, gxm, gym, gzm, gxs, gys, gzs; in main() local
111 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); in main()
146 …xecutionSpace, Rank<3, Iterate::Right, Iterate::Right>>({zs, ys, xs}, {zs + zm, ys + ym, xs + xm}), in main()
H A Dex47.c30 for (k = info.zs; k < info.zs + info.zm; k++) { in test_3d()
149 for (k = info.zs; k < info.zs + info.zm; k++) { in test_3d_nocoord()
H A Dex42.c32 for (k = info.zs; k < info.zs + info.zm; k++) { in test_3d()
151 for (k = info.zs; k < info.zs + info.zm; k++) { in test_3d_nocoord()
H A Dex48.c45 for (k = info.zs; k < info.zs + info.zm; k++) { in test_3d()
131 for (k = info.zs; k < info.zs + info.zm; k++) { in test_3d_compat()
/petsc/src/dm/impls/da/kokkos/
H A Ddagetov.kokkos.cxx8 #define DMDA_VEC_GET_SHAPE(da, vec, xs, ys, zs, xm, ym, zm, gxs, gys, gzs, gxm, gym, gzm, N, dim, d… argument
10 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); \
15 if (N == xm * ym * zm * dof) { \
18 gzm = zm; \
22 …ible with DMDA local sizes %" PetscInt_FMT " %" PetscInt_FMT, N, xm * ym * zm * dof, gxm * gym * g…
29 …PetscInt xs, ys, zs, xm, ym, zm, gxs, gys, gzs, gxm, gym, gzm, N, di… in DMDAVecGetKokkosOffsetView_Private() local
36 DMDA_VEC_GET_SHAPE(da, vec, xs, ys, zs, xm, ym, zm, gxs, gys, gzs, gxm, gym, gzm, N, dim, dof); in DMDAVecGetKokkosOffsetView_Private()
63 …PetscInt xs, ys, zs, xm, ym, zm, gxs, gys, gzs, gxm, gym, gzm, … in DMDAVecGetKokkosOffsetView() local
70 DMDA_VEC_GET_SHAPE(da, vec, xs, ys, zs, xm, ym, zm, gxs, gys, gzs, gxm, gym, gzm, N, dim, dof); in DMDAVecGetKokkosOffsetView()
95 …PetscInt xs, ys, zs, xm, ym, zm, gxs, gys, gzs, gxm, gym, gzm, N, di… in DMDAVecGetKokkosOffsetView_Private() local
[all …]
/petsc/src/binding/petsc4py/demo/legacy/wrap-swig/
H A DBratu3D.c24 PetscInt i,j,k,Mx,My,Mz,xs,ys,zs,xm,ym,zm; in FormInitGuess() local
55 PetscCall(DMDAGetCorners(da,&xs,&ys,&zs,&xm,&ym,&zm)); in FormInitGuess()
60 for (k=zs; k<zs+zm; k++) { in FormInitGuess()
84 PetscInt i,j,k,Mx,My,Mz,xs,ys,zs,xm,ym,zm; in FormFunction() local
120 PetscCall(DMDAGetCorners(da,&xs,&ys,&zs,&xm,&ym,&zm)); in FormFunction()
125 for (k=zs; k<zs+zm; k++) { in FormFunction()
161 PetscInt i,j,k,Mx,My,Mz,xs,ys,zs,xm,ym,zm; in FormJacobian() local
197 PetscCall(DMDAGetCorners(da,&xs,&ys,&zs,&xm,&ym,&zm)); in FormJacobian()
214 for (k=zs; k<zs+zm; k++) { in FormJacobian()
/petsc/src/binding/petsc4py/demo/legacy/wrap-cython/
H A DBratu3Dimpl.c24 PetscInt i,j,k,Mx,My,Mz,xs,ys,zs,xm,ym,zm; in FormInitGuess() local
56 PetscCall(DMDAGetCorners(da,&xs,&ys,&zs,&xm,&ym,&zm)); in FormInitGuess()
61 for (k=zs; k<zs+zm; k++) { in FormInitGuess()
85 PetscInt i,j,k,Mx,My,Mz,xs,ys,zs,xm,ym,zm; in FormFunction() local
124 PetscCall(DMDAGetCorners(da,&xs,&ys,&zs,&xm,&ym,&zm)); in FormFunction()
129 for (k=zs; k<zs+zm; k++) { in FormFunction()
165 PetscInt i,j,k,Mx,My,Mz,xs,ys,zs,xm,ym,zm; in FormJacobian() local
204 PetscCall(DMDAGetCorners(da,&xs,&ys,&zs,&xm,&ym,&zm)); in FormJacobian()
221 for (k=zs; k<zs+zm; k++) { in FormJacobian()
/petsc/src/snes/tutorials/
H A Dex16.c602 PetscInt xm = info->xm, ym = info->ym, zm = info->zm; in FormJacobianLocal() local
623 zee = zs + zm; in FormJacobianLocal()
629 if (zs + zm == mz) zee = zs + zm - 1; in FormJacobianLocal()
646 …= xs && i + ii < xm + xs && j + jj >= ys && j + jj < ys + ym && k + kk >= zs && k + kk < zs + zm) { in FormJacobianLocal()
676 for (k = zs; k < zs + zm; k++) { in FormJacobianLocal()
711 PetscInt xm = info->xm, ym = info->ym, zm = info->zm; in FormFunctionLocal() local
723 PetscCall(DMDAGetCorners(info->da, &xs, &ys, &zs, &xm, &ym, &zm)); in FormFunctionLocal()
726 for (k = zs; k < zs + zm; k++) { in FormFunctionLocal()
739 zee = zs + zm; in FormFunctionLocal()
745 if (zs + zm == mz) zee = zs + zm - 1; in FormFunctionLocal()
[all …]
H A Dex14.c204 PetscInt i, j, k, Mx, My, Mz, xs, ys, zs, xm, ym, zm; in FormInitialGuess() local
232 PetscCall(DMDAGetCorners(user->da, &xs, &ys, &zs, &xm, &ym, &zm)); in FormInitialGuess()
237 for (k = zs; k < zs + zm; k++) { in FormInitialGuess()
273 PetscInt i, j, k, Mx, My, Mz, xs, ys, zs, xm, ym, zm; in FormFunctionLocal() local
300 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); in FormFunctionLocal()
305 for (k = zs; k < zs + zm; k++) { in FormFunctionLocal()
389 PetscInt xs, ys, zs, xm, ym, zm; in FormJacobian() local
424 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); in FormJacobian()
437 for (k = zs; k < zs + zm; k++) { in FormJacobian()
H A Dex48.c290 static void PrmHexGetZ(const PrmNode pn[], PetscInt k, PetscInt zm, PetscReal zn[]) in PrmHexGetZ() argument
292 …const PetscScalar zm1 = zm - 1, znl[8] = {pn[0].b + pn[0].h * (PetscScalar)k / zm1, pn[1].b … in PrmHexGetZ()
681 PetscInt i, j, k, xs, xm, ys, ym, zs, zm, mx, my; in THIInitial() local
691 PetscCall(DMDAGetCorners(da, &zs, &ys, &xs, &zm, &ym, &xm)); in THIInitial()
698 for (k = zs; k < zs + zm; k++) { in THIInitial()
699 …const PetscScalar zm1 = zm - 1, drivingx = thi->rhog * (prm[i + 1][j].b + prm[i + 1][j].h - prm[i … in THIInitial()
754 PetscInt xs, ys, xm, ym, zm, i, j, k, q, l; in THIFunctionLocal() local
761 xm = info->zm; in THIFunctionLocal()
763 zm = info->xm; in THIFunctionLocal()
778 for (k = 0; k < zm - 1; k++) { in THIFunctionLocal()
[all …]
/petsc/src/ksp/ksp/tutorials/
H A Dex45.c76 PetscInt i, j, k, mx, my, mz, xm, ym, zm, xs, ys, zs; in ComputeRHS() local
91 PetscCall(DMDAGetCorners(dm, &xs, &ys, &zs, &xm, &ym, &zm)); in ComputeRHS()
94 for (k = zs; k < zs + zm; k++) { in ComputeRHS()
120 PetscInt i, j, k, mx, my, mz, xm, ym, zm, xs, ys, zs; in ComputeMatrix() local
134 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); in ComputeMatrix()
136 for (k = zs; k < zs + zm; k++) { in ComputeMatrix()
H A Dex34.c31 PetscInt i, j, k, mx, my, mz, xm, ym, zm, xs, ys, zs, d, dof; in main() local
67 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); in main()
70 for (k = zs; k < zs + zm; k++) { in main()
97 PetscInt d, dof, i, j, k, mx, my, mz, xm, ym, zm, xs, ys, zs; in ComputeRHS() local
109 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); in ComputeRHS()
111 for (k = zs; k < zs + zm; k++) { in ComputeRHS()
135 PetscInt dof, i, j, k, d, mx, my, mz, xm, ym, zm, xs, ys, zs, num, numi, numj, numk; in ComputeMatrix() local
151 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); in ComputeMatrix()
152 for (k = zs; k < zs + zm; k++) { in ComputeMatrix()
H A Dex22f.F9043 PetscInt ym, zm, xs, ys, zs, i1, i7
60 PetscCall(DMDAGetCorners(da, xs, ys, zs, xm, ym, zm, ierr))
62 do k = zs, zs + zm - 1
/petsc/src/ts/tutorials/
H A Dex14.c276 static void PrmHexGetZ(const PrmNode pn[], PetscInt k, PetscInt zm, PetscReal zn[]) in PrmHexGetZ() argument
278 …const PetscScalar zm1 = zm - 1, znl[8] = {pn[0].b + pn[0].h * (PetscScalar)k / zm1, pn[1].b … in PrmHexGetZ()
688 PetscInt i, j, k, xs, xm, ys, ym, zs, zm, mx, my; in THIInitial() local
700 PetscCall(DMDAGetCorners(da3, &zs, &ys, &xs, &zm, &ym, &xm)); in THIInitial()
710 for (k = zs; k < zs + zm; k++) { in THIInitial()
711 …const PetscScalar zm1 = zm - 1, drivingx = thi->rhog * (prm[i + 1][j].b + prm[i + 1][j].h - prm[i … in THIInitial()
752 PetscInt xs, ys, xm, ym, zm, i, j, k, q, l; in THIFunctionLocal_3D() local
758 xm = info->zm; in THIFunctionLocal_3D()
760 zm = info->xm; in THIFunctionLocal_3D()
774 for (k = 0; k < zm - 1; k++) { in THIFunctionLocal_3D()
[all …]
/petsc/include/
H A Dpetscdmdatypes.h63 PetscInt xm, ym, zm; /* number of grid points on this processor, excluding ghosts */ member
/petsc/src/binding/petsc4py/src/petsc4py/PETSc/
H A Dpetscdmda.pxi224 cdef PetscInt xs, ys, zs, xm, ym, zm
227 xm, ym, zm = lxm, lym, lzm
230 xm, ym, zm = gxm, gym, gzm
238 cdef tuple sizes = toDims(dim, xm, ym, zm)
242 cdef tuple shape = toDims(dim, xm, ym, zm)
/petsc/src/ksp/ksp/tests/
H A Dex32.c127 PetscInt i, j, k, mx, my, mz, xm, ym, zm, xs, ys, zs, dof, k1, k2, k3; in ComputeMatrix() local
159 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); in ComputeMatrix()
161 for (k = zs; k < zs + zm; k++) { in ComputeMatrix()
/petsc/src/mat/tests/
H A Dex129.c185 PetscInt i, j, k, mx, my, mz, xm, ym, zm, xs, ys, zs, dof, k1, k2, k3; in ComputeMatrix() local
226 PetscCall(DMDAGetCorners(da, &xs, &ys, &zs, &xm, &ym, &zm)); in ComputeMatrix()
228 for (k = zs; k < zs + zm; k++) { in ComputeMatrix()
/petsc/src/dm/tutorials/
H A Dex14.c28 for (k = info.zs; k < info.zs + info.zm; k++) { in FillLocalSubdomain()
107 upper.k = info.zs + info.zm; in main()

12