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()
208           PetscInt pos = n_obs_vertex - 1;  in DMPlexGetLETKFLocalizationMatrix()
218           current_max_dist = best_dists_dev(i, n_obs_vertex - 1);  in DMPlexGetLETKFLocalizationMatrix()
223       PetscReal radius2 = best_dists_dev(i, n_obs_vertex - 1);  in DMPlexGetLETKFLocalizationMatrix()
227       for (PetscInt k = 0; k < n_obs_vertex; ++k) {  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()
259 …PetscCall(MatSetValues(*Q, 1, &globalRow, n_obs_vertex, &indices_host(i, 0), &values_host(i, 0), I…  in DMPlexGetLETKFLocalizationMatrix()