Back to PBT run
output.generic.mlir
"builtin.module"() ({
"func.func"() <{function_type = (index) -> index, sym_name = "native_helper"}> ({
^bb0(%arg18: index):
"func.return"(%arg18) : (index) -> ()
}) : () -> ()
"dataflow.thread"() <{domain = #dataflow.thread_domain<dense>, function_type = (memref<8xindex>, memref<8xindex>, memref<4x4xindex>, index) -> (), sym_name = "t0", sym_visibility = "private"}> ({
^bb0(%arg13: memref<8xindex>, %arg14: memref<8xindex>, %arg15: memref<4x4xindex>, %arg16: index, %arg17: none):
%145 = "dataflow.graph.launch"(%arg17, %arg16, %arg14, %arg15) <{callee = @g_t0_0, operandSegmentSizes = array<i32: 1, 1, 0, 2, 0>, resultSegmentSizes = array<i32: 0, 0, 1>, source_maps = []}> : (none, index, memref<8xindex>, memref<4x4xindex>) -> none
"dataflow.thread.yield"(%145) : (none) -> ()
}) : () -> ()
"dataflow.thread"() <{domain = #dataflow.thread_domain<dense>, function_type = (memref<8xindex>, memref<8xindex>, memref<4x4xindex>, index) -> (), sym_name = "t1", sym_visibility = "private"}> ({
^bb0(%arg8: memref<8xindex>, %arg9: memref<8xindex>, %arg10: memref<4x4xindex>, %arg11: index, %arg12: none):
%142 = "arith.constant"() <{value = 0 : index}> : () -> index
%143 = "arith.constant"() <{value = 3 : index}> : () -> index
"memref.store"(%143, %arg8, %142) : (index, memref<8xindex>, index) -> ()
%144 = "dataflow.graph.launch"(%arg12, %arg11, %arg9, %arg10) <{callee = @g_t1_0, operandSegmentSizes = array<i32: 1, 1, 0, 2, 0>, resultSegmentSizes = array<i32: 0, 0, 1>, source_maps = []}> : (none, index, memref<8xindex>, memref<4x4xindex>) -> none
"dataflow.thread.yield"(%144) : (none) -> ()
}) : () -> ()
"dataflow.graph"() <{function_type = (index, memref<8xindex>, memref<4x4xindex>) -> (), input_segments = array<i32: 1, 0, 2>, result_segments = array<i32: 0, 0, 0>, sym_name = "g_t0_0", sym_visibility = "private"}> ({
^bb0(%arg4: none, %arg5: index, %arg6: memref<8xindex>, %arg7: memref<4x4xindex>):
%26 = "dataflow.constant"(%arg4) <{const_value = 0 : index}> : (none) -> index
%27 = "dataflow.constant"(%arg4) <{const_value = 1 : index}> : (none) -> index
%28 = "dataflow.constant"(%arg4) <{const_value = 4 : index}> : (none) -> index
%29 = "arith.index_cast"(%26) : (index) -> i32
%30 = "arith.index_cast"(%arg5) : (index) -> i32
%31 = "arith.index_cast"(%27) : (index) -> i32
%32:2 = "dataflow.stream"(%29, %30, %31) <{predicate = 2 : i64, step_kind = 0 : i32}> : (i32, i32, i32) -> (i32, i1)
%33 = "dataflow.carry"(%32#1, %arg4, %133#0) : (i1, none, none) -> none
%34:2 = "dataflow.demux"(%32#1, %33) : (i1, none) -> (none, none)
%35 = "dataflow.invariant"(%32#1, %arg5) : (i1, index) -> index
%36:2 = "dataflow.gate"(%32#1, %35) : (i1, index) -> (i1, index)
%37:2 = "dataflow.demux"(%36#0, %36#1) : (i1, index) -> (index, index)
%38 = "dataflow.invariant"(%32#1, %27) : (i1, index) -> index
%39:2 = "dataflow.gate"(%32#1, %38) : (i1, index) -> (i1, index)
%40:2 = "dataflow.demux"(%39#0, %39#1) : (i1, index) -> (index, index)
%41 = "dataflow.invariant"(%32#1, %26) : (i1, index) -> index
%42:2 = "dataflow.gate"(%32#1, %41) : (i1, index) -> (i1, index)
%43:2 = "dataflow.demux"(%42#0, %42#1) : (i1, index) -> (index, index)
%44 = "dataflow.invariant"(%32#1, %28) : (i1, index) -> index
%45:2 = "dataflow.gate"(%32#1, %44) : (i1, index) -> (i1, index)
%46:2 = "dataflow.demux"(%45#0, %45#1) : (i1, index) -> (index, index)
%47 = "dataflow.carry"(%32#1, %arg4, %134#0) : (i1, none, none) -> none
%48:2 = "dataflow.demux"(%32#1, %47) : (i1, none) -> (none, none)
%49 = "dataflow.constant"(%34#1) <{const_value = 0 : index}> : (none) -> index
%50 = "dataflow.carry"(%53, %34#1, %55#1) : (i1, none, none) -> none
%51 = "dataflow.carry"(%53, %42#1, %61) : (i1, index, index) -> index
%52 = "dataflow.invariant"(%53, %36#1) : (i1, index) -> index
%53 = "arith.cmpi"(%51, %52) <{predicate = 2 : i64}> : (index, index) -> i1
%54:2 = "dataflow.demux"(%53, %50) : (i1, none) -> (none, none)
%55:2 = "dataflow.gate"(%53, %50) : (i1, none) -> (i1, none)
%56:2 = "dataflow.demux"(%55#0, %55#1) : (i1, none) -> (none, none)
%57:2 = "dataflow.demux"(%53, %51) : (i1, index) -> (index, index)
%58 = "dataflow.invariant"(%53, %39#1) : (i1, index) -> index
%59:2 = "dataflow.gate"(%53, %58) : (i1, index) -> (i1, index)
%60:2 = "dataflow.demux"(%59#0, %59#1) : (i1, index) -> (index, index)
%61 = "arith.addi"(%57#1, %59#1) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
%62:2 = "dataflow.demux"(%55#0, %55#1) : (i1, none) -> (none, none)
%63:3 = "dataflow.sync"(%62#0, %56#0, %60#0) : (none, none, index) -> (none, none, index)
%64 = "dataflow.mux"(%55#0, %63#0, %62#1) : (i1, none, none) -> none
%65 = "dataflow.carry"(%53, %34#1, %64) : (i1, none, none) -> none
%66:2 = "dataflow.demux"(%53, %65) : (i1, none) -> (none, none)
%67:2 = "dataflow.sync"(%54#0, %66#0) : (none, none) -> (none, none)
%68:2 = "dataflow.sync"(%67#0, %48#1) : (none, none) -> (none, none)
%69 = "dataflow.store"(%arg6, %49, %57#0, %68#0) : (memref<8xindex>, index, index, none) -> none
%70 = "dataflow.constant"(%34#1) <{const_value = 1 : index}> : (none) -> index
%71 = "dataflow.carry"(%74, %34#1, %76#1) : (i1, none, none) -> none
%72 = "dataflow.carry"(%74, %42#1, %82) : (i1, index, index) -> index
%73 = "dataflow.invariant"(%74, %36#1) : (i1, index) -> index
%74 = "arith.cmpi"(%72, %73) <{predicate = 2 : i64}> : (index, index) -> i1
%75:2 = "dataflow.demux"(%74, %71) : (i1, none) -> (none, none)
%76:2 = "dataflow.gate"(%74, %71) : (i1, none) -> (i1, none)
%77:2 = "dataflow.demux"(%76#0, %76#1) : (i1, none) -> (none, none)
%78:2 = "dataflow.demux"(%74, %72) : (i1, index) -> (index, index)
%79 = "dataflow.invariant"(%74, %39#1) : (i1, index) -> index
%80:2 = "dataflow.gate"(%74, %79) : (i1, index) -> (i1, index)
%81:2 = "dataflow.demux"(%80#0, %80#1) : (i1, index) -> (index, index)
%82 = "arith.addi"(%78#1, %80#1) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
%83:2 = "dataflow.demux"(%76#0, %76#1) : (i1, none) -> (none, none)
%84:3 = "dataflow.sync"(%83#0, %77#0, %81#0) : (none, none, index) -> (none, none, index)
%85 = "dataflow.mux"(%76#0, %84#0, %83#1) : (i1, none, none) -> none
%86 = "dataflow.carry"(%74, %34#1, %85) : (i1, none, none) -> none
%87:2 = "dataflow.demux"(%74, %86) : (i1, none) -> (none, none)
%88:2 = "dataflow.sync"(%75#0, %87#0) : (none, none) -> (none, none)
%89:2 = "dataflow.sync"(%88#0, %48#1) : (none, none) -> (none, none)
%90 = "dataflow.store"(%arg6, %70, %78#0, %89#0) : (memref<8xindex>, index, index, none) -> none
%91 = "dataflow.constant"(%34#1) <{const_value = 2 : index}> : (none) -> index
%92 = "dataflow.carry"(%95, %34#1, %97#1) : (i1, none, none) -> none
%93 = "dataflow.carry"(%95, %42#1, %103) : (i1, index, index) -> index
%94 = "dataflow.invariant"(%95, %36#1) : (i1, index) -> index
%95 = "arith.cmpi"(%93, %94) <{predicate = 2 : i64}> : (index, index) -> i1
%96:2 = "dataflow.demux"(%95, %92) : (i1, none) -> (none, none)
%97:2 = "dataflow.gate"(%95, %92) : (i1, none) -> (i1, none)
%98:2 = "dataflow.demux"(%97#0, %97#1) : (i1, none) -> (none, none)
%99:2 = "dataflow.demux"(%95, %93) : (i1, index) -> (index, index)
%100 = "dataflow.invariant"(%95, %39#1) : (i1, index) -> index
%101:2 = "dataflow.gate"(%95, %100) : (i1, index) -> (i1, index)
%102:2 = "dataflow.demux"(%101#0, %101#1) : (i1, index) -> (index, index)
%103 = "arith.addi"(%99#1, %101#1) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
%104:2 = "dataflow.demux"(%97#0, %97#1) : (i1, none) -> (none, none)
%105:3 = "dataflow.sync"(%104#0, %98#0, %102#0) : (none, none, index) -> (none, none, index)
%106 = "dataflow.mux"(%97#0, %105#0, %104#1) : (i1, none, none) -> none
%107 = "dataflow.carry"(%95, %34#1, %106) : (i1, none, none) -> none
%108:2 = "dataflow.demux"(%95, %107) : (i1, none) -> (none, none)
%109:2 = "dataflow.sync"(%96#0, %108#0) : (none, none) -> (none, none)
%110:2 = "dataflow.sync"(%109#0, %48#1) : (none, none) -> (none, none)
%111 = "dataflow.store"(%arg6, %91, %99#0, %110#0) : (memref<8xindex>, index, index, none) -> none
%112 = "dataflow.constant"(%34#1) <{const_value = 3 : index}> : (none) -> index
%113 = "dataflow.carry"(%116, %34#1, %118#1) : (i1, none, none) -> none
%114 = "dataflow.carry"(%116, %42#1, %124) : (i1, index, index) -> index
%115 = "dataflow.invariant"(%116, %36#1) : (i1, index) -> index
%116 = "arith.cmpi"(%114, %115) <{predicate = 2 : i64}> : (index, index) -> i1
%117:2 = "dataflow.demux"(%116, %113) : (i1, none) -> (none, none)
%118:2 = "dataflow.gate"(%116, %113) : (i1, none) -> (i1, none)
%119:2 = "dataflow.demux"(%118#0, %118#1) : (i1, none) -> (none, none)
%120:2 = "dataflow.demux"(%116, %114) : (i1, index) -> (index, index)
%121 = "dataflow.invariant"(%116, %39#1) : (i1, index) -> index
%122:2 = "dataflow.gate"(%116, %121) : (i1, index) -> (i1, index)
%123:2 = "dataflow.demux"(%122#0, %122#1) : (i1, index) -> (index, index)
%124 = "arith.addi"(%120#1, %122#1) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
%125:2 = "dataflow.demux"(%118#0, %118#1) : (i1, none) -> (none, none)
%126:3 = "dataflow.sync"(%125#0, %119#0, %123#0) : (none, none, index) -> (none, none, index)
%127 = "dataflow.mux"(%118#0, %126#0, %125#1) : (i1, none, none) -> none
%128 = "dataflow.carry"(%116, %34#1, %127) : (i1, none, none) -> none
%129:2 = "dataflow.demux"(%116, %128) : (i1, none) -> (none, none)
%130:2 = "dataflow.sync"(%117#0, %129#0) : (none, none) -> (none, none)
%131:2 = "dataflow.sync"(%130#0, %48#1) : (none, none) -> (none, none)
%132 = "dataflow.store"(%arg6, %112, %120#0, %131#0) : (memref<8xindex>, index, index, none) -> none
%133:4 = "dataflow.sync"(%67#0, %88#0, %109#0, %130#0) : (none, none, none, none) -> (none, none, none, none)
%134:4 = "dataflow.sync"(%69, %90, %111, %132) : (none, none, none, none) -> (none, none, none, none)
%135 = "arith.cmpi"(%29, %30) <{predicate = 2 : i64}> : (i32, i32) -> i1
%136:2 = "dataflow.demux"(%135, %34#0) : (i1, none) -> (none, none)
%137:2 = "dataflow.sync"(%136#1, %37#0) : (none, index) -> (none, index)
%138:2 = "dataflow.sync"(%137#0, %40#0) : (none, index) -> (none, index)
%139:2 = "dataflow.sync"(%43#0, %46#0) : (index, index) -> (index, index)
%140:2 = "dataflow.sync"(%138#0, %139#0) : (none, index) -> (none, index)
%141 = "dataflow.mux"(%135, %136#0, %140#0) : (i1, none, none) -> none
"dataflow.graph.return"(%141, %48#0) <{operandSegmentSizes = array<i32: 0, 0, 0, 2>}> : (none, none) -> ()
}) : () -> ()
"dataflow.graph"() <{function_type = (index, memref<8xindex>, memref<4x4xindex>) -> (), input_segments = array<i32: 1, 0, 2>, result_segments = array<i32: 0, 0, 0>, sym_name = "g_t1_0", sym_visibility = "private"}> ({
^bb0(%arg0: none, %arg1: index, %arg2: memref<8xindex>, %arg3: memref<4x4xindex>):
%0 = "dataflow.constant"(%arg0) <{const_value = 0 : index}> : (none) -> index
%1 = "dataflow.constant"(%arg0) <{const_value = 1 : index}> : (none) -> index
%2 = "dataflow.constant"(%arg0) <{const_value = 7 : index}> : (none) -> index
%3 = "arith.index_cast"(%0) : (index) -> i32
%4 = "arith.index_cast"(%arg1) : (index) -> i32
%5 = "arith.index_cast"(%1) : (index) -> i32
%6:2 = "dataflow.stream"(%3, %4, %5) <{predicate = 2 : i64, step_kind = 0 : i32}> : (i32, i32, i32) -> (i32, i1)
%7 = "dataflow.carry"(%6#1, %arg0, %8#1) : (i1, none, none) -> none
%8:2 = "dataflow.demux"(%6#1, %7) : (i1, none) -> (none, none)
%9 = "dataflow.invariant"(%6#1, %2) : (i1, index) -> index
%10:2 = "dataflow.gate"(%6#1, %9) : (i1, index) -> (i1, index)
%11:2 = "dataflow.demux"(%10#0, %10#1) : (i1, index) -> (index, index)
%12 = "dataflow.invariant"(%6#1, %0) : (i1, index) -> index
%13:2 = "dataflow.gate"(%6#1, %12) : (i1, index) -> (i1, index)
%14:2 = "dataflow.demux"(%13#0, %13#1) : (i1, index) -> (index, index)
%15 = "dataflow.carry"(%6#1, %arg0, %21) : (i1, none, none) -> none
%16:2 = "dataflow.demux"(%6#1, %15) : (i1, none) -> (none, none)
%17:2 = "dataflow.sync"(%8#1, %16#1) : (none, none) -> (none, none)
%18 = "dataflow.constant"(%8#1) <{const_value = 4 : index}> : (none) -> index
%19 = "arith.muli"(%13#1, %18) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
%20 = "arith.addi"(%19, %13#1) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
%21 = "dataflow.store"(%arg3, %20, %10#1, %17#0) : (memref<4x4xindex>, index, index, none) -> none
%22 = "arith.cmpi"(%3, %4) <{predicate = 2 : i64}> : (i32, i32) -> i1
%23:2 = "dataflow.demux"(%22, %8#0) : (i1, none) -> (none, none)
%24:3 = "dataflow.sync"(%23#1, %11#0, %14#0) : (none, index, index) -> (none, index, index)
%25 = "dataflow.mux"(%22, %23#0, %24#0) : (i1, none, none) -> none
"dataflow.graph.return"(%25, %16#0) <{operandSegmentSizes = array<i32: 0, 0, 0, 2>}> : (none, none) -> ()
}) : () -> ()
}) : () -> ()