| /petsc/src/ts/utils/dmplexlandau/kokkos/ |
| H A D | landau.kokkos.cxx | 343 …const PetscInt Nb, PetscInt moffset, const PetscInt elem, const PetscInt fieldA, const P4estVertex… in landau_mat_assemble() argument
346 const LandauIdx *const Idxs = &d_maps->gIdx[elem][fieldA][0]; in landau_mat_assemble()
373 …const int idx0 = bid_coo_sz_batch + coo_elem_offsets[glb_elem_idx] + fieldA * fullNb2 + fullNb * c… in landau_mat_assemble()
666 …Kokkos::parallel_for(Kokkos::ThreadVectorRange(team, (int)loc_Nf), [&](const int &fieldA) { // \al… in LandauKokkosJacobian() argument
669 gg2(d2, fieldA, myQi) = gg_temp.gg2[d2] * d_alpha[fieldA + f_off]; in LandauKokkosJacobian()
670 …d3 = 0; d3 < dim; d3++) gg3(d2, d3, fieldA, myQi) = -gg_temp.gg3[d2][d3] * d_alpha[fieldA + f_off]… in LandauKokkosJacobian()
674 …eadVectorRange(team, (int)loc_Nf), [&](const int &fieldA) { gg2(dim - 1, fieldA, myQi) += d_Eq_m[f… in LandauKokkosJacobian() argument
675 … Kokkos::parallel_for(Kokkos::ThreadVectorRange(team, (int)loc_Nf), [=](const int &fieldA) { in LandauKokkosJacobian() argument
679 g2(d, fieldA, myQi) = 0; in LandauKokkosJacobian()
681 g2(d, fieldA, myQi) += invJ[d * dim + d2] * gg2(d2, fieldA, myQi); in LandauKokkosJacobian()
[all …]
|
| /petsc/src/ts/utils/dmplexlandau/ |
| H A D | plexland.c | 139 for (PetscInt fieldA = 0; fieldA < ctx->num_species; fieldA++) { in LandauFormJacobian_Internal() local
140 …Eq_m[fieldA] = ctx->Ez * ctx->t_0 * ctx->charges[fieldA] / (ctx->v_0 * ctx->masses[fieldA]); /* no… in LandauFormJacobian_Internal()
141 …if (dim == 2) Eq_m[fieldA] *= 2 * PETSC_PI; /* ad… in LandauFormJacobian_Internal()
407 … for (PetscInt fieldA = 0, f_off = ctx->species_offset[grid]; fieldA < loc_Nf; ++fieldA) { in LandauFormJacobian_Internal() local
409 gg2[fieldA][d2] = gg2_temp[d2] * nu_alpha[fieldA + f_off]; in LandauFormJacobian_Internal()
410 … (d3 = 0; d3 < LANDAU_DIM; d3++) gg3[fieldA][d2][d3] = -gg3_temp[d2][d3] * nu_alpha[fieldA + f_off… in LandauFormJacobian_Internal()
414 …for (PetscInt fieldA = 0, f_off = ctx->species_offset[grid]; fieldA < loc_Nf; ++fieldA) gg2[fieldA… in LandauFormJacobian_Internal() local
416 for (PetscInt fieldA = 0; fieldA < loc_Nf; ++fieldA) { in LandauFormJacobian_Internal() local
418 g2[fieldA][d] = 0.0; in LandauFormJacobian_Internal()
420 g2[fieldA][d] += invJj[d * dim + d2] * gg2[fieldA][d2]; in LandauFormJacobian_Internal()
[all …]
|
| /petsc/src/dm/impls/plex/tests/ |
| H A D | ex64.c | 24 const PetscInt fieldA = 2; in main() local
54 PetscCall(PetscSectionSetFieldName(section, fieldA, "Alpha")); in main()
62 PetscCall(PetscSectionSetFieldComponents(section, fieldA, 1)); in main()
165 PetscCall(PetscSectionSetFieldDof(section, closure[2 * p], fieldA, dofA[d])); in main()
185 PetscCall(DMCreateSubDM(dm, 1, &fieldA, &seqisA, &seqdmA)); in main()
281 PetscCall(DMCreateSubDM(pdm, 1, &fieldA, &isA, &dmA)); in main()
|
| H A D | ex26.c | 21 const PetscInt fieldA = 2; in main() local
134 PetscCall(PetscSectionSetFieldName(section, fieldA, "Alpha")); in main()
142 PetscCall(PetscSectionSetFieldComponents(section, fieldA, 1)); in main()
245 PetscCall(PetscSectionSetFieldDof(section, closure[2 * p], fieldA, dofA[d])); in main()
283 PetscCall(DMCreateSubDM(pdm, 1, &fieldA, &isA, &dmA)); in main()
|
| H A D | ex26f90.F90 | 19 PetscInt :: fieldA = 2
198 PetscCallA(PetscSectionSetFieldName(section, fieldA, 'Alpha', ierr))
211 PetscCallA(PetscSectionSetFieldComponents(section, fieldA, 1_kPI, ierr))
288 PetscCallA(PetscSectionSetFieldDof(section, closure(p), fieldA, dofA(d), ierr))
323 PetscCallA(DMCreateSubDM(pdm, 1_kPI, [fieldA], isA, dmA, ierr))
|
| H A D | ex62f90.F90 | 19 PetscInt :: fieldA = 2
185 PetscCallA(PetscSectionSetFieldName(section, fieldA, 'Alpha', ierr))
198 PetscCallA(PetscSectionSetFieldComponents(section, fieldA, 1_kPI, ierr))
271 PetscCallA(PetscSectionSetFieldDof(section, closure(p), fieldA, dofA(d), ierr))
307 PetscCallA(DMCreateSubDM(pdm, 1_kPI, [fieldA], isA, dmA, ierr))
|