Searched refs:n_vert_local (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/dm/impls/plex/kokkos/ |
| H A D | plexlocalizationletkf.kokkos.cxx | 71 PetscInt dim = 0, n_vert_local, d, N, n_obs_global, n_state_local; in DMPlexGetLETKFLocalizationMatrix() local
95 n_vert_local = n_state_local / n_dof; in DMPlexGetLETKFLocalizationMatrix()
114 PetscCall(MatSetSizes(*Q, n_vert_local, n_obs_local, PETSC_DETERMINE, n_obs_global)); in DMPlexGetLETKFLocalizationMatrix()
121 …ices, %" PetscInt_FMT " observations, %" PetscInt_FMT " neighbors\n", n_vert_local, n_obs_global, … in DMPlexGetLETKFLocalizationMatrix()
129 …etscScalar **, Kokkos::LayoutLeft, MemSpace> vertex_coords_dev("vertex_coords", n_vert_local, dim); in DMPlexGetLETKFLocalizationMatrix()
132 …Kokkos::LayoutLeft, Kokkos::HostSpace> vertex_coords_host("vertex_coords_host", n_vert_local, dim); in DMPlexGetLETKFLocalizationMatrix()
137 for (PetscInt i = 0; i < n_vert_local; ++i) vertex_coords_host(i, d) = local_coords_array[i]; in DMPlexGetLETKFLocalizationMatrix()
170 …Kokkos::View<PetscInt **, Kokkos::LayoutLeft, MemSpace> indices_dev("indices", n_vert_local, n_… in DMPlexGetLETKFLocalizationMatrix()
171 …Kokkos::View<PetscScalar **, Kokkos::LayoutLeft, MemSpace> values_dev("values", n_vert_local, n_ob… in DMPlexGetLETKFLocalizationMatrix()
178 …tscReal **, Kokkos::LayoutLeft, MemSpace> best_dists_dev("best_dists", n_vert_local, n_obs_vertex); in DMPlexGetLETKFLocalizationMatrix()
[all …]
|