Merge pull request #110 from ChevronETC/mpisetindex

ChevronETC · May 13, 2022 · f6dd212 · f6dd212
2 parents e29ce74 + 256afb8
commit f6dd212
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/Devito.jl b/src/Devito.jl
@@ -101,7 +101,7 @@ function Base.fill!(x::DevitoMPIAbstractArray, v)
 end
 
 
-Base.setindex!(x::DevitoMPIAbstractArray{T,N}, v, i) where {T,N} = error("not implemented")
+Base.setindex!(x::DevitoMPIAbstractArray{T,N}, v, I::Vararg{Int,N}) where {T,N} = error("not implemented")
 Base.IndexStyle(::Type{<:DevitoMPIAbstractArray}) = IndexCartesian()
 
 struct DevitoMPIArray{T,N,A<:AbstractArray{T,N},D} <: DevitoMPIAbstractArray{T,N}

diff --git a/test/mpitests_2ranks.jl b/test/mpitests_2ranks.jl
@@ -621,6 +621,12 @@ end
     end
 end
 
+@testset "MPI Setindex Not Implemented" begin
+    grid = Grid(shape=(5,6,7))
+    f = Devito.Function(name="f", grid=grid)
+    @test_throws ErrorException("not implemented") data(f)[2,2,2] = 1.0
+end
+
 @testset "MPI Getindex for Function n=$n" for n in ( (11,10), (5,4), (7,2), (4,5,6), (2,3,4) )
     N = length(n)
     rnk = MPI.Comm_rank(MPI.COMM_WORLD)