Lines Matching refs:b_id
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()
522 …const PetscInt idx0 = b_id * coo_elem_offsets[elem_offset[num_grids]] + coo_elem_offsets[glb_elem_… in LandauFormJacobian_Internal()
553 for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) { // OpenMP in LandauFormJacobian_Internal() local
555 …const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offs… in LandauFormJacobian_Internal()
559 Mat B = subJ[LAND_PACK_IDX(b_id, grid)]; in LandauFormJacobian_Internal()
849 …etscReal time, PetscReal temps[], PetscReal ns[], PetscInt grid, PetscInt b_id, PetscInt n_batch, … in DMPlexLandauAddMaxwellians() argument
892 static PetscErrorCode LandauSetInitialCondition(DM dm, Vec X, PetscInt grid, PetscInt b_id, PetscIn… in LandauSetInitialCondition() argument
899 …PetscCall(DMPlexLandauAddMaxwellians(dm, X, 0.0, ctx->thermal_temps, ctx->n, grid, b_id, n_batch, … in LandauSetInitialCondition()
1650 for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) { in CreateStaticData() local
1652 … const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offset); in CreateStaticData()
1673 …const PetscInt idx0 = b_id * coo_elem_offsets[ncellsTot] + coo_elem_offsets[glb_elem_idx] + fieldA… in CreateStaticData()
1914 …for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) { // add batch size DMs for this species grid in LandauCreateJacobianMatrix() local
1916 PetscInt N = ctx->mat_offset[ctx->num_grids], n0 = ctx->mat_offset[grid] + b_id * N; in LandauCreateJacobianMatrix()
1919 PetscInt n0 = ctx->mat_offset[grid] * ctx->batch_sz + b_id * n; in LandauCreateJacobianMatrix()
1935 for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) { in LandauCreateJacobianMatrix() local
1936 …const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offs… in LandauCreateJacobianMatrix()
2096 …for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) { // add batch size DMs for this species grid in DMPlexLandauCreateVelocitySpace() local
2104 for (PetscInt b_id = 1; b_id < ctx->batch_sz; b_id++) { in DMPlexLandauCreateVelocitySpace() local
2128 for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) { in DMPlexLandauCreateVelocitySpace() local
2130 …const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offs… in DMPlexLandauCreateVelocitySpace()
2131 … PetscCall(LandauSetInitialCondition(ctx->plex[grid], Xsub[grid], grid, b_id, ctx->batch_sz, ctx)); in DMPlexLandauCreateVelocitySpace()
2235 for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) { in DMPlexLandauAccess() local
2236 … const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offset); in DMPlexLandauAccess()
2239 PetscCall(func(vdm, vec, i0, grid, b_id, user_ctx)); in DMPlexLandauAccess()
2586 …for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) { // add batch size DMs for this species grid in DMPlexLandauCreateMassMatrix() local
2594 for (PetscInt b_id = 1; b_id < ctx->batch_sz; b_id++) { in DMPlexLandauCreateMassMatrix() local
2623 for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) { in DMPlexLandauCreateMassMatrix() local
2624 …const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offs… in DMPlexLandauCreateMassMatrix()