validators.escape: support arrays in multiple assignment.

artiq/compiler/validators/escape.py

@@ -32,8 +32,8 @@ class Region:
                     self.range.end_pos >= other.range.end_pos
 
     def intersects(self, other):
-        assert self.range.source_buffer == other.range.source_buffer
         assert self.range
+        assert self.range.source_buffer == other.range.source_buffer
 
         return (self.range.begin_pos <= other.range.begin_pos <= self.range.end_pos and \
                     other.range.end_pos > self.range.end_pos) or \
@@ -172,7 +172,7 @@ class AssignedNamesOf(algorithm.Visitor):
     visit_SubscriptT = visit_accessor
 
     def visit_sequence(self, node):
-        return reduce(list.__add__, map(self.visit, node.elts))
+        return functools.reduce(list.__add__, map(self.visit, node.elts))
 
     visit_TupleT = visit_sequence
     visit_ListT = visit_sequence
@@ -259,7 +259,6 @@ class EscapeValidator(algorithm.Visitor):
     # this property. But we will need to, if string operations are ever added.
 
     def visit_assignment(self, target, value, is_aug_assign=False):
-        target_region = self._region_of(target)
         value_region  = self._region_of(value) if not is_aug_assign else self.youngest_region
 
         # If this is a variable, we might need to contract the live range.
@@ -270,6 +269,8 @@ class EscapeValidator(algorithm.Visitor):
                     region.contract(value_region)
 
         # The assigned value should outlive the assignee
+        target_regions = [self._region_of(name) for name in self._names_of(target)]
+        for target_region in target_regions:
             if not Region.outlives(value_region, target_region):
                 if is_aug_assign:
                     target_desc = "the assignment target, allocated here,"

lit-test/test/local_access/multiple_asgn.py

@@ -0,0 +1,5 @@
+# RUN: %python -m artiq.compiler.testbench.signature %s
+
+def f():
+    x, y = [0], [0]
+    x[0], y[0]