Back to PBT run
input.mlir
dataflow.thread private @thread_0 domain(#dataflow.thread_domain<dense>)(
%limit: index, %memory: memref<?xindex>) ctrl (%ctrl: none) {
"loom.spatial_region"(%limit, %memory)
<{operandSegmentSizes = array<i32: 1, 0, 1, 0>,
resultSegmentSizes = array<i32: 0, 0>}> ({
^bb0(%lim: index, %target: memref<?xindex>):
%zero = arith.constant 0 : index
%one = arith.constant 1 : index
%two = arith.constant 2 : index
%flag = arith.cmpi ult, %zero, %two : index
scf.for %iv0 = %zero to %two step %one {
scf.for %iv1 = %zero to %two step %one {
memref.store %one, %target[%zero] : memref<?xindex>
%ld3 = memref.load %target[%zero] : memref<?xindex>
%ad3 = arith.addi %ld3, %one : index
memref.store %ad3, %target[%zero] : memref<?xindex>
}
}
scf.if %flag {
%ld5 = memref.load %target[%zero] : memref<?xindex>
%ad5 = arith.addi %ld5, %one : index
memref.store %ad5, %target[%zero] : memref<?xindex>
%ld6 = memref.load %target[%zero] : memref<?xindex>
%ad6 = arith.addi %ld6, %one : index
memref.store %ad6, %target[%zero] : memref<?xindex>
} else {
%wr7 = scf.while (%wa7 = %zero) : (index) -> index {
%wc7 = arith.cmpi ult, %wa7, %two : index
scf.condition(%wc7) %wa7 : index
} do {
^bb0(%wb7: index):
%ld8 = memref.load %target[%zero] : memref<?xindex>
%ad8 = arith.addi %ld8, %one : index
memref.store %ad8, %target[%zero] : memref<?xindex>
%wn7 = arith.addi %wb7, %one : index
scf.yield %wn7 : index
}
}
"loom.spatial_yield"()
<{operandSegmentSizes = array<i32: 0, 0>}> : () -> ()
}) {graph_name = "g_thread_0", source_maps = []} :
(index, memref<?xindex>) -> ()
dataflow.thread.yield
}