Searched refs:myQi (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/ts/utils/dmplexlandau/kokkos/ |
| H A D | landau.kokkos.cxx | 551 Kokkos::parallel_for(Kokkos::TeamThreadRange(team, 0, Nq), [=](int myQi) { in LandauKokkosJacobian() argument
552 … const PetscReal *const invJ = &invJe[myQi * dim * dim]; // b_elem_idx: batch element index in LandauKokkosJacobian()
553 const PetscReal *Bq = &d_BB[myQi * Nb], *Dq = &d_DD[myQi * Nb * dim]; in LandauKokkosJacobian()
557 const PetscInt idx = d_ipf_offset[grid] + f * loc_nip + loc_elem * Nq + myQi; in LandauKokkosJacobian()
603 Kokkos::parallel_for(Kokkos::TeamThreadRange(team, 0, Nq), [=](int myQi) { in LandauKokkosJacobian() argument
605 const PetscInt jpidx_glb = d_ip_offset[grid] + loc_elem * Nq + myQi; in LandauKokkosJacobian()
669 gg2(d2, fieldA, myQi) = gg_temp.gg2[d2] * d_alpha[fieldA + f_off]; in LandauKokkosJacobian()
670 …for (d3 = 0; d3 < dim; d3++) gg3(d2, d3, fieldA, myQi) = -gg_temp.gg3[d2][d3] * d_alpha[fieldA + f… in LandauKokkosJacobian()
674 …orRange(team, (int)loc_Nf), [&](const int &fieldA) { gg2(dim - 1, fieldA, myQi) += d_Eq_m[fieldA +… in LandauKokkosJacobian()
679 g2(d, fieldA, myQi) = 0; in LandauKokkosJacobian()
[all …]
|