Lines Matching refs:n_vert_local
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()
179 …Kokkos::View<PetscInt **, Kokkos::LayoutLeft, MemSpace> best_idxs_dev("best_idxs", n_vert_local, … in DMPlexGetLETKFLocalizationMatrix()
183 …"ComputeLocalization", Kokkos::RangePolicy<ExecSpace>(0, n_vert_local), KOKKOS_LAMBDA(const PetscI… in DMPlexGetLETKFLocalizationMatrix()
241 …okkos::LayoutRight, Kokkos::HostSpace> indices_host("indices_host", n_vert_local, n_obs_vertex); in DMPlexGetLETKFLocalizationMatrix()
242 …**, Kokkos::LayoutRight, Kokkos::HostSpace> values_host("values_host", n_vert_local, n_obs_vertex); in DMPlexGetLETKFLocalizationMatrix()
257 for (PetscInt i = 0; i < n_vert_local; ++i) { in DMPlexGetLETKFLocalizationMatrix()