Searched refs:glb_elem_idx (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/ts/utils/dmplexlandau/ |
| H A D | plexland.c | 315 const PetscInt glb_elem_idx = tid % b_Nelem, b_id = tid / b_Nelem; in LandauFormJacobian_Internal() local
321 while (glb_elem_idx >= elem_offset[grid + 1]) grid++; in LandauFormJacobian_Internal()
323 …ctx->species_offset[grid + 1] - ctx->species_offset[grid], loc_elem = glb_elem_idx - elem_offset[g… in LandauFormJacobian_Internal()
521 const PetscInt fullNb = coo_elem_fullNb[glb_elem_idx], fullNb2 = fullNb * fullNb; in LandauFormJacobian_Internal()
522 …elem_offsets[glb_elem_idx] + fieldA * fullNb2 + fullNb * coo_elem_point_offsets[glb_elem_idx][f] +… in LandauFormJacobian_Internal()
1501 for (PetscInt grid = 0, glb_elem_idx = 0; grid < ctx->num_grids; grid++) { in CreateStaticData() local
1516 for (PetscInt ej = cStart, eidx = 0; ej < cEnd; ++ej, ++eidx, glb_elem_idx++) { in CreateStaticData()
1517 …if (coo_elem_offsets) coo_elem_offsets[glb_elem_idx + 1] = coo_elem_offsets[glb_elem_idx]; // star… in CreateStaticData()
1594 …coo_elem_offsets[glb_elem_idx + 1] += fullNb * fullNb; // one species block, adds a block for each… in CreateStaticData()
1596 coo_elem_fullNb[glb_elem_idx] = fullNb; in CreateStaticData()
[all …]
|
| /petsc/src/ts/utils/dmplexlandau/kokkos/ |
| H A D | landau.kokkos.cxx | 343 …dauIdx (*coo_elem_point_offsets)[LANDAU_MAX_NQND + 1], const PetscInt glb_elem_idx, const PetscInt… in landau_mat_assemble() argument
349 const int fullNb = coo_elem_fullNb[glb_elem_idx], fullNb2 = fullNb * fullNb; in landau_mat_assemble()
373 …elem_offsets[glb_elem_idx] + fieldA * fullNb2 + fullNb * coo_elem_point_offsets[glb_elem_idx][f] +… in landau_mat_assemble()
|