Update Halo for Higher dimension Tensors?

[tianshu-wang]

Sometimes there are a lot tensors whose halos need be updated. Is it possible to update the halo of a higher dimension tensor A with size {nx, ny, nz, m}, where nx, ny, nz are the size of grids?

I know it is equivalent to update_halo!(A_1, A_2, ..., A_m). Just want to know if it can be done more easily when m is a large number.

Thank you in advance!

[omlins]

Thanks for reporting this issue; that's a good point!

If you use a vector of 3-D arrays to store your tensor A, then the solution is straightforward: you can simply splat the m 3-D arrays into update_halo!. Here is a minimal 2-D example (1 process, periodic boundaries):

julia> using ImplicitGlobalGrid

julia> init_global_grid(3, 3, 1, periodx=1, periody=1);
Global grid: 1x1x1 (nprocs: 1, dims: 1x1x1)

julia> A = [[1 2; 3 4; 5 6], [7 8 9; 10 11 12]]
2-element Vector{Matrix{Int64}}:
 [1 2; 3 4; 5 6]
 [7 8 9; 10 11 12]

julia> update_halo!(A...)

julia> A
2-element Vector{Matrix{Int64}}:
 [3 4; 3 4; 3 4]
 [8 8 8; 11 11 11]

Does that seem a good solution for you?

[tianshu-wang]

Yes! Thank you very much!