Add DistArray support to assign
Merge br 'assign' into 'swaption' for DistArray