diff options
Diffstat (limited to 'mlir/test/python/ir/blocks.py')
| -rw-r--r-- | mlir/test/python/ir/blocks.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/mlir/test/python/ir/blocks.py b/mlir/test/python/ir/blocks.py index 811dcd7..5aa1bf8 100644 --- a/mlir/test/python/ir/blocks.py +++ b/mlir/test/python/ir/blocks.py @@ -5,6 +5,7 @@ import io import itertools from mlir.ir import * from mlir.dialects import builtin +from mlir.dialects import cf # Note: std dialect needed for terminators. from mlir.dialects import std @@ -43,11 +44,11 @@ def testBlockCreation(): with InsertionPoint(entry_block) as entry_ip: assert entry_ip.block == entry_block - std.BranchOp([i16_arg], dest=middle_block) + cf.BranchOp([i16_arg], dest=middle_block) with InsertionPoint(middle_block) as middle_ip: assert middle_ip.block == middle_block - std.BranchOp([i32_arg], dest=successor_block) + cf.BranchOp([i32_arg], dest=successor_block) print(module.operation) # Ensure region back references are coherent. assert entry_block.region == middle_block.region == successor_block.region |