Back to PBT run
test.patch
diff --git a/test/raise/spectriad-mlir-stage-06-v1.mlir b/test/raise/spectriad-mlir-stage-06-v1.mlir
new file mode 100644
--- /dev/null
+++ b/test/raise/spectriad-mlir-stage-06-v1.mlir
@@ -0,0 +1,271 @@
+// RUN: loom-raise-opt --loom-lower-scf-to-dfg --mlir-print-op-generic %s | FileCheck %s
+
+llvm.func @imported_callable(i64) -> i64
+func.func private @native_callable(%arg_a: i32) -> i32 {
+ return %arg_a : i32
+}
+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.parallel (%iv0) = (%zero) to (%two) step (%one) {
+ %mx0 = arith.muli %zero, %two : index
+ %ix0 = arith.addi %mx0, %iv0 : index
+ scf.parallel (%iv1) = (%zero) to (%two) step (%one) {
+ %mx1 = arith.muli %ix0, %two : index
+ %ix1 = arith.addi %mx1, %iv1 : index
+ %ld2 = memref.load %target[%ix1] : memref<?xindex>
+ %ad2 = arith.addi %ld2, %one : index
+ memref.store %ad2, %target[%ix1] : memref<?xindex>
+ scf.reduce
+ }
+ scf.reduce
+ }
+ "loom.spatial_yield"()
+ <{operandSegmentSizes = array<i32: 0, 0>}> : () -> ()
+ }) {graph_name = "g_thread_0", source_maps = []} :
+ (index, memref<?xindex>) -> ()
+ dataflow.thread.yield
+}
+dataflow.thread private @thread_1 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
+ %wr3 = scf.while (%wa3 = %zero) : (index) -> index {
+ %wc3 = arith.cmpi ult, %wa3, %two : index
+ scf.condition(%wc3) %wa3 : index
+ } do {
+ ^bb0(%wb3: index):
+ %wr4 = scf.while (%wa4 = %zero) : (index) -> index {
+ %wc4 = arith.cmpi ult, %wa4, %two : index
+ scf.condition(%wc4) %wa4 : index
+ } do {
+ ^bb0(%wb4: index):
+ memref.store %one, %target[%zero] : memref<?xindex>
+ %ld6 = memref.load %target[%zero] : memref<?xindex>
+ %ad6 = arith.addi %ld6, %one : index
+ memref.store %ad6, %target[%zero] : memref<?xindex>
+ %wn4 = arith.addi %wb4, %one : index
+ scf.yield %wn4 : index
+ }
+ memref.store %one, %target[%zero] : memref<?xindex>
+ %wn3 = arith.addi %wb3, %one : index
+ scf.yield %wn3 : index
+ }
+ %wr8 = scf.while (%wa8 = %zero) : (index) -> index {
+ %wc8 = arith.cmpi ult, %wa8, %two : index
+ scf.condition(%wc8) %wa8 : index
+ } do {
+ ^bb0(%wb8: index):
+ %wr9 = scf.while (%wa9 = %zero) : (index) -> index {
+ %wc9 = arith.cmpi ult, %wa9, %two : index
+ scf.condition(%wc9) %wa9 : index
+ } do {
+ ^bb0(%wb9: index):
+ memref.store %one, %target[%zero] : memref<?xindex>
+ %wn9 = arith.addi %wb9, %one : index
+ scf.yield %wn9 : index
+ }
+ %ld11 = memref.load %target[%zero] : memref<?xindex>
+ %ad11 = arith.addi %ld11, %one : index
+ memref.store %ad11, %target[%zero] : memref<?xindex>
+ %wn8 = arith.addi %wb8, %one : index
+ scf.yield %wn8 : index
+ }
+ "loom.spatial_yield"()
+ <{operandSegmentSizes = array<i32: 0, 0>}> : () -> ()
+ }) {graph_name = "g_thread_1", source_maps = []} :
+ (index, memref<?xindex>) -> ()
+ dataflow.thread.yield
+}
+
+// CHECK: "builtin.module"() ({
+// CHECK-NEXT: "llvm.func"() <{CConv = #llvm.cconv<ccc>, function_type = !llvm.func<i64 (i64)>, linkage = #llvm.linkage<external>, sym_name = "imported_callable", unnamed_addr = 0 : i64, visibility_ = 0 : i64}> ({
+// CHECK-NEXT: }) : () -> ()
+// CHECK-NEXT: "func.func"() <{function_type = (i32) -> i32, sym_name = "native_callable", sym_visibility = "private"}> ({
+// CHECK-NEXT: ^bb0(%arg12: i32):
+// CHECK-NEXT: "func.return"(%arg12) : (i32) -> ()
+// CHECK-NEXT: }) : () -> ()
+// CHECK-NEXT: "dataflow.thread"() <{domain = #dataflow.thread_domain<dense>, function_type = (index, memref<?xindex>) -> (), sym_name = "thread_0", sym_visibility = "private"}> ({
+// CHECK-NEXT: ^bb0(%arg9: index, %arg10: memref<?xindex>, %arg11: none):
+// CHECK-NEXT: %152 = "dataflow.graph.launch"(%arg11, %arg9, %arg10) <{callee = @g_thread_0, operandSegmentSizes = array<i32: 1, 1, 0, 1, 0>, resultSegmentSizes = array<i32: 0, 0, 1>, source_maps = []}> : (none, index, memref<?xindex>) -> none
+// CHECK-NEXT: "dataflow.thread.yield"(%152) : (none) -> ()
+// CHECK-NEXT: }) : () -> ()
+// CHECK-NEXT: "dataflow.thread"() <{domain = #dataflow.thread_domain<dense>, function_type = (index, memref<?xindex>) -> (), sym_name = "thread_1", sym_visibility = "private"}> ({
+// CHECK-NEXT: ^bb0(%arg6: index, %arg7: memref<?xindex>, %arg8: none):
+// CHECK-NEXT: %151 = "dataflow.graph.launch"(%arg8, %arg6, %arg7) <{callee = @g_thread_1, operandSegmentSizes = array<i32: 1, 1, 0, 1, 0>, resultSegmentSizes = array<i32: 0, 0, 1>, source_maps = []}> : (none, index, memref<?xindex>) -> none
+// CHECK-NEXT: "dataflow.thread.yield"(%151) : (none) -> ()
+// CHECK-NEXT: }) : () -> ()
+// CHECK-NEXT: "dataflow.graph"() <{arg_attrs = [{}, {llvm.noalias}], function_type = (index, memref<?xindex>) -> (), input_segments = array<i32: 1, 0, 1>, result_segments = array<i32: 0, 0, 0>, sym_name = "g_thread_0", sym_visibility = "private"}> ({
+// CHECK-NEXT: ^bb0(%arg3: none, %arg4: index, %arg5: memref<?xindex>):
+// CHECK-NEXT: %127 = "dataflow.constant"(%arg3) <{const_value = 1 : index}> : (none) -> index
+// CHECK-NEXT: %128 = "dataflow.constant"(%arg3) <{const_value = 2 : index}> : (none) -> index
+// CHECK-NEXT: %129 = "dataflow.constant"(%arg3) <{const_value = 0 : index}> : (none) -> index
+// CHECK-NEXT: %130 = "arith.muli"(%129, %128) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %131 = "arith.addi"(%130, %129) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %132 = "arith.addi"(%142#0, %127) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %133 = "arith.muli"(%129, %128) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %134 = "arith.addi"(%133, %127) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %135 = "arith.addi"(%144#0, %127) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %136 = "arith.muli"(%127, %128) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %137 = "arith.addi"(%136, %129) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %138 = "arith.addi"(%146#0, %127) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %139 = "arith.muli"(%127, %128) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %140 = "arith.addi"(%139, %127) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %141 = "arith.addi"(%148#0, %127) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %142:2 = "dataflow.load"(%arg5, %131, %arg3) : (memref<?xindex>, index, none) -> (index, none)
+// CHECK-NEXT: %143 = "dataflow.store"(%arg5, %131, %132, %142#1) : (memref<?xindex>, index, index, none) -> none
+// CHECK-NEXT: %144:2 = "dataflow.load"(%arg5, %134, %arg3) : (memref<?xindex>, index, none) -> (index, none)
+// CHECK-NEXT: %145 = "dataflow.store"(%arg5, %134, %135, %144#1) : (memref<?xindex>, index, index, none) -> none
+// CHECK-NEXT: %146:2 = "dataflow.load"(%arg5, %137, %arg3) : (memref<?xindex>, index, none) -> (index, none)
+// CHECK-NEXT: %147 = "dataflow.store"(%arg5, %137, %138, %146#1) : (memref<?xindex>, index, index, none) -> none
+// CHECK-NEXT: %148:2 = "dataflow.load"(%arg5, %140, %arg3) : (memref<?xindex>, index, none) -> (index, none)
+// CHECK-NEXT: %149 = "dataflow.store"(%arg5, %140, %141, %148#1) : (memref<?xindex>, index, index, none) -> none
+// CHECK-NEXT: %150:4 = "dataflow.sync"(%143, %145, %147, %149) : (none, none, none, none) -> (none, none, none, none)
+// CHECK-NEXT: "dataflow.graph.return"(%150#0) <{operandSegmentSizes = array<i32: 0, 0, 0, 1>}> : (none) -> ()
+// CHECK-NEXT: }) : () -> ()
+// CHECK-NEXT: "dataflow.graph"() <{arg_attrs = [{}, {llvm.noalias}], function_type = (index, memref<?xindex>) -> (), input_segments = array<i32: 1, 0, 1>, result_segments = array<i32: 0, 0, 0>, sym_name = "g_thread_1", sym_visibility = "private"}> ({
+// CHECK-NEXT: ^bb0(%arg0: none, %arg1: index, %arg2: memref<?xindex>):
+// CHECK-NEXT: %0 = "dataflow.constant"(%arg0) <{const_value = 0 : index}> : (none) -> index
+// CHECK-NEXT: %1 = "dataflow.constant"(%arg0) <{const_value = 1 : index}> : (none) -> index
+// CHECK-NEXT: %2 = "dataflow.constant"(%arg0) <{const_value = 2 : index}> : (none) -> index
+// CHECK-NEXT: %3 = "dataflow.carry"(%8, %arg0, %51#0) : (i1, none, none) -> none
+// CHECK-NEXT: %4 = "dataflow.carry"(%8, %0, %54) : (i1, index, index) -> index
+// CHECK-NEXT: %5 = "dataflow.invariant"(%8, %2) : (i1, index) -> index
+// CHECK-NEXT: %6 = "dataflow.carry"(%8, %arg0, %53) : (i1, none, none) -> none
+// CHECK-NEXT: %7 = "dataflow.carry"(%8, %arg0, %53) : (i1, none, none) -> none
+// CHECK-NEXT: %8 = "arith.cmpi"(%4, %5) <{predicate = 6 : i64}> : (index, index) -> i1
+// CHECK-NEXT: %9:2 = "dataflow.demux"(%8, %3) : (i1, none) -> (none, none)
+// CHECK-NEXT: %10:2 = "dataflow.gate"(%8, %3) : (i1, none) -> (i1, none)
+// CHECK-NEXT: %11:2 = "dataflow.demux"(%10#0, %10#1) : (i1, none) -> (none, none)
+// CHECK-NEXT: %12:2 = "dataflow.demux"(%8, %6) : (i1, none) -> (none, none)
+// CHECK-NEXT: %13:2 = "dataflow.demux"(%8, %7) : (i1, none) -> (none, none)
+// CHECK-NEXT: %14:2 = "dataflow.demux"(%8, %4) : (i1, index) -> (index, index)
+// CHECK-NEXT: %15 = "dataflow.invariant"(%8, %2) : (i1, index) -> index
+// CHECK-NEXT: %16:2 = "dataflow.gate"(%8, %15) : (i1, index) -> (i1, index)
+// CHECK-NEXT: %17:2 = "dataflow.demux"(%16#0, %16#1) : (i1, index) -> (index, index)
+// CHECK-NEXT: %18 = "dataflow.invariant"(%8, %1) : (i1, index) -> index
+// CHECK-NEXT: %19:2 = "dataflow.gate"(%8, %18) : (i1, index) -> (i1, index)
+// CHECK-NEXT: %20:2 = "dataflow.demux"(%19#0, %19#1) : (i1, index) -> (index, index)
+// CHECK-NEXT: %21 = "dataflow.invariant"(%8, %0) : (i1, index) -> index
+// CHECK-NEXT: %22:2 = "dataflow.gate"(%8, %21) : (i1, index) -> (i1, index)
+// CHECK-NEXT: %23:2 = "dataflow.demux"(%22#0, %22#1) : (i1, index) -> (index, index)
+// CHECK-NEXT: %24 = "dataflow.carry"(%28, %10#1, %30#1) : (i1, none, none) -> none
+// CHECK-NEXT: %25 = "dataflow.carry"(%28, %22#1, %45) : (i1, index, index) -> index
+// CHECK-NEXT: %26 = "dataflow.invariant"(%28, %16#1) : (i1, index) -> index
+// CHECK-NEXT: %27 = "dataflow.carry"(%28, %13#1, %44) : (i1, none, none) -> none
+// CHECK-NEXT: %28 = "arith.cmpi"(%25, %26) <{predicate = 6 : i64}> : (index, index) -> i1
+// CHECK-NEXT: %29:2 = "dataflow.demux"(%28, %24) : (i1, none) -> (none, none)
+// CHECK-NEXT: %30:2 = "dataflow.gate"(%28, %24) : (i1, none) -> (i1, none)
+// CHECK-NEXT: %31:2 = "dataflow.demux"(%30#0, %30#1) : (i1, none) -> (none, none)
+// CHECK-NEXT: %32:2 = "dataflow.demux"(%28, %27) : (i1, none) -> (none, none)
+// CHECK-NEXT: %33:2 = "dataflow.demux"(%28, %25) : (i1, index) -> (index, index)
+// CHECK-NEXT: %34 = "dataflow.invariant"(%28, %19#1) : (i1, index) -> index
+// CHECK-NEXT: %35:2 = "dataflow.gate"(%28, %34) : (i1, index) -> (i1, index)
+// CHECK-NEXT: %36:2 = "dataflow.demux"(%35#0, %35#1) : (i1, index) -> (index, index)
+// CHECK-NEXT: %37 = "dataflow.invariant"(%28, %22#1) : (i1, index) -> index
+// CHECK-NEXT: %38:2 = "dataflow.gate"(%28, %37) : (i1, index) -> (i1, index)
+// CHECK-NEXT: %39:2 = "dataflow.demux"(%38#0, %38#1) : (i1, index) -> (index, index)
+// CHECK-NEXT: %40:2 = "dataflow.sync"(%30#1, %32#1) : (none, none) -> (none, none)
+// CHECK-NEXT: %41 = "dataflow.store"(%arg2, %38#1, %35#1, %40#0) : (memref<?xindex>, index, index, none) -> none
+// CHECK-NEXT: %42:2 = "dataflow.load"(%arg2, %38#1, %41) : (memref<?xindex>, index, none) -> (index, none)
+// CHECK-NEXT: %43 = "arith.addi"(%42#0, %35#1) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %44 = "dataflow.store"(%arg2, %38#1, %43, %42#1) : (memref<?xindex>, index, index, none) -> none
+// CHECK-NEXT: %45 = "arith.addi"(%33#1, %35#1) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %46:2 = "dataflow.demux"(%30#0, %30#1) : (i1, none) -> (none, none)
+// CHECK-NEXT: %47:4 = "dataflow.sync"(%46#0, %31#0, %36#0, %39#0) : (none, none, index, index) -> (none, none, index, index)
+// CHECK-NEXT: %48 = "dataflow.mux"(%30#0, %47#0, %46#1) : (i1, none, none) -> none
+// CHECK-NEXT: %49 = "dataflow.carry"(%28, %10#1, %48) : (i1, none, none) -> none
+// CHECK-NEXT: %50:2 = "dataflow.demux"(%28, %49) : (i1, none) -> (none, none)
+// CHECK-NEXT: %51:2 = "dataflow.sync"(%29#0, %50#0) : (none, none) -> (none, none)
+// CHECK-NEXT: %52:2 = "dataflow.sync"(%51#0, %32#0) : (none, none) -> (none, none)
+// CHECK-NEXT: %53 = "dataflow.store"(%arg2, %22#1, %19#1, %52#0) : (memref<?xindex>, index, index, none) -> none
+// CHECK-NEXT: %54 = "arith.addi"(%14#1, %19#1) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %55:2 = "dataflow.demux"(%10#0, %51#0) : (i1, none) -> (none, none)
+// CHECK-NEXT: %56:2 = "dataflow.sync"(%55#0, %11#0) : (none, none) -> (none, none)
+// CHECK-NEXT: %57:2 = "dataflow.sync"(%56#0, %17#0) : (none, index) -> (none, index)
+// CHECK-NEXT: %58:2 = "dataflow.sync"(%20#0, %23#0) : (index, index) -> (index, index)
+// CHECK-NEXT: %59:2 = "dataflow.sync"(%57#0, %58#0) : (none, index) -> (none, index)
+// CHECK-NEXT: %60 = "dataflow.mux"(%10#0, %59#0, %55#1) : (i1, none, none) -> none
+// CHECK-NEXT: %61 = "dataflow.carry"(%8, %arg0, %60) : (i1, none, none) -> none
+// CHECK-NEXT: %62:2 = "dataflow.demux"(%8, %61) : (i1, none) -> (none, none)
+// CHECK-NEXT: %63:2 = "dataflow.sync"(%9#0, %62#0) : (none, none) -> (none, none)
+// CHECK-NEXT: %64 = "dataflow.carry"(%69, %63#0, %111#0) : (i1, none, none) -> none
+// CHECK-NEXT: %65 = "dataflow.carry"(%69, %0, %117) : (i1, index, index) -> index
+// CHECK-NEXT: %66 = "dataflow.invariant"(%69, %2) : (i1, index) -> index
+// CHECK-NEXT: %67 = "dataflow.carry"(%69, %12#0, %116) : (i1, none, none) -> none
+// CHECK-NEXT: %68 = "dataflow.carry"(%69, %13#0, %116) : (i1, none, none) -> none
+// CHECK-NEXT: %69 = "arith.cmpi"(%65, %66) <{predicate = 6 : i64}> : (index, index) -> i1
+// CHECK-NEXT: %70:2 = "dataflow.demux"(%69, %64) : (i1, none) -> (none, none)
+// CHECK-NEXT: %71:2 = "dataflow.gate"(%69, %64) : (i1, none) -> (i1, none)
+// CHECK-NEXT: %72:2 = "dataflow.demux"(%71#0, %71#1) : (i1, none) -> (none, none)
+// CHECK-NEXT: %73:2 = "dataflow.demux"(%69, %67) : (i1, none) -> (none, none)
+// CHECK-NEXT: %74:2 = "dataflow.demux"(%69, %68) : (i1, none) -> (none, none)
+// CHECK-NEXT: %75:2 = "dataflow.demux"(%69, %65) : (i1, index) -> (index, index)
+// CHECK-NEXT: %76 = "dataflow.invariant"(%69, %2) : (i1, index) -> index
+// CHECK-NEXT: %77:2 = "dataflow.gate"(%69, %76) : (i1, index) -> (i1, index)
+// CHECK-NEXT: %78:2 = "dataflow.demux"(%77#0, %77#1) : (i1, index) -> (index, index)
+// CHECK-NEXT: %79 = "dataflow.invariant"(%69, %1) : (i1, index) -> index
+// CHECK-NEXT: %80:2 = "dataflow.gate"(%69, %79) : (i1, index) -> (i1, index)
+// CHECK-NEXT: %81:2 = "dataflow.demux"(%80#0, %80#1) : (i1, index) -> (index, index)
+// CHECK-NEXT: %82 = "dataflow.invariant"(%69, %0) : (i1, index) -> index
+// CHECK-NEXT: %83:2 = "dataflow.gate"(%69, %82) : (i1, index) -> (i1, index)
+// CHECK-NEXT: %84:2 = "dataflow.demux"(%83#0, %83#1) : (i1, index) -> (index, index)
+// CHECK-NEXT: %85 = "dataflow.carry"(%90, %71#1, %92#1) : (i1, none, none) -> none
+// CHECK-NEXT: %86 = "dataflow.carry"(%90, %83#1, %105) : (i1, index, index) -> index
+// CHECK-NEXT: %87 = "dataflow.invariant"(%90, %77#1) : (i1, index) -> index
+// CHECK-NEXT: %88 = "dataflow.carry"(%90, %73#1, %104) : (i1, none, none) -> none
+// CHECK-NEXT: %89 = "dataflow.carry"(%90, %74#1, %104) : (i1, none, none) -> none
+// CHECK-NEXT: %90 = "arith.cmpi"(%86, %87) <{predicate = 6 : i64}> : (index, index) -> i1
+// CHECK-NEXT: %91:2 = "dataflow.demux"(%90, %85) : (i1, none) -> (none, none)
+// CHECK-NEXT: %92:2 = "dataflow.gate"(%90, %85) : (i1, none) -> (i1, none)
+// CHECK-NEXT: %93:2 = "dataflow.demux"(%92#0, %92#1) : (i1, none) -> (none, none)
+// CHECK-NEXT: %94:2 = "dataflow.demux"(%90, %88) : (i1, none) -> (none, none)
+// CHECK-NEXT: %95:2 = "dataflow.demux"(%90, %89) : (i1, none) -> (none, none)
+// CHECK-NEXT: %96:2 = "dataflow.demux"(%90, %86) : (i1, index) -> (index, index)
+// CHECK-NEXT: %97 = "dataflow.invariant"(%90, %80#1) : (i1, index) -> index
+// CHECK-NEXT: %98:2 = "dataflow.gate"(%90, %97) : (i1, index) -> (i1, index)
+// CHECK-NEXT: %99:2 = "dataflow.demux"(%98#0, %98#1) : (i1, index) -> (index, index)
+// CHECK-NEXT: %100 = "dataflow.invariant"(%90, %83#1) : (i1, index) -> index
+// CHECK-NEXT: %101:2 = "dataflow.gate"(%90, %100) : (i1, index) -> (i1, index)
+// CHECK-NEXT: %102:2 = "dataflow.demux"(%101#0, %101#1) : (i1, index) -> (index, index)
+// CHECK-NEXT: %103:2 = "dataflow.sync"(%92#1, %95#1) : (none, none) -> (none, none)
+// CHECK-NEXT: %104 = "dataflow.store"(%arg2, %101#1, %98#1, %103#0) : (memref<?xindex>, index, index, none) -> none
+// CHECK-NEXT: %105 = "arith.addi"(%96#1, %98#1) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %106:2 = "dataflow.demux"(%92#0, %92#1) : (i1, none) -> (none, none)
+// CHECK-NEXT: %107:4 = "dataflow.sync"(%106#0, %93#0, %99#0, %102#0) : (none, none, index, index) -> (none, none, index, index)
+// CHECK-NEXT: %108 = "dataflow.mux"(%92#0, %107#0, %106#1) : (i1, none, none) -> none
+// CHECK-NEXT: %109 = "dataflow.carry"(%90, %71#1, %108) : (i1, none, none) -> none
+// CHECK-NEXT: %110:2 = "dataflow.demux"(%90, %109) : (i1, none) -> (none, none)
+// CHECK-NEXT: %111:2 = "dataflow.sync"(%91#0, %110#0) : (none, none) -> (none, none)
+// CHECK-NEXT: %112:2 = "dataflow.sync"(%111#0, %94#0) : (none, none) -> (none, none)
+// CHECK-NEXT: %113:2 = "dataflow.load"(%arg2, %83#1, %112#0) : (memref<?xindex>, index, none) -> (index, none)
+// CHECK-NEXT: %114:2 = "dataflow.sync"(%95#0, %113#1) : (none, none) -> (none, none)
+// CHECK-NEXT: %115 = "arith.addi"(%113#0, %80#1) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %116 = "dataflow.store"(%arg2, %83#1, %115, %114#0) : (memref<?xindex>, index, index, none) -> none
+// CHECK-NEXT: %117 = "arith.addi"(%75#1, %80#1) <{overflowFlags = #arith.overflow<none>}> : (index, index) -> index
+// CHECK-NEXT: %118:2 = "dataflow.demux"(%71#0, %111#0) : (i1, none) -> (none, none)
+// CHECK-NEXT: %119:2 = "dataflow.sync"(%118#0, %72#0) : (none, none) -> (none, none)
+// CHECK-NEXT: %120:2 = "dataflow.sync"(%119#0, %78#0) : (none, index) -> (none, index)
+// CHECK-NEXT: %121:2 = "dataflow.sync"(%81#0, %84#0) : (index, index) -> (index, index)
+// CHECK-NEXT: %122:2 = "dataflow.sync"(%120#0, %121#0) : (none, index) -> (none, index)
+// CHECK-NEXT: %123 = "dataflow.mux"(%71#0, %122#0, %118#1) : (i1, none, none) -> none
+// CHECK-NEXT: %124 = "dataflow.carry"(%69, %63#0, %123) : (i1, none, none) -> none
+// CHECK-NEXT: %125:2 = "dataflow.demux"(%69, %124) : (i1, none) -> (none, none)
+// CHECK-NEXT: %126:2 = "dataflow.sync"(%70#0, %125#0) : (none, none) -> (none, none)
+// CHECK-NEXT: "dataflow.graph.return"(%126#0, %74#0) <{operandSegmentSizes = array<i32: 0, 0, 0, 2>}> : (none, none) -> ()
+// CHECK-NEXT: }) : () -> ()
+// CHECK-NEXT: }) : () -> ()
+// CHECK-EMPTY: