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.if %flag {
%ld1 = memref.load %target[%zero] : memref<?xindex>
%ad1 = arith.addi %ld1, %one : index
memref.store %ad1, %target[%zero] : memref<?xindex>
} else {
memref.store %one, %target[%zero] : memref<?xindex>
%wr3 = scf.while (%wa3 = %zero) : (index) -> index {
%wc3 = arith.cmpi ult, %wa3, %two : index
scf.condition(%wc3) %wa3 : index
} do {
^bb0(%wb3: index):
%ld4 = memref.load %target[%zero] : memref<?xindex>
%ad4 = arith.addi %ld4, %one : index
memref.store %ad4, %target[%zero] : memref<?xindex>
%wn3 = arith.addi %wb3, %one : index
scf.yield %wn3 : index
}
}
scf.parallel (%iv5) = (%zero) to (%two) step (%one) {
%mx5 = arith.muli %zero, %two : index
%ix5 = arith.addi %mx5, %iv5 : index
%wr6 = scf.while (%wa6 = %zero) : (index) -> index {
%wc6 = arith.cmpi ult, %wa6, %two : index
scf.condition(%wc6) %wa6 : index
} do {
^bb0(%wb6: index):
memref.store %one, %target[%ix5] : memref<?xindex>
%wn6 = arith.addi %wb6, %one : index
scf.yield %wn6 : index
}
%ld8 = memref.load %target[%ix5] : memref<?xindex>
%ad8 = arith.addi %ld8, %one : index
memref.store %ad8, %target[%ix5] : memref<?xindex>
scf.reduce
}
"loom.spatial_yield"()
<{operandSegmentSizes = array<i32: 0, 0>}> : () -> ()
}) {graph_name = "g_thread_0", source_maps = []} :
(index, memref<?xindex>) -> ()
dataflow.thread.yield
}