fixup! fixup! Enhance DArray Distribution with Processor Assignment

Author: jpsamaroo

src/array/alloc.jl

@@ -82,7 +82,7 @@ function stage(ctx, a::AllocateArray)
         args = a.want_index ? (i, size(x)) : (size(x),)
 
         if isnothing(a.procgrid)
-            scope = nothing
+            scope = get_options(:compute_scope, get_options(:scope, DefaultScope()))
         else
             scope = ExactScope(a.procgrid[CartesianIndex(mod1.(Tuple(I), size(a.procgrid))...)])
         end

src/array/darray.jl

@@ -458,11 +458,11 @@ function stage(ctx::Context, d::Distribute)
             # TODO: fix hashing
             #hash = uhash(idx, Base.hash(Distribute, Base.hash(d.data)))
             if isnothing(d.procgrid)
-                options = Options(compute_scope=nothing)
+                scope = get_options(:compute_scope, get_options(:scope, DefaultScope()))
             else
                 scope = ExactScope(d.procgrid[CartesianIndex(mod1.(Tuple(I), size(d.procgrid))...)])
-                options = Options(compute_scope=scope)
             end
+            options = Options(compute_scope=scope)
             Dagger.spawn(options, shape, chunks...) do shape, parts...
                 if prod(shape) == 0
                     return Array{T}(undef, shape)
@@ -478,7 +478,7 @@ function stage(ctx::Context, d::Distribute)
             #hash = uhash(c, Base.hash(Distribute, Base.hash(d.data)))
             c = d.domainchunks[I]
             if isnothing(d.procgrid)
-                scope = nothing
+                scope = get_options(:compute_scope, get_options(:scope, DefaultScope()))
             else
                 scope = ExactScope(d.procgrid[CartesianIndex(mod1.(Tuple(I), size(d.procgrid))...)])
             end