dialects: (affine) add result_types to For from_region helper

xdslproject · Jun 27, 2023 · e8f9b2e · e8f9b2e
1 parent 097be14
commit e8f9b2e
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/tests/dialects/test_affine.py b/tests/dialects/test_affine.py
@@ -6,7 +6,7 @@
 
 
 def test_simple_for():
-    f = For.from_region([], 0, 5, Region())
+    f = For.from_region([], [], 0, 5, Region())
     assert f.lower_bound.value.data == 0
     assert f.upper_bound.value.data == 5
 

diff --git a/tests/test_frontend_op_inserter.py b/tests/test_frontend_op_inserter.py
@@ -28,7 +28,7 @@ def test_raises_exception_on_op_with_no_regions():
 
 def test_raises_exception_on_op_with_no_blocks():
     inserter = OpInserter(Block())
-    op_with_no_region = For.from_region([], 0, 10, Region())
+    op_with_no_region = For.from_region([], [], 0, 10, Region())
     with pytest.raises(FrontendProgramException) as err:
         inserter.set_insertion_point_from_op(op_with_no_region)
     assert err.value.msg == (

diff --git a/xdsl/dialects/affine.py b/xdsl/dialects/affine.py
@@ -63,6 +63,7 @@ def verify_(self) -> None:
     @staticmethod
     def from_region(
         operands: Sequence[Operation | SSAValue],
+        result_types: Sequence[Attribute],
         lower_bound: int | AnyIntegerAttr,
         upper_bound: int | AnyIntegerAttr,
         region: Region,
@@ -74,14 +75,13 @@ def from_region(
             upper_bound = IntegerAttr.from_index_int_value(upper_bound)
         if isinstance(step, int):
             step = IntegerAttr.from_index_int_value(step)
-        result_types = [SSAValue.get(op).typ for op in operands]
         attributes: dict[str, Attribute] = {
             "lower_bound": lower_bound,
             "upper_bound": upper_bound,
             "step": step,
         }
         return For.build(
-            operands=[[operand for operand in operands]],
+            operands=[operands],
             result_types=[result_types],
             attributes=attributes,
             regions=[region],

diff --git a/xdsl/frontend/code_generation.py b/xdsl/frontend/code_generation.py
@@ -447,7 +447,7 @@ def visit_For(self, node: ast.For):
                 and isinstance(end, int)
                 and isinstance(step, int)
             )
-            op = affine.For.from_region([], start, end, body, step)
+            op = affine.For.from_region([], [], start, end, body, step)
         else:
             self.inserter.insert_op(scf.Yield.get())
             assert (