Searched refs:b_id (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/ts/utils/dmplexlandau/kokkos/ |
| H A D | landau.kokkos.cxx | 515 …em = d_elem_offset[num_grids], b_elem_idx = team.league_rank() % b_Nelem, b_id = team.league_rank(… in LandauKokkosJacobian()
521 const PetscInt moffset = LAND_MOFFSET(b_id, grid, batch_sz, num_grids, d_mat_offset); 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()
558 d_fdf_k(b_id, 0, idx) = 0.0; in LandauKokkosJacobian()
561 d_fdf_k(b_id, 0, idx) += Bq[b] * PetscRealPart(coef[f * Nb + b]); in LandauKokkosJacobian()
565 …for (e = 0, d_fdf_k(b_id, d + 1, idx) = 0.0; e < dim; ++e) d_fdf_k(b_id, d + 1, idx) += invJ[e * d… in LandauKokkosJacobian()
587 …em = d_elem_offset[num_grids], b_elem_idx = team.league_rank() % b_Nelem, b_id = team.league_rank(… in LandauKokkosJacobian() local
593 const PetscInt moffset = LAND_MOFFSET(b_id, grid, batch_sz, num_grids, d_mat_offset); in LandauKokkosJacobian()
631 temp1[0] += d_fdf_k(b_id, 1, idx) * ff2; in LandauKokkosJacobian()
632 temp1[1] += d_fdf_k(b_id, 2, idx) * ff2; in LandauKokkosJacobian()
[all …]
|
| /petsc/src/ts/utils/dmplexlandau/ |
| H A D | plexland.c | 162 for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) { // OpenMP (once) in LandauFormJacobian_Internal() local
164 …Vec locX = locXArray[LAND_PACK_IDX(b_id, grid)], globX = globXArray[LAND_PACK_IDX(b_id, grid)… in LandauFormJacobian_Internal()
244 …const PetscInt b_Nelem = elem_offset[num_grids], b_elem_idx = tid % b_Nelem, b_id = tid / b_Nelem;… in LandauFormJacobian_Internal() local
250 …const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offset);… in LandauFormJacobian_Internal()
255 …coef = &cellClosure[b_id * IPf_sz_glb + ipf_offset[grid] + loc_elem * Nb * loc_Nf]; // this is con… in LandauFormJacobian_Internal()
284 … const PetscInt idx = b_id * IPf_sz_glb + ipf_offset[grid] + f * loc_nip + loc_elem * Nq + qi; in LandauFormJacobian_Internal()
315 const PetscInt glb_elem_idx = tid % b_Nelem, b_id = tid / b_Nelem; in LandauFormJacobian_Internal() local
324 …const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offset… in LandauFormJacobian_Internal()
367 … const PetscInt idx = b_id * IPf_sz_glb + ipf_offset[grid_r] + f * nip_loc_r + ei_r * Nq + qi; in LandauFormJacobian_Internal()
479 …ure(ctx->plex[grid], section[grid], globsection[grid], subJ[LAND_PACK_IDX(b_id, grid)], loc_elem +… in LandauFormJacobian_Internal()
[all …]
|
| /petsc/src/ts/utils/dmplexlandau/tutorials/ |
| H A D | ex1.c | 23 …print_access_callback(DM dm, Vec x, PetscInt local_field, PetscInt grid, PetscInt b_id, void *vctx) in landau_field_print_access_callback() argument
32 val = (PetscScalar)(LAND_PACK_IDX(b_id, grid) + (species + 1) * 10); in landau_field_print_access_callback()
34 … and local field %" PetscInt_FMT " with %" PetscInt_FMT " grids\n", grid, b_id, local_field, ctx->… in landau_field_print_access_callback()
373 for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) { in main() local
376 …PetscCall(SetMaxwellians(ctx->plex[grid], XsubArray[LAND_PACK_IDX(b_id, grid)], 0.0, ctx->thermal_… in main()
|