Back to PBT run

input.generic.mlir

"builtin.module"() ({
  "dataflow.thread"() <{domain = #dataflow.thread_domain<dense>, function_type = (memref<8xi32>, index, i32) -> (), sym_name = "thread_0", sym_visibility = "private"}> ({
  ^bb0(%arg11: memref<8xi32>, %arg12: index, %arg13: i32, %arg14: none, %arg15: index):
    %16 = "arith.constant"() <{value = 7 : i32}> : () -> i32
    "scf.forall"() <{operandSegmentSizes = array<i32: 0, 0, 0, 0>, staticLowerBound = array<i64: 0>, staticStep = array<i64: 1>, staticUpperBound = array<i64: 4>}> ({
    ^bb0(%arg16: index):
      "memref.store"(%16, %arg11, %arg16) : (i32, memref<8xi32>, index) -> ()
      "scf.forall.in_parallel"() ({
      ^bb0:
      }) : () -> ()
    }) : () -> ()
    "dataflow.thread.yield"(%arg14) : (none) -> ()
  }) : () -> ()
  "func.func"() <{function_type = () -> (), sym_name = "host_0"}> ({
    %11 = "memref.alloc"() <{operandSegmentSizes = array<i32: 0, 0>}> : () -> memref<8xi32>
    %12 = "arith.constant"() <{value = 2 : index}> : () -> index
    %13 = "arith.constant"() <{value = 3 : i32}> : () -> i32
    %14 = "arith.constant"() <{value = 4 : index}> : () -> index
    %15 = "dataflow.thread.launch"(%11, %12, %13, %14) <{callee = @thread_0, operandSegmentSizes = array<i32: 3, 1, 0>}> : (memref<8xi32>, index, i32, index) -> !dataflow.thread_token
    "func.return"() : () -> ()
  }) : () -> ()
  "dataflow.thread"() <{domain = #dataflow.thread_domain<dense>, function_type = (memref<8xi32>, index) -> (), sym_name = "thread_1", sym_visibility = "private"}> ({
  ^bb0(%arg6: memref<8xi32>, %arg7: index, %arg8: none, %arg9: index):
    %10 = "arith.constant"() <{value = 7 : i32}> : () -> i32
    "scf.forall"() <{operandSegmentSizes = array<i32: 0, 0, 0, 0>, staticLowerBound = array<i64: 0>, staticStep = array<i64: 1>, staticUpperBound = array<i64: 4>}> ({
    ^bb0(%arg10: index):
      "memref.store"(%10, %arg6, %arg10) : (i32, memref<8xi32>, index) -> ()
      "scf.forall.in_parallel"() ({
      ^bb0:
      }) : () -> ()
    }) : () -> ()
    "dataflow.thread.yield"(%arg8) : (none) -> ()
  }) : () -> ()
  "func.func"() <{function_type = () -> (), sym_name = "host_1"}> ({
    %6 = "memref.alloc"() <{operandSegmentSizes = array<i32: 0, 0>}> : () -> memref<8xi32>
    %7 = "arith.constant"() <{value = 2 : index}> : () -> index
    %8 = "arith.constant"() <{value = 4 : index}> : () -> index
    %9 = "dataflow.thread.launch"(%6, %7, %8) <{callee = @thread_1, operandSegmentSizes = array<i32: 2, 1, 0>}> : (memref<8xi32>, index, index) -> !dataflow.thread_token
    "func.return"() : () -> ()
  }) : () -> ()
  "dataflow.thread"() <{domain = #dataflow.thread_domain<dense>, function_type = (memref<8xi32>, index) -> (), sym_name = "thread_2", sym_visibility = "private"}> ({
  ^bb0(%arg0: memref<8xi32>, %arg1: index, %arg2: none, %arg3: index, %arg4: index):
    %5 = "arith.constant"() <{value = 7 : i32}> : () -> i32
    "scf.forall"() <{operandSegmentSizes = array<i32: 0, 0, 0, 0>, staticLowerBound = array<i64: 0>, staticStep = array<i64: 1>, staticUpperBound = array<i64: 2>}> ({
    ^bb0(%arg5: index):
      "memref.store"(%5, %arg0, %arg5) : (i32, memref<8xi32>, index) -> ()
      "scf.forall.in_parallel"() ({
      ^bb0:
      }) : () -> ()
    }) : () -> ()
    "dataflow.thread.yield"() : () -> ()
  }) : () -> ()
  "func.func"() <{function_type = () -> (), sym_name = "host_2"}> ({
    %0 = "memref.alloc"() <{operandSegmentSizes = array<i32: 0, 0>}> : () -> memref<8xi32>
    %1 = "arith.constant"() <{value = 2 : index}> : () -> index
    %2 = "arith.constant"() <{value = 4 : index}> : () -> index
    %3 = "arith.constant"() <{value = 4 : index}> : () -> index
    %4 = "dataflow.thread.launch"(%0, %1, %2, %3) <{callee = @thread_2, operandSegmentSizes = array<i32: 2, 2, 0>}> : (memref<8xi32>, index, index, index) -> !dataflow.thread_token
    "func.return"() : () -> ()
  }) : () -> ()
}) : () -> ()