Back to PBT run

output.generic.mlir

"builtin.module"() ({
  "llvm.func"() <{CConv = #llvm.cconv<ccc>, function_type = !llvm.func<void (i64)>, linkage = #llvm.linkage<external>, sym_name = "imported_kernel", unnamed_addr = 0 : i64, visibility_ = 0 : i64}> ({
  }) : () -> ()
  "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):
    %203 = "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"(%203) : (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):
    %202 = "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"(%202) : (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>):
    %44 = "dataflow.constant"(%arg4) <{const_value = 0 : index}> : (none) -> index
    %45 = "dataflow.constant"(%arg4) <{const_value = 1 : index}> : (none) -> index
    %46 = "dataflow.constant"(%arg4) <{const_value = 4 : index}> : (none) -> index
    %47 = "dataflow.constant"(%arg4) <{const_value = 7 : index}> : (none) -> index
    %48 = "arith.index_cast"(%44) : (index) -> i32
    %49 = "arith.index_cast"(%arg5) : (index) -> i32
    %50 = "arith.index_cast"(%45) : (index) -> i32
    %51:2 = "dataflow.stream"(%48, %49, %50) <{predicate = 2 : i64, step_kind = 0 : i32}> : (i32, i32, i32) -> (i32, i1)
    %52 = "dataflow.carry"(%51#1, %arg4, %53#1) : (i1, none, none) -> none
    %53:2 = "dataflow.demux"(%51#1, %52) : (i1, none) -> (none, none)
    %54 = "dataflow.invariant"(%51#1, %47) : (i1, index) -> index
    %55:2 = "dataflow.gate"(%51#1, %54) : (i1, index) -> (i1, index)
    %56:2 = "dataflow.demux"(%55#0, %55#1) : (i1, index) -> (index, index)
    %57 = "dataflow.invariant"(%51#1, %44) : (i1, index) -> index
    %58:2 = "dataflow.gate"(%51#1, %57) : (i1, index) -> (i1, index)
    %59:2 = "dataflow.demux"(%58#0, %58#1) : (i1, index) -> (index, index)
    %60 = "dataflow.invariant"(%51#1, %46) : (i1, index) -> index
    %61:2 = "dataflow.gate"(%51#1, %60) : (i1, index) -> (i1, index)
    %62:2 = "dataflow.demux"(%61#0, %61#1) : (i1, index) -> (index, index)
    %63 = "dataflow.invariant"(%51#1, %45) : (i1, index) -> index
    %64:2 = "dataflow.gate"(%51#1, %63) : (i1, index) -> (i1, index)
    %65:2 = "dataflow.demux"(%64#0, %64#1) : (i1, index) -> (index, index)
    %66 = "dataflow.carry"(%51#1, %arg4, %194#0) : (i1, none, none) -> none
    %67:2 = "dataflow.demux"(%51#1, %66) : (i1, none) -> (none, none)
    %68 = "dataflow.constant"(%53#1) <{const_value = 0 : index}> : (none) -> index
    %69 = "dataflow.constant"(%53#1) <{const_value = 0 : index}> : (none) -> index
    %70:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %71 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %72 = "arith.muli"(%68, %71) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %73 = "arith.addi"(%72, %69) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %74 = "dataflow.store"(%arg7, %73, %55#1, %70#0) : (memref<4x4xindex>, index, index, none) -> none
    %75 = "dataflow.constant"(%53#1) <{const_value = 0 : index}> : (none) -> index
    %76 = "dataflow.constant"(%53#1) <{const_value = 1 : index}> : (none) -> index
    %77:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %78 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %79 = "arith.muli"(%75, %78) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %80 = "arith.addi"(%79, %76) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %81 = "dataflow.store"(%arg7, %80, %55#1, %77#0) : (memref<4x4xindex>, index, index, none) -> none
    %82 = "dataflow.constant"(%53#1) <{const_value = 0 : index}> : (none) -> index
    %83 = "dataflow.constant"(%53#1) <{const_value = 2 : index}> : (none) -> index
    %84:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %85 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %86 = "arith.muli"(%82, %85) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %87 = "arith.addi"(%86, %83) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %88 = "dataflow.store"(%arg7, %87, %55#1, %84#0) : (memref<4x4xindex>, index, index, none) -> none
    %89 = "dataflow.constant"(%53#1) <{const_value = 0 : index}> : (none) -> index
    %90 = "dataflow.constant"(%53#1) <{const_value = 3 : index}> : (none) -> index
    %91:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %92 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %93 = "arith.muli"(%89, %92) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %94 = "arith.addi"(%93, %90) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %95 = "dataflow.store"(%arg7, %94, %55#1, %91#0) : (memref<4x4xindex>, index, index, none) -> none
    %96 = "dataflow.constant"(%53#1) <{const_value = 1 : index}> : (none) -> index
    %97 = "dataflow.constant"(%53#1) <{const_value = 0 : index}> : (none) -> index
    %98:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %99 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %100 = "arith.muli"(%96, %99) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %101 = "arith.addi"(%100, %97) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %102 = "dataflow.store"(%arg7, %101, %55#1, %98#0) : (memref<4x4xindex>, index, index, none) -> none
    %103 = "dataflow.constant"(%53#1) <{const_value = 1 : index}> : (none) -> index
    %104 = "dataflow.constant"(%53#1) <{const_value = 1 : index}> : (none) -> index
    %105:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %106 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %107 = "arith.muli"(%103, %106) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %108 = "arith.addi"(%107, %104) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %109 = "dataflow.store"(%arg7, %108, %55#1, %105#0) : (memref<4x4xindex>, index, index, none) -> none
    %110 = "dataflow.constant"(%53#1) <{const_value = 1 : index}> : (none) -> index
    %111 = "dataflow.constant"(%53#1) <{const_value = 2 : index}> : (none) -> index
    %112:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %113 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %114 = "arith.muli"(%110, %113) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %115 = "arith.addi"(%114, %111) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %116 = "dataflow.store"(%arg7, %115, %55#1, %112#0) : (memref<4x4xindex>, index, index, none) -> none
    %117 = "dataflow.constant"(%53#1) <{const_value = 1 : index}> : (none) -> index
    %118 = "dataflow.constant"(%53#1) <{const_value = 3 : index}> : (none) -> index
    %119:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %120 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %121 = "arith.muli"(%117, %120) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %122 = "arith.addi"(%121, %118) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %123 = "dataflow.store"(%arg7, %122, %55#1, %119#0) : (memref<4x4xindex>, index, index, none) -> none
    %124 = "dataflow.constant"(%53#1) <{const_value = 2 : index}> : (none) -> index
    %125 = "dataflow.constant"(%53#1) <{const_value = 0 : index}> : (none) -> index
    %126:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %127 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %128 = "arith.muli"(%124, %127) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %129 = "arith.addi"(%128, %125) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %130 = "dataflow.store"(%arg7, %129, %55#1, %126#0) : (memref<4x4xindex>, index, index, none) -> none
    %131 = "dataflow.constant"(%53#1) <{const_value = 2 : index}> : (none) -> index
    %132 = "dataflow.constant"(%53#1) <{const_value = 1 : index}> : (none) -> index
    %133:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %134 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %135 = "arith.muli"(%131, %134) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %136 = "arith.addi"(%135, %132) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %137 = "dataflow.store"(%arg7, %136, %55#1, %133#0) : (memref<4x4xindex>, index, index, none) -> none
    %138 = "dataflow.constant"(%53#1) <{const_value = 2 : index}> : (none) -> index
    %139 = "dataflow.constant"(%53#1) <{const_value = 2 : index}> : (none) -> index
    %140:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %141 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %142 = "arith.muli"(%138, %141) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %143 = "arith.addi"(%142, %139) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %144 = "dataflow.store"(%arg7, %143, %55#1, %140#0) : (memref<4x4xindex>, index, index, none) -> none
    %145 = "dataflow.constant"(%53#1) <{const_value = 2 : index}> : (none) -> index
    %146 = "dataflow.constant"(%53#1) <{const_value = 3 : index}> : (none) -> index
    %147:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %148 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %149 = "arith.muli"(%145, %148) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %150 = "arith.addi"(%149, %146) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %151 = "dataflow.store"(%arg7, %150, %55#1, %147#0) : (memref<4x4xindex>, index, index, none) -> none
    %152 = "dataflow.constant"(%53#1) <{const_value = 3 : index}> : (none) -> index
    %153 = "dataflow.constant"(%53#1) <{const_value = 0 : index}> : (none) -> index
    %154:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %155 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %156 = "arith.muli"(%152, %155) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %157 = "arith.addi"(%156, %153) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %158 = "dataflow.store"(%arg7, %157, %55#1, %154#0) : (memref<4x4xindex>, index, index, none) -> none
    %159 = "dataflow.constant"(%53#1) <{const_value = 3 : index}> : (none) -> index
    %160 = "dataflow.constant"(%53#1) <{const_value = 1 : index}> : (none) -> index
    %161:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %162 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %163 = "arith.muli"(%159, %162) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %164 = "arith.addi"(%163, %160) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %165 = "dataflow.store"(%arg7, %164, %55#1, %161#0) : (memref<4x4xindex>, index, index, none) -> none
    %166 = "dataflow.constant"(%53#1) <{const_value = 3 : index}> : (none) -> index
    %167 = "dataflow.constant"(%53#1) <{const_value = 2 : index}> : (none) -> index
    %168:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %169 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %170 = "arith.muli"(%166, %169) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %171 = "arith.addi"(%170, %167) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %172 = "dataflow.store"(%arg7, %171, %55#1, %168#0) : (memref<4x4xindex>, index, index, none) -> none
    %173 = "dataflow.constant"(%53#1) <{const_value = 3 : index}> : (none) -> index
    %174 = "dataflow.constant"(%53#1) <{const_value = 3 : index}> : (none) -> index
    %175:2 = "dataflow.sync"(%53#1, %67#1) : (none, none) -> (none, none)
    %176 = "dataflow.constant"(%53#1) <{const_value = 4 : index}> : (none) -> index
    %177 = "arith.muli"(%173, %176) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %178 = "arith.addi"(%177, %174) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %179 = "dataflow.store"(%arg7, %178, %55#1, %175#0) : (memref<4x4xindex>, index, index, none) -> none
    %180:2 = "dataflow.sync"(%74, %81) : (none, none) -> (none, none)
    %181:2 = "dataflow.sync"(%88, %95) : (none, none) -> (none, none)
    %182:2 = "dataflow.sync"(%180#0, %181#0) : (none, none) -> (none, none)
    %183:2 = "dataflow.sync"(%102, %109) : (none, none) -> (none, none)
    %184:2 = "dataflow.sync"(%116, %123) : (none, none) -> (none, none)
    %185:2 = "dataflow.sync"(%183#0, %184#0) : (none, none) -> (none, none)
    %186:2 = "dataflow.sync"(%182#0, %185#0) : (none, none) -> (none, none)
    %187:2 = "dataflow.sync"(%130, %137) : (none, none) -> (none, none)
    %188:2 = "dataflow.sync"(%144, %151) : (none, none) -> (none, none)
    %189:2 = "dataflow.sync"(%187#0, %188#0) : (none, none) -> (none, none)
    %190:2 = "dataflow.sync"(%158, %165) : (none, none) -> (none, none)
    %191:2 = "dataflow.sync"(%172, %179) : (none, none) -> (none, none)
    %192:2 = "dataflow.sync"(%190#0, %191#0) : (none, none) -> (none, none)
    %193:2 = "dataflow.sync"(%189#0, %192#0) : (none, none) -> (none, none)
    %194:2 = "dataflow.sync"(%186#0, %193#0) : (none, none) -> (none, none)
    %195 = "arith.cmpi"(%48, %49) <{predicate = 2 : i64}> : (i32, i32) -> i1
    %196:2 = "dataflow.demux"(%195, %53#0) : (i1, none) -> (none, none)
    %197:2 = "dataflow.sync"(%196#1, %56#0) : (none, index) -> (none, index)
    %198:2 = "dataflow.sync"(%197#0, %59#0) : (none, index) -> (none, index)
    %199:2 = "dataflow.sync"(%62#0, %65#0) : (index, index) -> (index, index)
    %200:2 = "dataflow.sync"(%198#0, %199#0) : (none, index) -> (none, index)
    %201 = "dataflow.mux"(%195, %196#0, %200#0) : (i1, none, none) -> none
    "dataflow.graph.return"(%201, %67#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.carry"(%5, %arg0, %7#1) : (i1, none, none) -> none
    %3 = "dataflow.carry"(%5, %0, %13) : (i1, index, index) -> index
    %4 = "dataflow.invariant"(%5, %arg1) : (i1, index) -> index
    %5 = "arith.cmpi"(%3, %4) <{predicate = 2 : i64}> : (index, index) -> i1
    %6:2 = "dataflow.demux"(%5, %2) : (i1, none) -> (none, none)
    %7:2 = "dataflow.gate"(%5, %2) : (i1, none) -> (i1, none)
    %8:2 = "dataflow.demux"(%7#0, %7#1) : (i1, none) -> (none, none)
    %9:2 = "dataflow.demux"(%5, %3) : (i1, index) -> (index, index)
    %10 = "dataflow.invariant"(%5, %1) : (i1, index) -> index
    %11:2 = "dataflow.gate"(%5, %10) : (i1, index) -> (i1, index)
    %12:2 = "dataflow.demux"(%11#0, %11#1) : (i1, index) -> (index, index)
    %13 = "arith.addi"(%9#1, %11#1) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %14:2 = "dataflow.demux"(%7#0, %7#1) : (i1, none) -> (none, none)
    %15:3 = "dataflow.sync"(%14#0, %8#0, %12#0) : (none, none, index) -> (none, none, index)
    %16 = "dataflow.mux"(%7#0, %15#0, %14#1) : (i1, none, none) -> none
    %17 = "dataflow.carry"(%5, %arg0, %16) : (i1, none, none) -> none
    %18:2 = "dataflow.demux"(%5, %17) : (i1, none) -> (none, none)
    %19:2 = "dataflow.sync"(%6#0, %18#0) : (none, none) -> (none, none)
    %20:2 = "dataflow.sync"(%19#0, %arg0) : (none, none) -> (none, none)
    %21 = "dataflow.store"(%arg2, %0, %9#0, %20#0) : (memref<8xindex>, index, index, none) -> none
    %22 = "dataflow.carry"(%25, %arg0, %27#1) : (i1, none, none) -> none
    %23 = "dataflow.carry"(%25, %0, %33) : (i1, index, index) -> index
    %24 = "dataflow.invariant"(%25, %arg1) : (i1, index) -> index
    %25 = "arith.cmpi"(%23, %24) <{predicate = 2 : i64}> : (index, index) -> i1
    %26:2 = "dataflow.demux"(%25, %22) : (i1, none) -> (none, none)
    %27:2 = "dataflow.gate"(%25, %22) : (i1, none) -> (i1, none)
    %28:2 = "dataflow.demux"(%27#0, %27#1) : (i1, none) -> (none, none)
    %29:2 = "dataflow.demux"(%25, %23) : (i1, index) -> (index, index)
    %30 = "dataflow.invariant"(%25, %1) : (i1, index) -> index
    %31:2 = "dataflow.gate"(%25, %30) : (i1, index) -> (i1, index)
    %32:2 = "dataflow.demux"(%31#0, %31#1) : (i1, index) -> (index, index)
    %33 = "arith.addi"(%29#1, %31#1) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
    %34:2 = "dataflow.demux"(%27#0, %27#1) : (i1, none) -> (none, none)
    %35:3 = "dataflow.sync"(%34#0, %28#0, %32#0) : (none, none, index) -> (none, none, index)
    %36 = "dataflow.mux"(%27#0, %35#0, %34#1) : (i1, none, none) -> none
    %37 = "dataflow.carry"(%25, %arg0, %36) : (i1, none, none) -> none
    %38:2 = "dataflow.demux"(%25, %37) : (i1, none) -> (none, none)
    %39:2 = "dataflow.sync"(%26#0, %38#0) : (none, none) -> (none, none)
    %40:2 = "dataflow.sync"(%39#0, %arg0) : (none, none) -> (none, none)
    %41 = "dataflow.store"(%arg2, %1, %29#0, %40#0) : (memref<8xindex>, index, index, none) -> none
    %42:2 = "dataflow.sync"(%19#0, %39#0) : (none, none) -> (none, none)
    %43:2 = "dataflow.sync"(%21, %41) : (none, none) -> (none, none)
    "dataflow.graph.return"(%42#0, %43#0) <{operandSegmentSizes = array<i32: 0, 0, 0, 2>}> : (none, none) -> ()
  }) : () -> ()
}) : () -> ()