ASSERT_THAT(remat.GetMemProfile(),
              ElementsAreArray({1, 3, 7, 15, 23, 19, 9, 4}));

  EXPECT_CALL(remat, ApplyRemat(FieldsAre(/*begin=*/2,
                                          /*end=*/3,
                                          /*insert=*/5)));
  EXPECT_CALL(remat, ApplyRemat(FieldsAre(/*begin=*/0,
                                          /*end=*/1,
                                          /*insert=*/8)));

  // operations will be re-inserted. For each rematerialization found,
  // ApplyRemat is invoked (which can be used to apply the rematerialization to
  // the higher- level representation, e.g., MLIR, flatbuffer, ...)
  void RunGreedyAlgorithm(int max_cost, int max_block_length,
                          SizeT min_savings);

  virtual void ApplyRemat(const RematSpec& remat) {}

 protected:

                            : std::min(max_block_length, max_cost - cost)),
                   2 * min_block_length));
      if (remat.begin == remat.end) break;
      Remat(remat);
      ApplyRemat(remat);
      cost += (remat.end - remat.begin);
    }
  }
}

}  // namespace TFL