Update batch.jl

PTsolvers · May 16, 2024 · e831650 · e831650
1 parent 0bfa360
commit e831650
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/BoundaryConditions/batch.jl b/src/BoundaryConditions/batch.jl
@@ -78,8 +78,11 @@ end
 # exchange-only interface
 
 function batch(grid; exchange=nothing)
-    exchange = regularise_exchange(grid, exchange)
-    return batch_set(grid, (), (), exchange)
+    exch = regularise_exchange(grid, exchange)
+    ntuple(Val(ndims(grid))) do D
+        batch(Side(1), Dim(D), grid, (), (); exchange=exch[D]),
+        batch(Side(2), Dim(D), grid, (), (); exchange=exch[D])
+    end
 end
 
 # custom batcher interface