Searched refs:b_elem_idx (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/ts/utils/dmplexlandau/kokkos/ |
| H A D | landau.kokkos.cxx | 515 …const PetscInt b_Nelem = d_elem_offset[num_grids], b_elem_idx = team.league_rank() % b_Nelem, b_id… in LandauKokkosJacobian()
518 while (b_elem_idx >= d_elem_offset[grid + 1]) grid++; in LandauKokkosJacobian()
520 …c_Nf = d_species_offset[grid + 1] - d_species_offset[grid], loc_elem = b_elem_idx - d_elem_offset[… in LandauKokkosJacobian()
587 …const PetscInt b_Nelem = d_elem_offset[num_grids], b_elem_idx = team.league_rank() % b_Nelem, b_id… in LandauKokkosJacobian() local
590 while (b_elem_idx >= d_elem_offset[grid + 1]) grid++; in LandauKokkosJacobian()
592 …c_Nf = d_species_offset[grid + 1] - d_species_offset[grid], loc_elem = b_elem_idx - d_elem_offset[… in LandauKokkosJacobian()
712 …rid], d_coo_elem_offsets, d_coo_elem_fullNb, d_coo_elem_point_offsets, b_elem_idx, b_id * coo_sz_b… in LandauKokkosJacobian()
739 …const PetscInt b_Nelem = d_elem_offset[num_grids], b_elem_idx = team.league_rank() % b_Nelem, b_id… in LandauKokkosJacobian()
742 while (b_elem_idx >= d_elem_offset[grid + 1]) grid++; in LandauKokkosJacobian()
744 …c_Nf = d_species_offset[grid + 1] - d_species_offset[grid], loc_elem = b_elem_idx - d_elem_offset[… in LandauKokkosJacobian()
[all …]
|
| /petsc/src/ts/utils/dmplexlandau/ |
| H A D | plexland.c | 244 …const PetscInt b_Nelem = elem_offset[num_grids], b_elem_idx = tid % b_Nelem, b_id = tid / b_Nelem;… in LandauFormJacobian_Internal() local
247 while (b_elem_idx >= elem_offset[grid + 1]) grid++; // yuck search for grid in LandauFormJacobian_Internal()
249 … ctx->species_offset[grid + 1] - ctx->species_offset[grid], loc_elem = b_elem_idx - elem_offset[gr… in LandauFormJacobian_Internal()
|