Lines Matching refs:elem_offset
210 …PetscInt ip_offset[LANDAU_MAX_GRIDS + 1], ipf_offset[LANDAU_MAX_GRIDS + 1], elem_offset[LA… in LandauFormJacobian_Internal() local
224 ip_offset[0] = ipf_offset[0] = elem_offset[0] = 0; in LandauFormJacobian_Internal()
227 elem_offset[grid + 1] = elem_offset[grid] + numCells[grid]; in LandauFormJacobian_Internal()
243 …for (PetscInt tid = 0; tid < ctx->batch_sz * elem_offset[num_grids]; tid++) { … 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()
247 while (b_elem_idx >= elem_offset[grid + 1]) grid++; // yuck search for grid in LandauFormJacobian_Internal()
249 …x->species_offset[grid + 1] - ctx->species_offset[grid], loc_elem = b_elem_idx - elem_offset[grid]; in LandauFormJacobian_Internal()
313 … for (PetscInt tid = 0; tid < ctx->batch_sz * elem_offset[num_grids]; tid++) { // for each element in LandauFormJacobian_Internal()
314 const PetscInt b_Nelem = elem_offset[num_grids]; in LandauFormJacobian_Internal()
321 while (glb_elem_idx >= elem_offset[grid + 1]) grid++; in LandauFormJacobian_Internal()
323 …>species_offset[grid + 1] - ctx->species_offset[grid], loc_elem = glb_elem_idx - elem_offset[grid]; in LandauFormJacobian_Internal()
522 …const PetscInt idx0 = b_id * coo_elem_offsets[elem_offset[num_grids]] + coo_elem_offsets[glb_elem_… in LandauFormJacobian_Internal()