89   PetscInt           numCells[LANDAU_MAX_GRIDS], Nq, Nb;  in LandauFormJacobian_Internal()  local
123   PetscCall(PetscFEGetDimension(ctx->fe[0], &Nb));  in LandauFormJacobian_Internal()
125 …PetscCheck(Nb <= LANDAU_MAX_NQND, ctx->comm, PETSC_ERR_ARG_WRONG, "Order too high. Nb = %" PetscIn…  in LandauFormJacobian_Internal()
155 …for (PetscInt grid = 0; grid < ctx->num_grids; grid++) cellClosure_sz += Nb * Nf[grid] * numCells[…  in LandauFormJacobian_Internal()
173 …PetscCall(PetscMemcpy(cellClosure_it, coef, Nb * Nf[grid] * sizeof(*cellClosure_it))); /* change i…  in LandauFormJacobian_Internal()
175             cellClosure_it += Nb * Nf[grid];  in LandauFormJacobian_Internal()
204 …PetscCall(LandauKokkosJacobian(ctx->plex, Nq, Nb, ctx->batch_sz, ctx->num_grids, numCells, Eq_m, c…  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()
260               for (b = 0; b < Nb; ++b) {  in LandauFormJacobian_Internal()
263                   coef[f * Nb + b] = xdata[idx + moffset];  in LandauFormJacobian_Internal()
266                   coef[f * Nb + b] = 0;  in LandauFormJacobian_Internal()
270                     coef[f * Nb + b] += scale * xdata[id + moffset];  in LandauFormJacobian_Internal()
279             const PetscReal *Bq   = &BB[qi * Nb];  in LandauFormJacobian_Internal()
280             const PetscReal *Dq   = &DD[qi * Nb * dim];  in LandauFormJacobian_Internal()
289               for (b = 0; b < Nb; ++b) {  in LandauFormJacobian_Internal()
291                 ff[idx] += Bq[cidx] * PetscRealPart(coef[f * Nb + cidx]);  in LandauFormJacobian_Internal()
292 …Int d = 0; d < dim; ++d) refSpaceDer[d] += Dq[cidx * dim + d] * PetscRealPart(coef[f * Nb + cidx]);  in LandauFormJacobian_Internal()
444             const PetscReal *BJq = &BB[qj * Nb], *DIq = &DD[qj * Nb * dim];  in LandauFormJacobian_Internal()
447               for (f = 0; f < Nb; f++) {  in LandauFormJacobian_Internal()
448                 const PetscInt i = fieldA * Nb + f; /* Element matrix row */  in LandauFormJacobian_Internal()
449                 for (g = 0; g < Nb; ++g) {  in LandauFormJacobian_Internal()
450                   const PetscInt j    = fieldA * Nb + g; /* Element matrix column */  in LandauFormJacobian_Internal()
488             for (f = 0; f < Nb; f++) {  in LandauFormJacobian_Internal()
502               for (g = 0; g < Nb; ++g) {  in LandauFormJacobian_Internal()
517                 const PetscInt    i   = fieldA * Nb + f; /* Element matrix row */  in LandauFormJacobian_Internal()
518                 const PetscInt    j   = fieldA * Nb + g; /* Element matrix column */  in LandauFormJacobian_Internal()
913   PetscInt        Nq, Nb, *Nb2, cStart, cEnd, c, dim, qj, k;  in adaptToleranceFEM()  local
929   PetscCall(PetscFEGetDimension(ctx->fe[0], &Nb));  in adaptToleranceFEM()
931 …PetscCheck(Nb2[0] == Nb, ctx->comm, PETSC_ERR_ARG_WRONG, " Nb = %" PetscInt_FMT " != Nb (%" PetscI…  in adaptToleranceFEM()
932 …PetscCheck(Nb <= LANDAU_MAX_NQND, ctx->comm, PETSC_ERR_ARG_WRONG, "Order too high. Nb = %" PetscIn…  in adaptToleranceFEM()
1409 …PetscInt         numCells[LANDAU_MAX_GRIDS], Nq, Nb, Nf[LANDAU_MAX_GRIDS], ncellsTot = 0, MAP_BF_S…  in CreateStaticData()  local
1414   PetscCall(PetscFEGetDimension(ctx->fe[0], &Nb));  in CreateStaticData()
1415 …PetscCheck(Nb <= LANDAU_MAX_NQND, ctx->comm, PETSC_ERR_ARG_WRONG, "Order too high. Nb = %" PetscIn…  in CreateStaticData()
1481     const PetscInt  *plex_batch = NULL, elMatSz = Nb * Nb * ctx->num_species * ctx->num_species;  in CreateStaticData()
1502       PetscInt cStart, cEnd, Nfloc = Nf[grid], totDim = Nfloc * Nb;  in CreateStaticData()
1520           for (PetscInt q = 0; q < Nb; ++q) {  in CreateStaticData()
1524             elMat[(fieldA * Nb + q) * totDim + fieldA * Nb + q] = 1;  in CreateStaticData()
1527 …eck(numindices == Nb, ctx->comm, PETSC_ERR_ARG_WRONG, "numindices != Nb numindices=%" PetscInt_FMT…  in CreateStaticData()
1630           for (PetscInt f = 0, cnt2 = 0; f < Nb; f++) {  in CreateStaticData()
1646 …Nb] == fullNb, PETSC_COMM_SELF, PETSC_ERR_PLIB, "coo_elem_point_offsets size %" PetscInt_FMT " != …  in CreateStaticData()
1659               for (PetscInt f = 0; f < Nb; ++f) {  in CreateStaticData()
1666                 for (PetscInt g = 0; g < Nb; ++g) {  in CreateStaticData()
1712 …" N=%" PetscInt_FMT " nnz= %" PetscInt_FMT "\n", 0, "FormLandau", nip_glb, ncells, Nb, Nq, dim, Nb,  in CreateStaticData()
1713                             ctx->num_species, Nb, dim, N, (PetscInt)info.nz_used));  in CreateStaticData()
1769             const PetscReal       *Dq = &DD[qj * Nb * dim];  in CreateStaticData()
1771             for (PetscInt b = 0; b < Nb; ++b) {  in CreateStaticData()
1816 …PetscCall(LandauKokkosStaticDataSet(ctx->plex[0], Nq, Nb, ctx->batch_sz, ctx->num_grids, numCells,…  in CreateStaticData()