| /petsc/src/mat/tutorials/ |
| H A D | ex18.c | 23 fe->Ne = 3; in CreateFEStruct()
24 PetscCall(PetscMalloc1(3 * fe->Ne, &fe->vertices)); in CreateFEStruct()
58 PetscCall(PetscMalloc2(3 * 3 * fe->Ne, &oor, 3 * 3 * fe->Ne, &ooc)); in CreateMatrix()
59 for (PetscInt e = 0; e < fe->Ne; e++) { in CreateMatrix()
67 PetscCall(MatSetPreallocationCOO(*A, 3 * 3 * fe->Ne, oor, ooc)); in CreateMatrix()
72 PetscCall(PetscMalloc1(fe->Ne, &fe->coo)); in CreateMatrix()
74 for (PetscInt e = 1; e < fe->Ne; e++) fe->coo[e] = fe->coo[e - 1] + 3 * 3; in CreateMatrix()
84 for (PetscInt e = 0; e < fe->Ne; e++) { in FillMatrixCPU()
105 PetscCall(PetscMalloc1(3 * 3 * fe->Ne, &v)); in FillMatrixCPUCOO()
106 for (PetscInt e = 0; e < fe->Ne; e++) { in FillMatrixCPUCOO()
[all …]
|
| H A D | ex18cu.cu | 21 PetscCallCUDA(cudaMalloc((void **)&v, 3 * 3 * fe->Ne * sizeof(PetscScalar))); in FillMatrixCUDACOO()
22 FillValues<<<(fe->Ne + 255) / 256, 256>>>(fe->Ne, v); in FillMatrixCUDACOO()
|
| H A D | ex18k.kokkos.cxx | 9 Kokkos::View<PetscScalar *, DefaultMemorySpace> v("v", 3 * 3 * fe->Ne); in FillMatrixKokkosCOO()
14 "AssembleElementMatrices", fe->Ne, KOKKOS_LAMBDA(PetscInt i) { in FillMatrixKokkosCOO()
|
| H A D | ex18.h | 7 PetscInt Ne; /* number of elements */ member
|
| /petsc/src/dm/dt/fe/impls/opencl/ |
| H A D | feopencl.c | 502 static PetscErrorCode PetscFEIntegrateResidual_OpenCL(PetscDS prob, PetscFormKey key, PetscInt Ne, … in PetscFEIntegrateResidual_OpenCL() argument
547 if (!Ne) { in PetscFEIntegrateResidual_OpenCL()
564 if (Ne % (N_cb * N_bc)) { /* Remainder cells */ in PetscFEIntegrateResidual_OpenCL()
565 …PetscCall(PetscFEIntegrateResidual_Basic(prob, key, Ne, cgeom, coefficients, coefficients_t, probA… in PetscFEIntegrateResidual_OpenCL()
568 PetscCall(PetscFEOpenCLCalculateGrid(fem, Ne, N_cb * N_bc, &x, &y, &z)); in PetscFEIntegrateResidual_OpenCL()
601 …PetscCall(PetscMalloc4(Ne * N_bt, &f_coeff, Ne, &f_coeffAux, Ne * dim * dim, &f_invJ, Ne, &f_detJ)… in PetscFEIntegrateResidual_OpenCL()
602 for (c = 0; c < Ne; ++c) { in PetscFEIntegrateResidual_OpenCL()
608 for (c = 0; c < Ne; ++c) f_coeffAux[c] = (float)coefficientsAux[c]; in PetscFEIntegrateResidual_OpenCL()
622 …PetscCall(PetscMalloc4(Ne * N_bt, &d_coeff, Ne, &d_coeffAux, Ne * dim * dim, &d_invJ, Ne, &d_detJ)… in PetscFEIntegrateResidual_OpenCL()
623 for (c = 0; c < Ne; ++c) { in PetscFEIntegrateResidual_OpenCL()
[all …]
|
| /petsc/src/dm/tests/ |
| H A D | ex10.c | 19 PetscInt Ne, Ni; in main() local
35 Ne = 2; in main()
37 …nodeOffset = (Ne + Ni) * rank; /* The global node index of the first node defined on this process … in main()
|
| /petsc/src/dm/impls/plex/ |
| H A D | plexegads.c | 632 int id, sh, Nsh, f, Nf, l, Nl, e, Ne, v, Nv; in DMPlexGeomPrintModel_Internal() local
661 PetscCall(EGlite_getBodyTopos(body, NULL, EDGE, &Ne, &eobjs)); in DMPlexGeomPrintModel_Internal()
663 PetscCall(EG_getBodyTopos(body, NULL, EDGE, &Ne, &eobjs)); in DMPlexGeomPrintModel_Internal()
665 PetscCall(PetscPrintf(PETSC_COMM_SELF, " Number of EDGES: %d \n", Ne)); in DMPlexGeomPrintModel_Internal()
767 … PetscCall(EGlite_getTopology(loop, &geom, &oclass, &mtype, NULL, &Ne, &eobjs, &esenses)); in DMPlexGeomPrintModel_Internal()
769 PetscCall(EG_getTopology(loop, &geom, &oclass, &mtype, NULL, &Ne, &eobjs, &esenses)); in DMPlexGeomPrintModel_Internal()
772 for (e = 0; e < Ne; ++e) { in DMPlexGeomPrintModel_Internal()
920 int Ner = 0, Ne, e, Nc; in DMPlexCreateGeom_Internal() local
923 PetscCall(EGlite_getTopology(loop, &geom, &oclass, &mtype, NULL, &Ne, &objs, &senses)); in DMPlexCreateGeom_Internal()
925 PetscCall(EG_getTopology(loop, &geom, &oclass, &mtype, NULL, &Ne, &objs, &senses)); in DMPlexCreateGeom_Internal()
[all …]
|
| H A D | plexfem.c | 2520 PetscInt numChunks, numBatches, batchSize, numBlocks, blockSize, Ne, Nr, offset; in DMPlexComputeIntegral_Internal() local
2538 Ne = numChunks * numBatches * batchSize; in DMPlexComputeIntegral_Internal()
2543 PetscCall(PetscFEIntegrate(prob, f, Ne, chunkGeom, u, probAux, a, cintegral)); in DMPlexComputeIntegral_Internal()
4496 PetscInt numChunks, numBatches, batchSize, numBlocks, blockSize, Ne, Nr, offset; in DMPlexComputeResidual_Patch_Internal() local
4517 Ne = numChunks * numBatches * batchSize; in DMPlexComputeResidual_Patch_Internal()
4523 … PetscCall(PetscFEIntegrateResidual(prob, key, Ne, chunkGeom, u, u_t, probAux, a, t, elemVec)); in DMPlexComputeResidual_Patch_Internal()
4530 Ne = numFaces; in DMPlexComputeResidual_Patch_Internal()
4533 PetscCall(PetscFVIntegrateRHSFunction(fv, prob, f, Ne, fgeom, vol, uL, uR, fluxL, fluxR)); in DMPlexComputeResidual_Patch_Internal()
5007 PetscInt numChunks, numBatches, numBlocks, Ne, blockSize, batchSize; in DMPlexComputeBdResidualSingleByKey() local
5019 Ne = numChunks * numBatches * batchSize; in DMPlexComputeBdResidualSingleByKey()
[all …]
|
| /petsc/src/dm/dt/fe/impls/basic/ |
| H A D | febasic.c | 162 PETSC_INTERN PetscErrorCode PetscFEIntegrate_Basic(PetscDS ds, PetscInt field, PetscInt Ne, PetscFE… in PetscFEIntegrate_Basic() argument
208 for (e = 0; e < Ne; ++e) { in PetscFEIntegrate_Basic()
255 …ateBd_Basic(PetscDS ds, PetscInt field, PetscBdPointFn *obj_func, PetscInt Ne, PetscFEGeom *fgeom,… in PetscFEIntegrateBd_Basic() argument
299 …SELF, "Field: %" PetscInt_FMT " Nface: %" PetscInt_FMT " Nq: %" PetscInt_FMT "\n", field, Ne, Nq)); in PetscFEIntegrateBd_Basic()
303 for (e = 0; e < Ne; ++e) { in PetscFEIntegrateBd_Basic()
389 PetscErrorCode PetscFEIntegrateResidual_Basic(PetscDS ds, PetscFormKey key, PetscInt Ne, PetscFEGeo… in PetscFEIntegrateResidual_Basic() argument
439 for (e = 0; e < Ne; ++e) { in PetscFEIntegrateResidual_Basic()
496 …eBdResidual_Basic(PetscDS ds, PetscWeakForm wf, PetscFormKey key, PetscInt Ne, PetscFEGeom *fgeom,… in PetscFEIntegrateBdResidual_Basic() argument
551 for (e = 0; e < Ne; ++e) { in PetscFEIntegrateBdResidual_Basic()
609 …ual_Basic(PetscDS ds, PetscDS dsIn, PetscFormKey key, PetscInt s, PetscInt Ne, PetscFEGeom *fgeom,… in PetscFEIntegrateHybridResidual_Basic() argument
[all …]
|
| /petsc/src/dm/dt/fe/interface/ |
| H A D | fe.c | 1399 PetscErrorCode PetscFEIntegrate(PetscDS prob, PetscInt field, PetscInt Ne, PetscFEGeom *cgeom, cons… in PetscFEIntegrate() argument
1406 …if (fe->ops->integrate) PetscCall((*fe->ops->integrate)(prob, field, Ne, cgeom, coefficients, prob… in PetscFEIntegrate()
1435 …const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]), PetscInt Ne, PetscFEGeom *geom, … in PetscFEIntegrateBd() argument
1442 …if (fe->ops->integratebd) PetscCall((*fe->ops->integratebd)(prob, field, obj_func, Ne, geom, coeff… in PetscFEIntegrateBd()
1479 PetscErrorCode PetscFEIntegrateResidual(PetscDS ds, PetscFormKey key, PetscInt Ne, PetscFEGeom *cge… in PetscFEIntegrateResidual() argument
1486 …if (fe->ops->integrateresidual) PetscCall((*fe->ops->integrateresidual)(ds, key, Ne, cgeom, coeffi… in PetscFEIntegrateResidual()
1514 …tegrateBdResidual(PetscDS ds, PetscWeakForm wf, PetscFormKey key, PetscInt Ne, PetscFEGeom *fgeom,… in PetscFEIntegrateBdResidual() argument
1521 …if (fe->ops->integratebdresidual) PetscCall((*fe->ops->integratebdresidual)(ds, wf, key, Ne, fgeom… in PetscFEIntegrateBdResidual()
1551 …dResidual(PetscDS ds, PetscDS dsIn, PetscFormKey key, PetscInt s, PetscInt Ne, PetscFEGeom *fgeom,… in PetscFEIntegrateHybridResidual() argument
1559 …dresidual) PetscCall((*fe->ops->integratehybridresidual)(ds, dsIn, key, s, Ne, fgeom, cgeom, coeff… in PetscFEIntegrateHybridResidual()
[all …]
|
| /petsc/src/dm/dt/fe/tests/ |
| H A D | ex2.c | 260 const PetscInt cS = cell, cE = PetscMin(cS + chunkSize, cEnd), Ne = cE - cS; in TestIntegration() local
275 … PetscCall(PetscFEIntegrateResidual(ds, key, Ne, chunkGeom, u, NULL, NULL, NULL, 0.0, elemVec)); in TestIntegration()
|
| H A D | ex1.c | 212 const PetscInt cS = cell, cE = PetscMin(cS + chunkSize, cEnd), Ne = cE - cS; in TestIntegration() local
228 … PetscCall(PetscFEIntegrateResidual(ds, key, Ne, chunkGeom, u, NULL, NULL, NULL, 0.0, elemVec)); in TestIntegration()
|