Searched refs:grid_r (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/ts/utils/dmplexlandau/kokkos/ |
| H A D | landau.kokkos.cxx | 614 PetscInt fieldB, d2, d3, f_off_r, grid_r, ipidx_g, nip_loc_r, loc_Nf_r; in LandauKokkosJacobian() local
622 grid_r = 0; in LandauKokkosJacobian()
623 while (ipidx >= d_ip_offset[grid_r + 1]) grid_r++; // yuck search for grid in LandauKokkosJacobian()
624 f_off_r = d_species_offset[grid_r]; in LandauKokkosJacobian()
625 ipidx_g = ipidx - d_ip_offset[grid_r]; in LandauKokkosJacobian()
626 nip_loc_r = d_numCells[grid_r] * Nq; in LandauKokkosJacobian()
627 loc_Nf_r = d_species_offset[grid_r + 1] - d_species_offset[grid_r]; in LandauKokkosJacobian()
629 const PetscInt idx = d_ipf_offset[grid_r] + fieldB * nip_loc_r + ipidx_g; in LandauKokkosJacobian()
630 …eal ff1 = d_beta[fieldB + f_off_r] * d_lambdas[LANDAU_MAX_GRIDS * grid + grid_r], ff2 = d_invMass[… in LandauKokkosJacobian()
|
| /petsc/src/ts/utils/dmplexlandau/ |
| H A D | plexland.c | 349 …for (PetscInt grid_r = 0, f_off = 0, ipidx = 0; grid_r < ctx->num_grids; grid_r++, f_off = ctx->sp… in LandauFormJacobian_Internal() local
350 PetscInt nip_loc_r = numCells[grid_r] * Nq, Nfloc_r = Nf[grid_r]; in LandauFormJacobian_Internal()
351 for (PetscInt ei_r = 0; ei_r < numCells[grid_r]; ++ei_r) { 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()
369 … temp1[0] += dudx[idx] * nu_beta[f + f_off] * invMass[f + f_off] * (*lambdas)[grid][grid_r]; in LandauFormJacobian_Internal()
370 … temp1[1] += dudy[idx] * nu_beta[f + f_off] * invMass[f + f_off] * (*lambdas)[grid][grid_r]; in LandauFormJacobian_Internal()
372 … temp1[2] += dudz[idx] * nu_beta[f + f_off] * invMass[f + f_off] * (*lambdas)[grid][grid_r]; in LandauFormJacobian_Internal()
374 temp2 += ff[idx] * nu_beta[f + f_off] * (*lambdas)[grid][grid_r]; in LandauFormJacobian_Internal()
|