xref: /aosp_15_r20/external/llvm/tools/llvm-split/llvm-split.cpp (revision 9880d6810fe72a1726cb53787c6711e909410d58)
1*9880d681SAndroid Build Coastguard Worker //===-- llvm-split: command line tool for testing module splitter ---------===//
2*9880d681SAndroid Build Coastguard Worker //
3*9880d681SAndroid Build Coastguard Worker //                     The LLVM Compiler Infrastructure
4*9880d681SAndroid Build Coastguard Worker //
5*9880d681SAndroid Build Coastguard Worker // This file is distributed under the University of Illinois Open Source
6*9880d681SAndroid Build Coastguard Worker // License. See LICENSE.TXT for details.
7*9880d681SAndroid Build Coastguard Worker //
8*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
9*9880d681SAndroid Build Coastguard Worker //
10*9880d681SAndroid Build Coastguard Worker // This program can be used to test the llvm::SplitModule function.
11*9880d681SAndroid Build Coastguard Worker //
12*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
13*9880d681SAndroid Build Coastguard Worker 
14*9880d681SAndroid Build Coastguard Worker #include "llvm/ADT/StringExtras.h"
15*9880d681SAndroid Build Coastguard Worker #include "llvm/Bitcode/ReaderWriter.h"
16*9880d681SAndroid Build Coastguard Worker #include "llvm/IR/LLVMContext.h"
17*9880d681SAndroid Build Coastguard Worker #include "llvm/IR/Verifier.h"
18*9880d681SAndroid Build Coastguard Worker #include "llvm/IRReader/IRReader.h"
19*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/CommandLine.h"
20*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/FileSystem.h"
21*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/SourceMgr.h"
22*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/ToolOutputFile.h"
23*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/raw_ostream.h"
24*9880d681SAndroid Build Coastguard Worker #include "llvm/Transforms/Utils/SplitModule.h"
25*9880d681SAndroid Build Coastguard Worker 
26*9880d681SAndroid Build Coastguard Worker using namespace llvm;
27*9880d681SAndroid Build Coastguard Worker 
28*9880d681SAndroid Build Coastguard Worker static cl::opt<std::string>
29*9880d681SAndroid Build Coastguard Worker InputFilename(cl::Positional, cl::desc("<input bitcode file>"),
30*9880d681SAndroid Build Coastguard Worker     cl::init("-"), cl::value_desc("filename"));
31*9880d681SAndroid Build Coastguard Worker 
32*9880d681SAndroid Build Coastguard Worker static cl::opt<std::string>
33*9880d681SAndroid Build Coastguard Worker OutputFilename("o", cl::desc("Override output filename"),
34*9880d681SAndroid Build Coastguard Worker                cl::value_desc("filename"));
35*9880d681SAndroid Build Coastguard Worker 
36*9880d681SAndroid Build Coastguard Worker static cl::opt<unsigned> NumOutputs("j", cl::Prefix, cl::init(2),
37*9880d681SAndroid Build Coastguard Worker                                     cl::desc("Number of output files"));
38*9880d681SAndroid Build Coastguard Worker 
39*9880d681SAndroid Build Coastguard Worker static cl::opt<bool>
40*9880d681SAndroid Build Coastguard Worker     PreserveLocals("preserve-locals", cl::Prefix, cl::init(false),
41*9880d681SAndroid Build Coastguard Worker                    cl::desc("Split without externalizing locals"));
42*9880d681SAndroid Build Coastguard Worker 
main(int argc,char ** argv)43*9880d681SAndroid Build Coastguard Worker int main(int argc, char **argv) {
44*9880d681SAndroid Build Coastguard Worker   LLVMContext Context;
45*9880d681SAndroid Build Coastguard Worker   SMDiagnostic Err;
46*9880d681SAndroid Build Coastguard Worker   cl::ParseCommandLineOptions(argc, argv, "LLVM module splitter\n");
47*9880d681SAndroid Build Coastguard Worker 
48*9880d681SAndroid Build Coastguard Worker   std::unique_ptr<Module> M = parseIRFile(InputFilename, Err, Context);
49*9880d681SAndroid Build Coastguard Worker 
50*9880d681SAndroid Build Coastguard Worker   if (!M) {
51*9880d681SAndroid Build Coastguard Worker     Err.print(argv[0], errs());
52*9880d681SAndroid Build Coastguard Worker     return 1;
53*9880d681SAndroid Build Coastguard Worker   }
54*9880d681SAndroid Build Coastguard Worker 
55*9880d681SAndroid Build Coastguard Worker   unsigned I = 0;
56*9880d681SAndroid Build Coastguard Worker   SplitModule(std::move(M), NumOutputs, [&](std::unique_ptr<Module> MPart) {
57*9880d681SAndroid Build Coastguard Worker     std::error_code EC;
58*9880d681SAndroid Build Coastguard Worker     std::unique_ptr<tool_output_file> Out(new tool_output_file(
59*9880d681SAndroid Build Coastguard Worker         OutputFilename + utostr(I++), EC, sys::fs::F_None));
60*9880d681SAndroid Build Coastguard Worker     if (EC) {
61*9880d681SAndroid Build Coastguard Worker       errs() << EC.message() << '\n';
62*9880d681SAndroid Build Coastguard Worker       exit(1);
63*9880d681SAndroid Build Coastguard Worker     }
64*9880d681SAndroid Build Coastguard Worker 
65*9880d681SAndroid Build Coastguard Worker     verifyModule(*MPart);
66*9880d681SAndroid Build Coastguard Worker     WriteBitcodeToFile(MPart.get(), Out->os());
67*9880d681SAndroid Build Coastguard Worker 
68*9880d681SAndroid Build Coastguard Worker     // Declare success.
69*9880d681SAndroid Build Coastguard Worker     Out->keep();
70*9880d681SAndroid Build Coastguard Worker   }, PreserveLocals);
71*9880d681SAndroid Build Coastguard Worker 
72*9880d681SAndroid Build Coastguard Worker   return 0;
73*9880d681SAndroid Build Coastguard Worker }
74