Skip to content

Commit 3c4038c

Browse files
committed
DArray: setindex! should occur on compatible processor
1 parent 8e724d0 commit 3c4038c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/array/indexing.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,7 @@ function Base.setindex!(A::DArray{T,N}, value, idx::NTuple{N,Int}) where {T,N}
127127
# Set the value
128128
part = A.chunks[part_idx...]
129129
space = memory_space(part)
130-
scope = Dagger.scope(worker=root_worker_id(space))
130+
scope = UnionScope(map(ExactScope, processors(space)))
131131
return fetch(Dagger.@spawn scope=scope setindex!(part, value, offset_idx...))
132132
end
133133
Base.setindex!(A::DArray, value, idx::Integer...) =

0 commit comments

Comments
 (0)