Back to PBT run

input.generic.mlir

"builtin.module"() ({
  "dataflow.thread"() <{domain = #dataflow.thread_domain<dense>, function_type = () -> (), sym_name = "t_0", sym_visibility = "private"}> ({
  ^bb0(%arg12: none, %arg13: index, %arg14: index):
    "dataflow.thread.yield"(%arg12) : (none) -> ()
  }) : () -> ()
  "dataflow.thread"() <{domain = #dataflow.thread_domain<dense>, function_type = (index, memref<?xf32>, i32) -> (), sym_name = "t_1", sym_visibility = "private"}> ({
  ^bb0(%arg6: index, %arg7: memref<?xf32>, %arg8: i32, %arg9: none, %arg10: index, %arg11: index):
    "dataflow.thread.yield"() : () -> ()
  }) : () -> ()
  "dataflow.thread"() <{domain = #dataflow.thread_domain<dense>, function_type = (index, memref<?xf32>, i32) -> (), sym_name = "t_2", sym_visibility = "private"}> ({
  ^bb0(%arg1: index, %arg2: memref<?xf32>, %arg3: i32, %arg4: none, %arg5: index):
    "dataflow.thread.yield"() : () -> ()
  }) : () -> ()
  "func.func"() <{function_type = (memref<?xf32>) -> (), sym_name = "host"}> ({
  ^bb0(%arg0: memref<?xf32>):
    %0 = "arith.constant"() <{value = 1 : i32}> : () -> i32
    %1 = "arith.constant"() <{value = 1.000000e+00 : f32}> : () -> f32
    %2 = "arith.constant"() <{value = 1 : i64}> : () -> i64
    %3 = "arith.constant"() <{value = 4 : index}> : () -> index
    %4 = "dataflow.thread.launch"(%3, %3) <{callee = @t_0, operandSegmentSizes = array<i32: 0, 2, 0>}> : (index, index) -> !dataflow.thread_token
    "dataflow.thread.wait"(%4) : (!dataflow.thread_token) -> ()
    %5 = "dataflow.thread.launch"(%3, %arg0, %0, %3, %3) <{callee = @t_1, operandSegmentSizes = array<i32: 3, 2, 0>}> : (index, memref<?xf32>, i32, index, index) -> !dataflow.thread_token
    "dataflow.thread.wait"(%5) : (!dataflow.thread_token) -> ()
    %6 = "dataflow.thread.launch"(%3, %arg0, %0, %3) <{callee = @t_2, operandSegmentSizes = array<i32: 3, 1, 0>}> : (index, memref<?xf32>, i32, index) -> !dataflow.thread_token
    "func.return"() : () -> ()
  }) : () -> ()
}) : () -> ()