1 files changed, 8 insertions, 13 deletions
diff --git a/llvm/tools/llvm-reduce/llvm-reduce.cpp b/llvm/tools/llvm-reduce/llvm-reduce.cpp
index 43dd15f..90b7ed5 100644
--- a/llvm/tools/llvm-reduce/llvm-reduce.cpp
+++ b/llvm/tools/llvm-reduce/llvm-reduce.cpp
@@ -86,7 +86,7 @@ static std::unique_ptr<Module> parseInputFile(StringRef Filename,
   return Result;
 }
 
-void writeOutput(Module *M, StringRef Message) {
+void writeOutput(Module &M, StringRef Message) {
   if (ReplaceInput) // In-place
     OutputFilename = InputFilename.c_str();
   else if (OutputFilename.empty() || OutputFilename == "-")
@@ -98,7 +98,7 @@ void writeOutput(Module *M, StringRef Message) {
     errs() << "Error opening output file: " << EC.message() << "!\n";
     exit(1);
   }
-  M->print(Out, /*AnnotationWriter=*/nullptr);
+  M.print(Out, /*AnnotationWriter=*/nullptr);
   errs() << Message << OutputFilename << "\n";
 }
 
@@ -122,21 +122,16 @@ int main(int Argc, char **Argv) {
   }
 
   // Initialize test environment
-  TestRunner Tester(TestFilename, TestArguments);
-  Tester.setProgram(std::move(OriginalProgram));
+  TestRunner Tester(TestFilename, TestArguments, std::move(OriginalProgram));
 
   // Try to reduce code
   runDeltaPasses(Tester);
 
-  if (!Tester.getProgram()) {
-    errs() << "\nCouldnt reduce input :/\n";
-  } else {
-    // Print reduced file to STDOUT
-    if (OutputFilename == "-")
-      Tester.getProgram()->print(outs(), nullptr);
-    else
-      writeOutput(Tester.getProgram(), "\nDone reducing! Reduced testcase: ");
-  }
+  // Print reduced file to STDOUT
+  if (OutputFilename == "-")
+    Tester.getProgram().print(outs(), nullptr);
+  else
+    writeOutput(Tester.getProgram(), "\nDone reducing! Reduced testcase: ");
 
   return 0;
 }