Home
last modified time | relevance | path

Searched refs:epe (Results 1 – 7 of 7) sorted by relevance

/petsc/src/dm/impls/stag/
H A Dstag1d.c451 const PetscInt epe = stag->entriesPerElement; in DMStagComputeLocationOffsets_1d() local
457 stag->locationOffsets[DMSTAG_RIGHT] = stag->locationOffsets[DMSTAG_LEFT] + epe; in DMStagComputeLocationOffsets_1d()
523 PetscInt dof[2], start, n, n_extra, stencil_width, N, epe; in DMCreateMatrix_Stag_1D_AIJ_Assemble() local
532 PetscCall(DMStagGetEntriesPerElement(dm, &epe)); in DMCreateMatrix_Stag_1D_AIJ_Assemble()
565 PetscCall(PetscMalloc1(epe, &row)); in DMCreateMatrix_Stag_1D_AIJ_Assemble()
579 PetscCall(PetscMalloc1(epe, &col)); in DMCreateMatrix_Stag_1D_AIJ_Assemble()
594 for (PetscInt i = 0; i < epe; ++i) row[i].i = e; in DMCreateMatrix_Stag_1D_AIJ_Assemble()
604 for (PetscInt i = 0; i < epe; ++i) col[i].i = e_offset; in DMCreateMatrix_Stag_1D_AIJ_Assemble()
605 PetscCall(DMStagMatSetValuesStencil(dm, A, epe, row, epe, col, NULL, INSERT_VALUES)); in DMCreateMatrix_Stag_1D_AIJ_Assemble()
H A Dstagstencil.c388 const PetscInt epe = stag->entriesPerElement; in DMStagStencilToIndexLocal() local
395 ix[idx] = eLocal * epe + stag->locationOffsets[pos[idx].loc] + pos[idx].c; in DMStagStencilToIndexLocal()
405 ix[idx] = eLocal * epe + stag->locationOffsets[pos[idx].loc] + pos[idx].c; in DMStagStencilToIndexLocal()
417 ix[idx] = eLocal * epe + stag->locationOffsets[pos[idx].loc] + pos[idx].c; in DMStagStencilToIndexLocal()
H A Dstag2d.c1145 const PetscInt epe = stag->entriesPerElement; in DMStagComputeLocationOffsets_2d() local
1146 const PetscInt epr = stag->nGhost[0] * epe; in DMStagComputeLocationOffsets_2d()
1152 stag->locationOffsets[DMSTAG_DOWN_RIGHT] = stag->locationOffsets[DMSTAG_DOWN_LEFT] + epe; in DMStagComputeLocationOffsets_2d()
1155 stag->locationOffsets[DMSTAG_RIGHT] = stag->locationOffsets[DMSTAG_LEFT] + epe; in DMStagComputeLocationOffsets_2d()
1158 stag->locationOffsets[DMSTAG_UP_RIGHT] = stag->locationOffsets[DMSTAG_UP_LEFT] + epe; in DMStagComputeLocationOffsets_2d()
1303 …PetscInt entries, dof[DMSTAG_MAX_STRATA], epe, stencil_width, N[2], start[2], n[2], n_ext… in DMCreateMatrix_Stag_2D_AIJ_Assemble() local
1312 PetscCall(DMStagGetEntriesPerElement(dm, &epe)); in DMCreateMatrix_Stag_2D_AIJ_Assemble()
1384 PetscCall(PetscMalloc1(epe, &row)); in DMCreateMatrix_Stag_2D_AIJ_Assemble()
1410 PetscCall(PetscMalloc1(epe, &col)); in DMCreateMatrix_Stag_2D_AIJ_Assemble()
1438 for (PetscInt i = 0; i < epe; ++i) { in DMCreateMatrix_Stag_2D_AIJ_Assemble()
[all …]
H A Dstagutils.c816 const PetscInt epe = stag->entriesPerElement; in DMStagMigrateVec() local
849 for (; d < b && dTo < bTo; ++d, ++dTo) arrTo[i * epeTo + dTo] = arr[i * epe + d]; in DMStagMigrateVec()
855 const PetscInt epr = stag->nGhost[0] * epe; in DMStagMigrateVec()
859 const PetscInt base = j * epr + i * epe; in DMStagMigrateVec()
874 const PetscInt epr = stag->nGhost[0] * epe; in DMStagMigrateVec()
882 const PetscInt base = k * epl + j * epr + i * epe; in DMStagMigrateVec()
H A Dstag3d.c3194 const PetscInt epe = stag->entriesPerElement; in DMStagComputeLocationOffsets_3d() local
3195 const PetscInt epr = stag->nGhost[0] * epe; in DMStagComputeLocationOffsets_3d()
3202 …tag->locationOffsets[DMSTAG_BACK_DOWN_RIGHT] = stag->locationOffsets[DMSTAG_BACK_DOWN_LEFT] + epe; in DMStagComputeLocationOffsets_3d()
3205 stag->locationOffsets[DMSTAG_BACK_RIGHT] = stag->locationOffsets[DMSTAG_BACK_LEFT] + epe; in DMStagComputeLocationOffsets_3d()
3208 stag->locationOffsets[DMSTAG_BACK_UP_RIGHT] = stag->locationOffsets[DMSTAG_BACK_UP_LEFT] + epe; in DMStagComputeLocationOffsets_3d()
3211 stag->locationOffsets[DMSTAG_DOWN_RIGHT] = stag->locationOffsets[DMSTAG_DOWN_LEFT] + epe; in DMStagComputeLocationOffsets_3d()
3214 stag->locationOffsets[DMSTAG_RIGHT] = stag->locationOffsets[DMSTAG_LEFT] + epe; in DMStagComputeLocationOffsets_3d()
3217 stag->locationOffsets[DMSTAG_UP_RIGHT] = stag->locationOffsets[DMSTAG_UP_LEFT] + epe; in DMStagComputeLocationOffsets_3d()
3220 …ag->locationOffsets[DMSTAG_FRONT_DOWN_RIGHT] = stag->locationOffsets[DMSTAG_FRONT_DOWN_LEFT] + epe; in DMStagComputeLocationOffsets_3d()
3223 stag->locationOffsets[DMSTAG_FRONT_RIGHT] = stag->locationOffsets[DMSTAG_FRONT_LEFT] + epe; in DMStagComputeLocationOffsets_3d()
[all …]
/petsc/src/dm/impls/stag/tests/
H A Dex40.c585 PetscInt epe; in FormFunction2D() local
588 PetscCall(DMStagGetEntriesPerElement(dm, &epe)); in FormFunction2D()
589 PetscCall(PetscMalloc1(epe, &row)); in FormFunction2D()
590 PetscCall(PetscMalloc1(epe, &col)); in FormFunction2D()
591 for (PetscInt i = 0; i < epe; ++i) { in FormFunction2D()
647 for (PetscInt i = 0; i < epe; ++i) { in FormFunction2D()
648 for (PetscInt j = 0; j < epe; ++j) { in FormFunction2D()
746 PetscInt epe; in FormJacobian2D() local
749 PetscCall(DMStagGetEntriesPerElement(dm, &epe)); in FormJacobian2D()
750 PetscCall(PetscMalloc1(epe, &row)); in FormJacobian2D()
[all …]
/petsc/src/binding/petsc4py/src/petsc4py/PETSc/
H A DDMStag.pyx330 cdef PetscInt epe=0
331 CHKERR(DMStagGetEntriesPerElement(self.dm, &epe))
332 return toInt(epe)