TheRock/compiler/amd-llvm/llvm/tools/reduce-chunk-list/reduce-chunk-list.cpp

//===-- reduce-chunk-list.cpp - Reduce a chunks list to its minimal size --===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// See the llvm-project/llvm/docs/ProgrammersManual.rst to see how to use this
// tool
//
//===----------------------------------------------------------------------===//

#include "llvm/Support/CommandLine.h"
#include "llvm/Support/IntegerInclusiveInterval.h"
#include "llvm/Support/Program.h"

using namespace llvm;

static cl::opt<std::string> ReproductionCmd(cl::Positional, cl::Required);

static cl::opt<std::string> StartChunks(cl::Positional, cl::Required);

static cl::opt<bool> Pessimist("pessimist", cl::init(false));

namespace {

bool isStillInteresting(ArrayRef<IntegerInclusiveInterval> Chunks) {
  IntegerInclusiveIntervalUtils::IntervalList SimpleChunks =
      IntegerInclusiveIntervalUtils::mergeAdjacentIntervals(Chunks);

  std::string ChunkStr;
  {
    raw_string_ostream OS(ChunkStr);
    IntegerInclusiveIntervalUtils::printIntervals(OS, SimpleChunks);
  }

  errs() << "Checking with: " << ChunkStr << "\n";

  std::vector<StringRef> Argv;
  Argv.push_back(ReproductionCmd);
  Argv.push_back(ChunkStr);

  std::string ErrMsg;
  bool ExecutionFailed;
  int Result = sys::ExecuteAndWait(Argv[0], Argv, std::nullopt, {}, 0, 0,
                                   &ErrMsg, &ExecutionFailed);
  if (ExecutionFailed) {
    errs() << "failed to execute : " << Argv[0] << " : " << ErrMsg << "\n";
    exit(1);
  }

  bool Res = Result != 0;
  if (Res) {
    errs() << "SUCCESS : Still Interesting\n";
  } else {
    errs() << "FAILURE : Not Interesting\n";
  }
  return Res;
}

bool increaseGranularity(IntegerInclusiveIntervalUtils::IntervalList &Chunks) {
  errs() << "Increasing granularity\n";
  IntegerInclusiveIntervalUtils::IntervalList NewChunks;
  bool SplitOne = false;

  for (auto &C : Chunks) {
    if (C.getBegin() == C.getEnd()) {
      NewChunks.push_back(C);
    } else {
      int64_t Half = (C.getBegin() + C.getEnd()) / 2;
      NewChunks.push_back(IntegerInclusiveInterval(C.getBegin(), Half));
      NewChunks.push_back(IntegerInclusiveInterval(Half + 1, C.getEnd()));
      SplitOne = true;
    }
  }
  if (SplitOne) {
    Chunks = std::move(NewChunks);
  }
  return SplitOne;
}

} // namespace

int main(int argc, char **argv) {
  cl::ParseCommandLineOptions(argc, argv);

  auto ExpectedChunks =
      IntegerInclusiveIntervalUtils::parseIntervals(StartChunks, ',');
  if (!ExpectedChunks) {
    handleAllErrors(ExpectedChunks.takeError(), [](const StringError &E) {
      errs() << "Error parsing chunks: " << E.getMessage() << "\n";
    });
    return 1;
  }
  IntegerInclusiveIntervalUtils::IntervalList CurrChunks =
      std::move(*ExpectedChunks);

  auto Program = sys::findProgramByName(ReproductionCmd);
  if (!Program) {
    errs() << "failed to find command : " << ReproductionCmd << "\n";
    return 1;
  }
  ReproductionCmd.setValue(Program.get());

  errs() << "Input Checking:\n";
  if (!isStillInteresting(CurrChunks)) {
    errs() << "starting chunks are not interesting\n";
    return 1;
  }
  if (CurrChunks.size() == 1)
    increaseGranularity(CurrChunks);
  if (Pessimist)
    while (increaseGranularity(CurrChunks))
      /* empty body */;
  while (1) {
    for (int Idx = (CurrChunks.size() - 1); Idx >= 0; Idx--) {
      if (CurrChunks.size() == 1)
        break;

      IntegerInclusiveInterval Testing = CurrChunks[Idx];
      errs() << "Trying to remove : ";
      Testing.print(errs());
      errs() << "\n";

      CurrChunks.erase(CurrChunks.begin() + Idx);

      if (!isStillInteresting(CurrChunks))
        CurrChunks.insert(CurrChunks.begin() + Idx, Testing);
    }
    bool HasSplit = increaseGranularity(CurrChunks);
    if (!HasSplit)
      break;
  }

  errs() << "Minimal Chunks = ";
  IntegerInclusiveIntervalUtils::printIntervals(
      llvm::errs(),
      IntegerInclusiveIntervalUtils::mergeAdjacentIntervals(CurrChunks));
  errs() << "\n";
}