Searched refs:loc_Nf (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/ts/utils/dmplexlandau/kokkos/ |
| H A D | landau.kokkos.cxx | 520 …const PetscInt loc_nip = d_numCells[grid] * Nq, loc_Nf = d_species_offset[grid + 1] - d_species_of… in LandauKokkosJacobian()
528 …coef = &d_vertex_f[b_id * IPf_sz_glb + d_ipf_offset[grid] + loc_elem * Nb * loc_Nf]; // closure an… in LandauKokkosJacobian()
531 Kokkos::parallel_for(Kokkos::TeamThreadRange(team, 0, (int)loc_Nf), [=](int f) { in LandauKokkosJacobian()
554 Kokkos::parallel_for(Kokkos::ThreadVectorRange(team, 0, (int)loc_Nf), [=](int f) { in LandauKokkosJacobian()
592 …const PetscInt loc_Nf = d_species_offset[grid + 1] - d_species_offset[grid], loc_elem = b_elem_idx… in LandauKokkosJacobian() local
596 const PetscInt totDim = loc_Nf * Nb; in LandauKokkosJacobian()
598 g2_scr_t g2(team.team_scratch(jac_shared_level), dim, loc_Nf, Nq); in LandauKokkosJacobian()
599 g3_scr_t g3(team.team_scratch(jac_shared_level), dim, dim, loc_Nf, Nq); in LandauKokkosJacobian()
600 g2_scr_t gg2(team.team_scratch(jac_shared_level), dim, loc_Nf, Nq); in LandauKokkosJacobian()
601 g3_scr_t gg3(team.team_scratch(jac_shared_level), dim, dim, loc_Nf, Nq); in LandauKokkosJacobian()
[all …]
|
| /petsc/src/ts/utils/dmplexlandau/ |
| H A D | plexland.c | 249 …const PetscInt loc_nip = numCells[grid] * Nq, loc_Nf = ctx->species_offset[grid + 1] - ctx->specie… in LandauFormJacobian_Internal() local
255 …coef = &cellClosure[b_id * IPf_sz_glb + ipf_offset[grid] + loc_elem * Nb * loc_Nf]; // this is con… in LandauFormJacobian_Internal()
258 for (f = 0; f < loc_Nf; ++f) { in LandauFormJacobian_Internal()
283 for (f = 0; f < loc_Nf; ++f) { in LandauFormJacobian_Internal()
323 …const PetscInt loc_Nf = ctx->species_offset[grid + 1] - ctx->species_offset[grid], loc_elem = gl… in LandauFormJacobian_Internal() local
324 …ET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offset), totDim = loc_Nf * Nq, elemMatSize … in LandauFormJacobian_Internal()
407 … for (PetscInt fieldA = 0, f_off = ctx->species_offset[grid]; fieldA < loc_Nf; ++fieldA) { in LandauFormJacobian_Internal()
414 …for (PetscInt fieldA = 0, f_off = ctx->species_offset[grid]; fieldA < loc_Nf; ++fieldA) gg2[fieldA… in LandauFormJacobian_Internal()
416 for (PetscInt fieldA = 0; fieldA < loc_Nf; ++fieldA) { in LandauFormJacobian_Internal()
433 for (PetscInt fieldA = 0; fieldA < loc_Nf; ++fieldA) { in LandauFormJacobian_Internal()
[all …]
|