Searched refs:a_numCells (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/ts/utils/dmplexlandau/kokkos/ |
| H A D | landau.kokkos.cxx | 137 …tscInt Nb, const PetscInt batch_sz, const PetscInt num_grids, PetscInt a_numCells[], PetscInt a_sp… in LandauKokkosStaticDataSet() argument
157 elem_offset[grid + 1] = elem_offset[grid] + a_numCells[grid]; in LandauKokkosStaticDataSet()
158 nip += a_numCells[grid] * Nq; in LandauKokkosStaticDataSet()
160 IPf_sz += Nq * nfloc * a_numCells[grid]; in LandauKokkosStaticDataSet()
216 …tLeft, Kokkos::HostSpace, Kokkos::MemoryTraits<Kokkos::Unmanaged>> h_NCells(a_numCells, num_grids); in LandauKokkosStaticDataSet()
381 … Nb, const PetscInt batch_sz, const PetscInt num_grids, const PetscInt a_numCells[], PetscReal a_E… in LandauKokkosJacobian() argument
475 if (a_numCells[grid] > num_cells_max) num_cells_max = a_numCells[grid]; in LandauKokkosJacobian()
476 num_cells_batch += a_numCells[grid]; // we don't have a host element offset here (but in ctx) in LandauKokkosJacobian()
497 closure_sz += Nb * nfloc * a_numCells[grid]; in LandauKokkosJacobian()
810 PetscPrintf(PETSC_COMM_SELF, "Kokkos Element matrix %d/%d\n", 1, (int)a_numCells[grid]); in LandauKokkosJacobian()
|