Back to PBT run

input.mlir

module {
  dataflow.graph private @graph_0(
      %start: none, %i: index, %c: i1, %m: vector<4xi1>,
      %av: vector<4xindex>, %val: i32,
      %a: memref<16xi32>, %b: memref<16xi32>) -> ()
      attributes {input_segments = array<i32: 5, 0, 2>,
                  result_segments = array<i32: 0, 0, 0>} {
    %pad = arith.constant 0 : i32
    %g0, %gd0 = dataflow.load %a[%av] %start mask %m : memref<16xi32>, vector<4xindex>, vector<4xi32>
    %sc0 = dataflow.store %b[%av] %g0 %start mask %m : memref<16xi32>, vector<4xindex>, vector<4xi32>
    memref.store %val, %b[%i] : memref<16xi32>
    %s1 = memref.load %b[%i] : memref<16xi32>
    dataflow.graph.return %start : none
  }
}