112     for (PetscInt i = 0; i < ctx->num_grids * ctx->batch_sz; i++) subJ[i] = NULL;  in LandauFormJacobian_Internal()
115     for (PetscInt tid = 0; tid < ctx->batch_sz; tid++) {  in LandauFormJacobian_Internal()
156       PetscCall(PetscMalloc1(cellClosure_sz * ctx->batch_sz, &cellClosure));  in LandauFormJacobian_Internal()
162       for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) { // OpenMP (once)  in LandauFormJacobian_Internal()
180 …batch_sz, PETSC_COMM_SELF, PETSC_ERR_PLIB, "iteration wrong %" PetscCount_FMT " != cellClosure_sz …  in LandauFormJacobian_Internal()
204 …PetscCall(LandauKokkosJacobian(ctx->plex, Nq, Nb, ctx->batch_sz, ctx->num_grids, numCells, Eq_m, c…  in LandauFormJacobian_Internal()
232     IPf_sz_tot = IPf_sz_glb * ctx->batch_sz;  in LandauFormJacobian_Internal()
243 …for (PetscInt tid = 0; tid < ctx->batch_sz * elem_offset[num_grids]; tid++) {                     …  in LandauFormJacobian_Internal()
250 …const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offset);…  in LandauFormJacobian_Internal()
313 …  for (PetscInt tid = 0; tid < ctx->batch_sz * elem_offset[num_grids]; tid++) { // for each element  in LandauFormJacobian_Internal()
324 …const PetscInt   moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offset…  in LandauFormJacobian_Internal()
553       for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) { // OpenMP  in LandauFormJacobian_Internal()
555 …const PetscInt     moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offs…  in LandauFormJacobian_Internal()
1084   ctx->batch_sz       = 1;  in ProcessOptions()
1163 …andau_batch_size", "Number of 'vertices' to batch", "ex2.c", ctx->batch_sz, &ctx->batch_sz, NULL));  in ProcessOptions()
1164 …Z >= ctx->batch_sz, ctx->comm, PETSC_ERR_ARG_WRONG, "LANDAU_MAX_BATCH_SZ %d < ctx->batch_sz %" Pet…  in ProcessOptions()
1166 …->batch_sz, ctx->comm, PETSC_ERR_ARG_WRONG, "-ctx->batch_view_idx %" PetscInt_FMT " > ctx->batch_s…  in ProcessOptions()
1351 …e)(ctx->v_0 / SPEED_OF_LIGHT), (double)ctx->n_0, (double)ctx->t_0, ctx->batch_sz, ctx->batch_view_…  in ProcessOptions()
1621       ctx->SData_d.coo_size = coo_elem_offsets[ncellsTot] * ctx->batch_sz;  in CreateStaticData()
1650       for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) {  in CreateStaticData()
1652 … const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offset);  in CreateStaticData()
1816 …PetscCall(LandauKokkosStaticDataSet(ctx->plex[0], Nq, Nb, ctx->batch_sz, ctx->num_grids, numCells,…  in CreateStaticData()
1884 …field_major_order) PetscCall(PetscMalloc1(ctx->mat_offset[ctx->num_grids] * ctx->batch_sz, &idxs));  in LandauCreateJacobianMatrix()
1914 …for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) { // add batch size DMs for this species grid  in LandauCreateJacobianMatrix()
1919         PetscInt n0 = ctx->mat_offset[grid] * ctx->batch_sz + b_id * n;  in LandauCreateJacobianMatrix()
1928 … PetscCall(ISCreateGeneral(comm, ctx->mat_offset[ctx->num_grids] * ctx->batch_sz, idxs, PETSC_OWN_…  in LandauCreateJacobianMatrix()
1935     for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) {  in LandauCreateJacobianMatrix()
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()
2104   for (PetscInt b_id = 1; b_id < ctx->batch_sz; b_id++) {  in DMPlexLandauCreateVelocitySpace()
2128     for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) {  in DMPlexLandauCreateVelocitySpace()
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()
2191     *pNf = ctx->batch_sz;  in DMPlexLandauCreateVelocitySpace()
2235       for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) {  in DMPlexLandauAccess()
2236 … const PetscInt moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offset);  in DMPlexLandauAccess()
2300 …MT " threads\n", ctx->times[LANDAU_EX2_TSSOLVE] - ctx->times[LANDAU_MATRIX_TOTAL], ctx->batch_sz));  in DMPlexLandauDestroyVelocitySpace()
2411 …grids * ctx->batch_sz, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "#DM wrong %" PetscInt_FMT " %" PetscInt_…  in DMPlexLandauPrintNorms()
2586 …for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) { // add batch size DMs for this species grid  in DMPlexLandauCreateMassMatrix()
2594   for (PetscInt b_id = 1; b_id < ctx->batch_sz; b_id++) {  in DMPlexLandauCreateMassMatrix()
2623     for (PetscInt b_id = 0; b_id < ctx->batch_sz; b_id++) {  in DMPlexLandauCreateMassMatrix()
2624 …const PetscInt     moffset = LAND_MOFFSET(b_id, grid, ctx->batch_sz, ctx->num_grids, ctx->mat_offs…  in DMPlexLandauCreateMassMatrix()