Searched refs:vector_size (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/ts/utils/dmplexlandau/kokkos/ |
| H A D | landau.kokkos.cxx | 388 int nfaces = 0, vector_size = 256 / Nq; in LandauKokkosJacobian() local
448 while (vector_size & (vector_size - 1)) vector_size = vector_size & (vector_size - 1); in LandauKokkosJacobian()
449 if (vector_size > 16) vector_size = 16; // printf("DEBUG\n"); in LandauKokkosJacobian()
512 …df, KOKKOS_SHARED_LEVEL, (int)(num_cells_batch * batch_sz), team_size, vector_size, nfaces, (int)N… in LandauKokkosJacobian()
514 …"f, df", Kokkos::TeamPolicy<>(num_cells_batch * batch_sz, team_size, vector_size).set_scratch_size… in LandauKokkosJacobian()
727 …maximum_shared_mem_size, (int)(num_cells_batch * batch_sz), team_size, vector_size, nfaces, (int)N… in LandauKokkosJacobian()
728 …os::LaunchBounds<256, lbound2>>(num_cells_batch * batch_sz, team_size, vector_size).set_scratch_si… in LandauKokkosJacobian()
736 …ctor size=%d #face=%d Nb=%" PetscInt_FMT ", %s assembly\n", team_size, vector_size, nfaces, Nb, d_… in LandauKokkosJacobian()
738 …"Mass", Kokkos::TeamPolicy<>(num_cells_batch * batch_sz, team_size, vector_size), KOKKOS_LAMBDA(co… in LandauKokkosJacobian()
|