Checking ranges

Printer.cpp

int i = 0;
  for (auto it : asmState.getOpDefs()) {
    const char *startOp = it.scopeLoc.Start.getPointer();
    const char *endOp = it.scopeLoc.End.getPointer();

    for (auto res : it.resultGroups) {
      SMRange range = res.definition.loc;
      startOp = std::min(startOp, range.Start.getPointer());
    }

    rewriteBuffer.InsertText(startOp - start, ("🌜<" + Twine(i) + ">").str());
    rewriteBuffer.InsertText(endOp - start, ("<" + Twine(i) + ">🌛").str());
    ++i;
  }

test_0before.mlir

// Two dynamic shapes: one of direct shape.shape_of(arg) and the other.
🌜<15>func.func @two_dynamic_one_direct_shape(%arg0: tensor<?x4x?xf32>, %arg1: tensor<2x4x?xf32>) -> tensor<?x4x?xf32> {
  // SHAPE-DAG: Shape for {{.*}} = "test.abs"({{.*}}> :: @shape_cal_0(<block argument> of type 'tensor<?x4x?xf32>' at index: 0)
  // SHAPE-DAG: Shape for {{.*}} = "test.concat"({{.*}}> :: @shape_cal_1(<block argument> of type 'tensor<?x4x?xf32>' at index: 0)
  🌜<0>%c2 = arith.constant 2 : index
  <0>🌛🌜<1>%c0 = arith.constant 0 : index
// Bar
  <1>🌛🌜<2>%c4 = arith.constant 4 : index

  // Foo

  <2>🌛🌜<3>%0 = shape.shape_of %arg0 : tensor<?x4x?xf32> -> tensor<3xindex>
  <3>🌛🌜<4>%1 = shape.get_extent %0, %c2 : tensor<3xindex>, index -> index
  <4>🌛🌜<5>%2 = "test.abs"(%arg0) : (tensor<?x4x?xf32>) -> tensor<?x4x?xf32>
  <5>🌛🌜<6>%3 = shape.with_shape %2, %0 : tensor<?x4x?xf32>, tensor<3xindex>
  <6>🌛🌜<7>%4 = shape.value_of %3 : tensor<?x4x?xf32>
  <7>🌛🌜<8>%5 = "test.concat"(%4, %arg1) {axis = 0 : i64} : (tensor<?x4x?xf32>, tensor<2x4x?xf32>) -> tensor<?x4x?xf32>
  <8>🌛🌜<9>%6 = shape.get_extent %0, %c0 : tensor<3xindex>, index -> index
  <9>🌛🌜<10>%7 = arith.addi %6, %c2 : index
  <10>🌛🌜<11>%8 = shape.from_extents %7, %c4, %1 : index, index, index
  <11>🌛🌜<12>%9 = shape.with_shape %5, %8 : tensor<?x4x?xf32>, !shape.shape
  <12>🌛🌜<13>%10 = shape.value_of %9 : tensor<?x4x?xf32>
  <13>🌛🌜<14>return %10 : tensor<?x4x?xf32>
<14>🌛}

<15>🌛

test_1after.mlir

// Two dynamic shapes: one of direct shape.shape_of(arg) and the other.
🌜<15>func.func @two_dynamic_one_direct_shape(%arg0: tensor<?x4x?xf32>, %arg1: tensor<2x4x?xf32>) -> tensor<?x4x?xf32> {
  // SHAPE-DAG: Shape for {{.*}} = "test.abs"({{.*}}> :: @shape_cal_0(<block argument> of type 'tensor<?x4x?xf32>' at index: 0)
  // SHAPE-DAG: Shape for {{.*}} = "test.concat"({{.*}}> :: @shape_cal_1(<block argument> of type 'tensor<?x4x?xf32>' at index: 0)
  🌜<0>%c2 = arith.constant 2 : index<0>🌛
  🌜<1>%c0 = arith.constant 0 : index<1>🌛
// Bar
  🌜<2>%c4 = arith.constant 4 : index<2>🌛

  // Foo

  🌜<3>%0 = shape.shape_of %arg0 : tensor<?x4x?xf32> -> tensor<3xindex><3>🌛
  🌜<4>%1 = shape.get_extent %0, %c2 : tensor<3xindex>, index -> index<4>🌛
  🌜<5>%2 = "test.abs"(%arg0) : (tensor<?x4x?xf32>) -> tensor<?x4x?xf32><5>🌛
  🌜<6>%3 = shape.with_shape %2, %0 : tensor<?x4x?xf32>, tensor<3xindex><6>🌛
  🌜<7>%4 = shape.value_of %3 : tensor<?x4x?xf32><7>🌛
  🌜<8>%5 = "test.concat"(%4, %arg1) {axis = 0 : i64} : (tensor<?x4x?xf32>, tensor<2x4x?xf32>) -> tensor<?x4x?xf32><8>🌛
  🌜<9>%6 = shape.get_extent %0, %c0 : tensor<3xindex>, index -> index<9>🌛
  🌜<10>%7 = arith.addi %6, %c2 : index<10>🌛
  🌜<11>%8 = shape.from_extents %7, %c4, %1 : index, index, index<11>🌛
  🌜<12>%9 = shape.with_shape %5, %8 : tensor<?x4x?xf32>, !shape.shape
  🌜<13>%10<12>🌛 = shape.value_of %9 : tensor<?x4x?xf32><13>🌛
  🌜<14>return %10 : tensor<?x4x?xf32><14>🌛
}<15>🌛