Back to PBT run

input.mlir

dataflow.graph private @pair_0(%start: none, %lb: i64, %ub: i64, %step: i64,
    %a: memref<?xi32>, %b: memref<?xi32>) -> ()
    attributes {input_segments = array<i32: 3, 0, 2>,
                result_segments = array<i32: 0, 0, 0>} {
  scf.for %i = %lb to %ub step %step : i64 {
    %idx = arith.index_cast %i : i64 to index
    %ploaded = memref.load %b[%idx] : memref<?xi32>
    memref.store %ploaded, %a[%idx] : memref<?xi32>
  }
  dataflow.graph.return %start : none
}