Fix induction variable type in scf.for py binding.

bythew3i · ftynse · commit 3d51b40c4a48 · 2023-09-21T08:07:26.000Z
- make sure that the type of induction variable should be determined by the type of the lower bound type. Reviewed By: ftynse Differential Revision: https://reviews.llvm.org/D159534
diff --git a/mlir/python/mlir/dialects/_scf_ops_ext.py b/mlir/python/mlir/dialects/_scf_ops_ext.py
@@ -52,7 +52,7 @@ def __init__(
                 ip=ip,
             )
         )
-        self.regions[0].blocks.append(IndexType.get(), *results)
+        self.regions[0].blocks.append(self.operands[0].type, *results)
 
     @property
     def body(self):

Original file line number	Diff line number	Diff line change
`@@ -52,7 +52,7 @@ def __init__(`
`52`	`52`	`ip=ip,`
`53`	`53`	`)`
`54`	`54`	`)`
`55`		`- self.regions[0].blocks.append(IndexType.get(), *results)`
	`55`	`+ self.regions[0].blocks.append(self.operands[0].type, *results)`
`56`	`56`
`57`	`57`	`@property`
`58`	`58`	`def body(self):`