Searched refs:num_cells_batch (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/ts/utils/dmplexlandau/kokkos/ |
| H A D | landau.kokkos.cxx | 387 PetscInt dim, num_cells_max, Nf_max, num_cells_batch; in LandauKokkosJacobian() local
471 num_cells_batch = Nf_max = num_cells_max = 0; in LandauKokkosJacobian()
476 num_cells_batch += a_numCells[grid]; // we don't have a host element offset here (but in ctx) in LandauKokkosJacobian()
512 …d, #face=%d, Nf_max=%d\n", scr_bytes_fdf, KOKKOS_SHARED_LEVEL, (int)(num_cells_batch * batch_sz), … in LandauKokkosJacobian()
514 …"f, df", Kokkos::TeamPolicy<>(num_cells_batch * batch_sz, team_size, vector_size).set_scratch_size… in LandauKokkosJacobian()
727 …ared_level == 0 ? "local" : "global", maximum_shared_mem_size, (int)(num_cells_batch * batch_sz), … in LandauKokkosJacobian()
728 …r("Jacobian", Kokkos::TeamPolicy<Kokkos::LaunchBounds<256, lbound2>>(num_cells_batch * batch_sz, t… in LandauKokkosJacobian()
738 …"Mass", Kokkos::TeamPolicy<>(num_cells_batch * batch_sz, team_size, vector_size), KOKKOS_LAMBDA(co… in LandauKokkosJacobian()
|