| //===- llvm-reduce.cpp - The LLVM Delta Reduction utility -----------------===// |
| // |
| // 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 |
| // |
| //===----------------------------------------------------------------------===// |
| // |
| // This program tries to reduce an IR test case for a given interesting-ness |
| // test. It runs multiple delta debugging passes in order to minimize the input |
| // file. It's worth noting that this is a part of the bugpoint redesign |
| // proposal, and thus a *temporary* tool that will eventually be integrated |
| // into the bugpoint tool itself. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "DeltaManager.h" |
| #include "ReducerWorkItem.h" |
| #include "TestRunner.h" |
| #include "llvm/ADT/SmallString.h" |
| #include "llvm/CodeGen/CommandFlags.h" |
| #include "llvm/IR/LLVMContext.h" |
| #include "llvm/IR/Verifier.h" |
| #include "llvm/IRReader/IRReader.h" |
| #include "llvm/MC/TargetRegistry.h" |
| #include "llvm/Support/CommandLine.h" |
| #include "llvm/Support/Host.h" |
| #include "llvm/Support/InitLLVM.h" |
| #include "llvm/Support/SourceMgr.h" |
| #include "llvm/Support/TargetSelect.h" |
| #include "llvm/Support/raw_ostream.h" |
| #include "llvm/Target/TargetMachine.h" |
| #include <system_error> |
| #include <vector> |
| |
| using namespace llvm; |
| |
| static cl::OptionCategory Options("llvm-reduce options"); |
| |
| static cl::opt<bool> Help("h", cl::desc("Alias for -help"), cl::Hidden, |
| cl::cat(Options)); |
| static cl::opt<bool> Version("v", cl::desc("Alias for -version"), cl::Hidden, |
| cl::cat(Options)); |
| |
| static cl::opt<bool> |
| PrintDeltaPasses("print-delta-passes", |
| cl::desc("Print list of delta passes, passable to " |
| "--delta-passes as a comma separated list"), |
| cl::cat(Options)); |
| |
| static cl::opt<std::string> InputFilename(cl::Positional, cl::Required, |
| cl::desc("<input llvm ll/bc file>"), |
| cl::cat(Options)); |
| |
| static cl::opt<std::string> |
| TestFilename("test", cl::Required, |
| cl::desc("Name of the interesting-ness test to be run"), |
| cl::cat(Options)); |
| |
| static cl::list<std::string> |
| TestArguments("test-arg", cl::ZeroOrMore, |
| cl::desc("Arguments passed onto the interesting-ness test"), |
| cl::cat(Options)); |
| |
| static cl::opt<std::string> OutputFilename( |
| "output", cl::desc("Specify the output file. default: reduced.ll|mir")); |
| static cl::alias OutputFileAlias("o", cl::desc("Alias for -output"), |
| cl::aliasopt(OutputFilename), |
| cl::cat(Options)); |
| |
| static cl::opt<bool> |
| ReplaceInput("in-place", |
| cl::desc("WARNING: This option will replace your input file " |
| "with the reduced version!"), |
| cl::cat(Options)); |
| |
| enum class InputLanguages { None, IR, MIR }; |
| |
| static cl::opt<InputLanguages> |
| InputLanguage("x", cl::ValueOptional, |
| cl::desc("Input language ('ir' or 'mir')"), |
| cl::init(InputLanguages::None), |
| cl::values(clEnumValN(InputLanguages::IR, "ir", ""), |
| clEnumValN(InputLanguages::MIR, "mir", "")), |
| cl::cat(Options)); |
| |
| static cl::opt<std::string> TargetTriple("mtriple", |
| cl::desc("Set the target triple"), |
| cl::cat(Options)); |
| |
| static codegen::RegisterCodeGenFlags CGF; |
| |
| void writeOutput(ReducerWorkItem &M, StringRef Message) { |
| if (ReplaceInput) // In-place |
| OutputFilename = InputFilename.c_str(); |
| else if (OutputFilename.empty() || OutputFilename == "-") |
| OutputFilename = M.isMIR() ? "reduced.mir" : "reduced.ll"; |
| std::error_code EC; |
| raw_fd_ostream Out(OutputFilename, EC); |
| if (EC) { |
| errs() << "Error opening output file: " << EC.message() << "!\n"; |
| exit(1); |
| } |
| M.print(Out, /*AnnotationWriter=*/nullptr); |
| errs() << Message << OutputFilename << "\n"; |
| } |
| |
| static std::unique_ptr<LLVMTargetMachine> createTargetMachine() { |
| InitializeAllTargets(); |
| InitializeAllTargetMCs(); |
| InitializeAllAsmPrinters(); |
| InitializeAllAsmParsers(); |
| |
| if (TargetTriple == "") |
| TargetTriple = sys::getDefaultTargetTriple(); |
| auto TT(Triple::normalize(TargetTriple)); |
| std::string CPU(codegen::getCPUStr()); |
| std::string FS(codegen::getFeaturesStr()); |
| |
| std::string Error; |
| const Target *TheTarget = TargetRegistry::lookupTarget(TT, Error); |
| |
| return std::unique_ptr<LLVMTargetMachine>( |
| static_cast<LLVMTargetMachine *>(TheTarget->createTargetMachine( |
| TT, CPU, FS, TargetOptions(), None, None, CodeGenOpt::Default))); |
| } |
| |
| int main(int Argc, char **Argv) { |
| InitLLVM X(Argc, Argv); |
| |
| cl::HideUnrelatedOptions({&Options, &getColorCategory()}); |
| cl::ParseCommandLineOptions(Argc, Argv, "LLVM automatic testcase reducer.\n"); |
| |
| bool ReduceModeMIR = false; |
| if (InputLanguage != InputLanguages::None) { |
| if (InputLanguage == InputLanguages::MIR) |
| ReduceModeMIR = true; |
| } else if (StringRef(InputFilename).endswith(".mir")) { |
| ReduceModeMIR = true; |
| } |
| |
| if (PrintDeltaPasses) { |
| printDeltaPasses(errs()); |
| return 0; |
| } |
| |
| LLVMContext Context; |
| std::unique_ptr<LLVMTargetMachine> TM; |
| std::unique_ptr<MachineModuleInfo> MMI; |
| std::unique_ptr<ReducerWorkItem> OriginalProgram; |
| if (ReduceModeMIR) { |
| TM = createTargetMachine(); |
| MMI = std::make_unique<MachineModuleInfo>(TM.get()); |
| } |
| OriginalProgram = parseReducerWorkItem(InputFilename, Context, MMI.get()); |
| if (!OriginalProgram) { |
| return 1; |
| } |
| |
| // Initialize test environment |
| TestRunner Tester(TestFilename, TestArguments, std::move(OriginalProgram)); |
| |
| // Try to reduce code |
| runDeltaPasses(Tester); |
| |
| // Print reduced file to STDOUT |
| if (OutputFilename == "-") |
| Tester.getProgram().print(outs(), nullptr); |
| else |
| writeOutput(Tester.getProgram(), "\nDone reducing! Reduced testcase: "); |
| |
| return 0; |
| } |