Fix copy_to when targeting GPUs

masterleinad · May 17, 2024 · 7b88824 · 7b88824
1 parent 9193862
commit 7b88824
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/source/HeatSources.hh b/source/HeatSources.hh
@@ -377,7 +377,8 @@ HeatSources<dim, MemorySpaceType>::copy_to(
            ++i)
         target_electron_beam_scan_path_segments[i] =
             Kokkos::create_mirror_view_and_copy(
-                Kokkos::HostSpace{}, _electron_beam_scan_path_segments[i]);
+                typename TargetMemorySpaceType::kokkos_space{},
+                _electron_beam_scan_path_segments[i]);
       auto target_copy_electron_beam_heat_sources =
           Kokkos::create_mirror_view_and_copy(Kokkos::HostSpace{},
                                               _electron_beam_heat_sources);