Proof of concept: TrixiMPIArray

Project.toml

@@ -4,6 +4,7 @@ authors = ["Michael Schlottke-Lakemper <[email protected]>", "Gregor
 version = "0.4.28-pre"
 
 [deps]
+ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
 CodeTracking = "da1fd8a2-8d9e-5ec2-8556-3022fb5608a2"
 ConstructionBase = "187b0558-2788-49d3-abe0-74a17ed4e7c9"
 CPUSummary = "2a0fbf3d-bb9c-48f3-b0a9-814d99fd7ab9"
@@ -41,6 +42,7 @@ TriplotRecipes = "808ab39a-a642-4abf-81ff-4cb34ebbffa3"
 UnPack = "3a884ed6-31ef-47d7-9d2a-63182c4928ed"
 
 [compat]
+ArrayInterface = "3"
 CodeTracking = "1.0.5"
 ConstructionBase = "1.3"
 CPUSummary = "=0.1.8" # see https://github.com/JuliaSIMD/CPUSummary.jl/issues/6

docs/src/reference-trixi.md

@@ -5,5 +5,5 @@ CurrentModule = Trixi
 ```
 
 ```@autodocs
-Modules = [Trixi]
+Modules = [Trixi, Trixi.TrixiMPIArrays]
 ```

src/Trixi.jl

@@ -25,6 +25,7 @@ using SparseArrays: AbstractSparseMatrix, AbstractSparseMatrixCSC, sparse, dropt
 # import @reexport now to make it available for further imports/exports
 using Reexport: @reexport
 
+using ArrayInterface: static_length
 using SciMLBase: CallbackSet, DiscreteCallback,
                  ODEProblem, ODESolution, ODEFunction
 import SciMLBase: get_du, get_tmp_cache, u_modified!,
@@ -39,7 +40,6 @@ using HDF5: h5open, attributes
 using IfElse: ifelse
 using LinearMaps: LinearMap
 using LoopVectorization: LoopVectorization, @turbo, indices
-using LoopVectorization.ArrayInterface: static_length
 using MPI: MPI
 using MuladdMacro: @muladd
 using GeometryBasics: GeometryBasics
@@ -99,6 +99,7 @@ include("basic_types.jl")
 # Include all top-level source files
 include("auxiliary/auxiliary.jl")
 include("auxiliary/mpi.jl")
+include("auxiliary/mpi_arrays.jl")
 include("auxiliary/p4est.jl")
 include("equations/equations.jl")
 include("meshes/meshes.jl")