Make GPU compatible

trixi-framework · Dec 23, 2024 · 4d2ff80 · 4d2ff80
1 parent 5a1cf0c
commit 4d2ff80
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/cell_lists/full_grid.jl b/src/cell_lists/full_grid.jl
@@ -199,9 +199,10 @@ function check_domain_bounds(cell_list::FullGridCellList, y, search_radius)
         error("particle coordinates contain NaNs")
     end
 
-    # Require one extra layer in each direction to make sure neighbor cells exist
-    min_corner = minimum(y, dims = 2) .- search_radius
-    max_corner = maximum(y, dims = 2) .+ search_radius
+    # Require one extra layer in each direction to make sure neighbor cells exist.
+    # Convert to CPU in case this lives on the GPU.
+    min_corner = Array(minimum(y, dims = 2) .- search_radius)
+    max_corner = Array(maximum(y, dims = 2) .+ search_radius)
 
     if any(min_corner .< cell_list.min_corner) || any(max_corner .> cell_list.max_corner)
         error("particle coordinates are outside the domain bounds of the cell list")