Lines Matching refs:Kokkos

124   using ExecSpace = Kokkos::DefaultExecutionSpace;  in DMPlexGetLETKFLocalizationMatrix()
129Kokkos::View<PetscScalar **, Kokkos::LayoutLeft, MemSpace> vertex_coords_dev("vertex_coords", n_ve… in DMPlexGetLETKFLocalizationMatrix()
132Kokkos::View<PetscScalar **, Kokkos::LayoutLeft, Kokkos::HostSpace> vertex_coords_host("vertex_coo… in DMPlexGetLETKFLocalizationMatrix()
140 Kokkos::deep_copy(vertex_coords_dev, vertex_coords_host); in DMPlexGetLETKFLocalizationMatrix()
144Kokkos::View<PetscReal **, Kokkos::LayoutRight, MemSpace> obs_coords_dev("obs_coords", n_obs_globa… in DMPlexGetLETKFLocalizationMatrix()
146Kokkos::View<PetscReal **, Kokkos::LayoutRight, Kokkos::HostSpace> obs_coords_host("obs_coords_hos… in DMPlexGetLETKFLocalizationMatrix()
162 Kokkos::deep_copy(obs_coords_dev, obs_coords_host); in DMPlexGetLETKFLocalizationMatrix()
170Kokkos::View<PetscInt **, Kokkos::LayoutLeft, MemSpace> indices_dev("indices", n_vert_local, n_… in DMPlexGetLETKFLocalizationMatrix()
171Kokkos::View<PetscScalar **, Kokkos::LayoutLeft, MemSpace> values_dev("values", n_vert_local, n_ob… in DMPlexGetLETKFLocalizationMatrix()
178Kokkos::View<PetscReal **, Kokkos::LayoutLeft, MemSpace> best_dists_dev("best_dists", n_vert_local… in DMPlexGetLETKFLocalizationMatrix()
179Kokkos::View<PetscInt **, Kokkos::LayoutLeft, MemSpace> best_idxs_dev("best_idxs", n_vert_local, … in DMPlexGetLETKFLocalizationMatrix()
182 Kokkos::parallel_for( in DMPlexGetLETKFLocalizationMatrix()
183 …"ComputeLocalization", Kokkos::RangePolicy<ExecSpace>(0, n_vert_local), KOKKOS_LAMBDA(const PetscI… in DMPlexGetLETKFLocalizationMatrix()
241Kokkos::View<PetscInt **, Kokkos::LayoutRight, Kokkos::HostSpace> indices_host("indices_host", … in DMPlexGetLETKFLocalizationMatrix()
242Kokkos::View<PetscScalar **, Kokkos::LayoutRight, Kokkos::HostSpace> values_host("values_host", n_… in DMPlexGetLETKFLocalizationMatrix()
247Kokkos::View<PetscInt **, Kokkos::LayoutLeft, Kokkos::HostSpace> indices_host_left = Kokkos::cr… in DMPlexGetLETKFLocalizationMatrix()
248Kokkos::View<PetscScalar **, Kokkos::LayoutLeft, Kokkos::HostSpace> values_host_left = Kokkos::cr… in DMPlexGetLETKFLocalizationMatrix()
250 Kokkos::deep_copy(indices_host_left, indices_dev); in DMPlexGetLETKFLocalizationMatrix()
251 Kokkos::deep_copy(values_host_left, values_dev); in DMPlexGetLETKFLocalizationMatrix()
254 Kokkos::deep_copy(indices_host, indices_host_left); in DMPlexGetLETKFLocalizationMatrix()
255 Kokkos::deep_copy(values_host, values_host_left); in DMPlexGetLETKFLocalizationMatrix()