dmplexlandau/kokkos/landau.kokkos.cxx

515         const PetscInt b_Nelem = d_elem_offset[num_grids], b_elem_idx = team.league_rank() % b_Nelem, b_id = team.league_rank() / b_Nelem, IPf_sz_glb = d_ipf_offset[num_grids];
521           const PetscInt moffset = LAND_MOFFSET(b_id, grid, batch_sz, num_grids, d_mat_offset);
528               coef = &d_vertex_f[b_id * IPf_sz_glb + d_ipf_offset[grid] + loc_elem * Nb * loc_Nf]; // closure and IP indexing are the same
558                 d_fdf_k(b_id, 0, idx) = 0.0;
561                   d_fdf_k(b_id, 0, idx) += Bq[b] * PetscRealPart(coef[f * Nb + b]);
565                   for (e = 0, d_fdf_k(b_id, d + 1, idx) = 0.0; e < dim; ++e) d_fdf_k(b_id, d + 1, idx) += invJ[e * dim + d] * refSpaceDer[e];
587       const PetscInt b_Nelem = d_elem_offset[num_grids], b_elem_idx = team.league_rank() % b_Nelem, b_id = team.league_rank() / b_Nelem;
593         const PetscInt moffset = LAND_MOFFSET(b_id, grid, batch_sz, num_grids, d_mat_offset);
631                 temp1[0] += d_fdf_k(b_id, 1, idx) * ff2;
632                 temp1[1] += d_fdf_k(b_id, 2, idx) * ff2;
634                 temp1[2] += d_fdf_k(b_id, 3, idx) * ff2;
636                 temp2 += d_fdf_k(b_id, 0, idx) * ff1;
709                   d_elem_mats(b_id, grid, loc_elem, fOff) = t;
712                   static_cast<void>(landau_mat_assemble(d_coo_vals, t, f, g, Nb, moffset, loc_elem, fieldA, maps[grid], d_coo_elem_offsets, d_coo_elem_fullNb, d_coo_elem_point_offsets, b_elem_idx, b_id * coo_sz_batch));
739         const PetscInt b_Nelem = d_elem_offset[num_grids], b_elem_idx = team.league_rank() % b_Nelem, b_id = team.league_rank() / b_Nelem;
748           const PetscInt moffset = LAND_MOFFSET(b_id, grid, batch_sz, num_grids, d_mat_offset);
766                   d_elem_mats(b_id, grid, loc_elem, fOff) = t;
769                   static_cast<void>(landau_mat_assemble(d_coo_vals, t, f, g, Nb, moffset, loc_elem, fieldA, maps[grid], d_coo_elem_offsets, d_coo_elem_fullNb, d_coo_elem_point_offsets, b_elem_idx, b_id * coo_sz_batch));
790     for (PetscInt b_id = 0; b_id < batch_sz; b_id++) { // OpenMP (once)
794         Mat                B       = subJ[LAND_PACK_IDX(b_id, grid)];
795         PetscInt           moffset = LAND_MOFFSET(b_id, grid, batch_sz, num_grids, a_mat_offset), nloc, nzl, colbuf[1024], row;
805           const PetscScalar *elMat = &h_elem_mats(b_id, grid, ej - cStart, 0);
830         PetscCall(MatDestroy(&subJ[LAND_PACK_IDX(b_id, grid)]));