Searched refs:n_obs_vertex (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/dm/impls/plex/kokkos/ |
| H A D | plexlocalizationletkf.kokkos.cxx | 69 PetscErrorCode DMPlexGetLETKFLocalizationMatrix(const PetscInt n_obs_vertex, const PetscInt n_obs_l… in DMPlexGetLETKFLocalizationMatrix() argument
91 PetscCheck(n_obs_vertex > 0, comm, PETSC_ERR_ARG_WRONG, "n_obs_vertex must be > 0"); in DMPlexGetLETKFLocalizationMatrix()
116 PetscCall(MatSeqAIJSetPreallocation(*Q, n_obs_vertex, NULL)); in DMPlexGetLETKFLocalizationMatrix()
117 PetscCall(MatMPIAIJSetPreallocation(*Q, n_obs_vertex, NULL, n_obs_vertex, NULL)); in DMPlexGetLETKFLocalizationMatrix()
121 …Int_FMT " observations, %" PetscInt_FMT " neighbors\n", n_vert_local, n_obs_global, n_obs_vertex)); in DMPlexGetLETKFLocalizationMatrix()
170 …w<PetscInt **, Kokkos::LayoutLeft, MemSpace> indices_dev("indices", n_vert_local, n_obs_vertex); in DMPlexGetLETKFLocalizationMatrix()
171 …iew<PetscScalar **, Kokkos::LayoutLeft, MemSpace> values_dev("values", n_vert_local, n_obs_vertex); in DMPlexGetLETKFLocalizationMatrix()
178 …tscReal **, Kokkos::LayoutLeft, MemSpace> best_dists_dev("best_dists", n_vert_local, n_obs_vertex); in DMPlexGetLETKFLocalizationMatrix()
179 …PetscInt **, Kokkos::LayoutLeft, MemSpace> best_idxs_dev("best_idxs", n_vert_local, n_obs_vertex); in DMPlexGetLETKFLocalizationMatrix()
192 for (PetscInt k = 0; k < n_obs_vertex; ++k) { in DMPlexGetLETKFLocalizationMatrix()
[all …]
|