X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=tools%2Fllc%2Fllc.cpp;h=357eedd0b68e433540e5915902e951496d39b5c6;hb=966083fd1ad5adf801b767bba008ebaf0b9e06fa;hp=8d9094c4c36cd9dfb4885f9cb87a75d98c888baf;hpb=cd50d3fafabec10c09f3ed0466924a951943d6d0;p=oota-llvm.git diff --git a/tools/llc/llc.cpp b/tools/llc/llc.cpp index 8d9094c4c36..357eedd0b68 100644 --- a/tools/llc/llc.cpp +++ b/tools/llc/llc.cpp @@ -1,45 +1,96 @@ -//===-- llc.cpp - Implement the LLVM Compiler -----------------------------===// +//===-- llc.cpp - Implement the LLVM Native Code Generator ----------------===// // -// This is the llc compiler driver. +// The LLVM Compiler Infrastructure +// +// This file was developed by the LLVM research group and is distributed under +// the University of Illinois Open Source License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// +// +// This is the llc code generator driver. It provides a convenient +// command-line interface for generating native assembly-language code +// or C code, given LLVM bytecode. // //===----------------------------------------------------------------------===// #include "llvm/Bytecode/Reader.h" -#include "llvm/Target/Sparc.h" +#include "llvm/Target/SubtargetFeature.h" #include "llvm/Target/TargetMachine.h" -#include "llvm/Transforms/Instrumentation/TraceValues.h" -#include "llvm/Transforms/ChangeAllocations.h" -#include "llvm/Transforms/HoistPHIConstants.h" -#include "llvm/Assembly/PrintModulePass.h" -#include "llvm/Bytecode/WriteBytecodePass.h" -#include "llvm/Transforms/ConstantMerge.h" +#include "llvm/Target/TargetMachineRegistry.h" +#include "llvm/Transforms/Scalar.h" #include "llvm/Module.h" -#include "llvm/Method.h" #include "llvm/PassManager.h" -#include "Support/CommandLine.h" -#include -#include +#include "llvm/Pass.h" +#include "llvm/Support/CommandLine.h" +#include "llvm/Support/PluginLoader.h" +#include "llvm/Support/FileUtilities.h" +#include "llvm/Analysis/Verifier.h" +#include "llvm/System/Signals.h" +#include "llvm/Config/config.h" #include -using std::string; - -cl::String InputFilename ("", "Input filename", cl::NoFlags, "-"); -cl::String OutputFilename("o", "Output filename", cl::NoFlags, ""); -cl::Flag Force ("f", "Overwrite output files"); -cl::Flag DumpAsm ("d", "Print bytecode before native code generation", - cl::Hidden); -cl::Flag DoNotEmitAssembly("noasm", "Do not emit assembly code", cl::Hidden); -cl::Flag TraceBBValues ("trace", - "Trace values at basic block and method exits"); -cl::Flag TraceMethodValues("tracem", "Trace values only at method exits"); - - -// GetFileNameRoot - Helper function to get the basename of a filename... -static inline string GetFileNameRoot(const string &InputFilename) { - string IFN = InputFilename; - string outputFilename; +#include +#include + +using namespace llvm; + +// General options for llc. Other pass-specific options are specified +// within the corresponding llc passes, and target-specific options +// and back-end code generation options are specified with the target machine. +// +static cl::opt +InputFilename(cl::Positional, cl::desc(""), cl::init("-")); + +static cl::opt +OutputFilename("o", cl::desc("Output filename"), cl::value_desc("filename")); + +static cl::opt Force("f", cl::desc("Overwrite output files")); + +static cl::opt Fast("fast", + cl::desc("Generate code quickly, potentially sacrificing code quality")); + +static cl::opt +TargetTriple("mtriple", cl::desc("Override target triple for module")); + +static cl::opt +MArch("march", cl::desc("Architecture to generate code for:")); + +static cl::opt +MCPU("mcpu", + cl::desc("Target a specific cpu type (-mcpu=help for details)"), + cl::value_desc("cpu-name"), + cl::init("")); + +static cl::list +MAttrs("mattr", + cl::CommaSeparated, + cl::desc("Target specific attributes (-mattr=help for details)"), + cl::value_desc("a1,+a2,-a3,...")); + +cl::opt +FileType("filetype", cl::init(TargetMachine::AssemblyFile), + cl::desc("Choose a file type (not all types are supported by all targets):"), + cl::values( + clEnumValN(TargetMachine::AssemblyFile, "asm", + " Emit an assembly ('.s') file"), + clEnumValN(TargetMachine::ObjectFile, "obj", + " Emit a native object ('.o') file [experimental]"), + clEnumValN(TargetMachine::DynamicLibrary, "dynlib", + " Emit a native dynamic library ('.so') file"), + clEnumValEnd)); + +cl::opt NoVerify("disable-verify", cl::Hidden, + cl::desc("Do not verify input module")); + + +// GetFileNameRoot - Helper function to get the basename of a filename. +static inline std::string +GetFileNameRoot(const std::string &InputFilename) { + std::string IFN = InputFilename; + std::string outputFilename; int Len = IFN.length(); - if (IFN[Len-3] == '.' && IFN[Len-2] == 'b' && IFN[Len-1] == 'c') { - outputFilename = string(IFN.begin(), IFN.end()-3); // s/.bc/.s/ + if ((Len > 2) && + IFN[Len-3] == '.' && IFN[Len-2] == 'b' && IFN[Len-1] == 'c') { + outputFilename = std::string(IFN.begin(), IFN.end()-3); // s/.bc/.s/ } else { outputFilename = IFN; } @@ -47,168 +98,146 @@ static inline string GetFileNameRoot(const string &InputFilename) { } -//===---------------------------------------------------------------------===// -// GenerateCodeForTarget Pass -// -// Native code generation for a specified target. -//===---------------------------------------------------------------------===// - -class GenerateCodeForTarget : public MethodPass { - TargetMachine &Target; -public: - inline GenerateCodeForTarget(TargetMachine &T) : Target(T) {} +// main - Entry point for the llc compiler. +// +int main(int argc, char **argv) { + try { + cl::ParseCommandLineOptions(argc, argv, " llvm system compiler\n"); + sys::PrintStackTraceOnErrorSignal(); + + // Load the module to be compiled... + std::auto_ptr M(ParseBytecodeFile(InputFilename)); + if (M.get() == 0) { + std::cerr << argv[0] << ": bytecode didn't read correctly.\n"; + return 1; + } + Module &mod = *M.get(); - // runOnMethod - This method does the actual work of generating code for - // the specified method. - // - bool runOnMethod(Method *M) { - if (Target.compileMethod(M)) - cerr << "Error compiling " << InputFilename << "!\n"; + // If we are supposed to override the target triple, do so now. + if (!TargetTriple.empty()) + mod.setTargetTriple(TargetTriple); - return true; - } -}; - - -//===---------------------------------------------------------------------===// -// EmitAssembly Pass -// -// Write assembly code to specified output stream -//===---------------------------------------------------------------------===// - -class EmitAssembly : public Pass { - const TargetMachine &Target; // Target to compile for - std::ostream *Out; // Stream to print on - bool DeleteStream; // Delete stream in dtor? -public: - inline EmitAssembly(const TargetMachine &T, std::ostream *O, bool D) - : Target(T), Out(O), DeleteStream(D) {} - - virtual bool run(Module *M) { - Target.emitAssembly(M, *Out); - - if (DeleteStream) delete Out; - return false; - } -}; + // Allocate target machine. First, check whether the user has + // explicitly specified an architecture to compile for. + if (MArch == 0) { + std::string Err; + MArch = TargetMachineRegistry::getClosestStaticTargetForModule(mod, Err); + if (MArch == 0) { + std::cerr << argv[0] << ": error auto-selecting target for module '" + << Err << "'. Please use the -march option to explicitly " + << "pick a target.\n"; + return 1; + } + } + // Package up features to be passed to target/subtarget + std::string FeaturesStr; + if (MCPU.size() || MAttrs.size()) { + SubtargetFeatures Features; + Features.setCPU(MCPU); + for (unsigned i = 0; i != MAttrs.size(); ++i) + Features.AddFeature(MAttrs[i]); + FeaturesStr = Features.getString(); + } + std::auto_ptr target(MArch->CtorFn(mod, FeaturesStr)); + assert(target.get() && "Could not allocate target machine!"); + TargetMachine &Target = *target.get(); + const TargetData &TD = Target.getTargetData(); -//===---------------------------------------------------------------------===// -// Function main() -// -// Entry point for the llc compiler. -//===---------------------------------------------------------------------===// + // Build up all of the passes that we want to do to the module... + PassManager Passes; + Passes.add(new TargetData(TD)); -int main(int argc, char **argv) { - cl::ParseCommandLineOptions(argc, argv, " llvm system compiler\n"); - - // Allocate a target... in the future this will be controllable on the - // command line. - std::auto_ptr target(allocateSparcTargetMachine()); - assert(target.get() && "Could not allocate target machine!"); - - TargetMachine &Target = *target.get(); - - // Load the module to be compiled... - std::auto_ptr M(ParseBytecodeFile(InputFilename)); - if (M.get() == 0) { - cerr << "bytecode didn't read correctly.\n"; - return 1; - } +#ifndef NDEBUG + if(!NoVerify) + Passes.add(createVerifierPass()); +#endif - // Build up all of the passes that we want to do to the module... - PassManager Passes; - - // Hoist constants out of PHI nodes into predecessor BB's - Passes.add(new HoistPHIConstants()); - - if (TraceBBValues || TraceMethodValues) { // If tracing enabled... - // Insert trace code in all methods in the module - Passes.add(new InsertTraceCode(TraceBBValues, - TraceBBValues ||TraceMethodValues)); - - // Eliminate duplication in constant pool - Passes.add(new DynamicConstantMerge()); - - // Then write out the module with tracing code before code generation - assert(InputFilename != "-" && - "files on stdin not supported with tracing"); - string traceFileName = GetFileNameRoot(InputFilename) + ".trace.bc"; - - if (!Force && std::ifstream(OutputFilename.c_str())) { - // If force is not specified, make sure not to overwrite a file! - cerr << "Error opening '" << OutputFilename << "': File exists!\n" - << "Use -f command line argument to force output\n"; - return 1; - } - - std::ostream *os = new std::ofstream(traceFileName.c_str()); - if (!os->good()) { - cerr << "Error opening " << traceFileName - << "! SKIPPING OUTPUT OF TRACE CODE\n"; - delete os; - return 1; - } - - Passes.add(new WriteBytecodePass(os, true)); - } - - // Replace malloc and free instructions with library calls. - // Do this after tracing until lli implements these lib calls. - // For now, it will emulate malloc and free internally. - Passes.add(new LowerAllocations(Target.DataLayout)); - - // If LLVM dumping after transformations is requested, add it to the pipeline - if (DumpAsm) - Passes.add(new PrintMethodPass("Code after xformations: \n",&cerr)); - - // Generate Target code... - Passes.add(new GenerateCodeForTarget(Target)); - - if (!DoNotEmitAssembly) { // If asm output is enabled... // Figure out where we are going to send the output... std::ostream *Out = 0; - if (OutputFilename != "") { // Specified an output filename? - if (!Force && std::ifstream(OutputFilename.c_str())) { - // If force is not specified, make sure not to overwrite a file! - cerr << "Error opening '" << OutputFilename << "': File exists!\n" - << "Use -f command line argument to force output\n"; - return 1; + if (OutputFilename != "") { + if (OutputFilename != "-") { + // Specified an output filename? + if (!Force && std::ifstream(OutputFilename.c_str())) { + // If force is not specified, make sure not to overwrite a file! + std::cerr << argv[0] << ": error opening '" << OutputFilename + << "': file exists!\n" + << "Use -f command line argument to force output\n"; + return 1; + } + Out = new std::ofstream(OutputFilename.c_str()); + + // Make sure that the Out file gets unlinked from the disk if we get a + // SIGINT + sys::RemoveFileOnSignal(sys::Path(OutputFilename)); + } else { + Out = &std::cout; } - Out = new std::ofstream(OutputFilename.c_str()); } else { if (InputFilename == "-") { OutputFilename = "-"; Out = &std::cout; } else { - string OutputFilename = GetFileNameRoot(InputFilename); - OutputFilename += ".s"; + OutputFilename = GetFileNameRoot(InputFilename); + + switch (FileType) { + case TargetMachine::AssemblyFile: + if (MArch->Name[0] != 'c' || MArch->Name[1] != 0) // not CBE + OutputFilename += ".s"; + else + OutputFilename += ".cbe.c"; + break; + case TargetMachine::ObjectFile: + OutputFilename += ".o"; + break; + case TargetMachine::DynamicLibrary: + OutputFilename += LTDL_SHLIB_EXT; + break; + } if (!Force && std::ifstream(OutputFilename.c_str())) { // If force is not specified, make sure not to overwrite a file! - cerr << "Error opening '" << OutputFilename << "': File exists!\n" - << "Use -f command line argument to force output\n"; + std::cerr << argv[0] << ": error opening '" << OutputFilename + << "': file exists!\n" + << "Use -f command line argument to force output\n"; return 1; } Out = new std::ofstream(OutputFilename.c_str()); if (!Out->good()) { - cerr << "Error opening " << OutputFilename << "!\n"; + std::cerr << argv[0] << ": error opening " << OutputFilename << "!\n"; delete Out; return 1; } + + // Make sure that the Out file gets unlinked from the disk if we get a + // SIGINT + sys::RemoveFileOnSignal(sys::Path(OutputFilename)); } } - // Output assembly language to the .s file - Passes.add(new EmitAssembly(Target, Out, Out != &std::cout)); - } - - // Run our queue of passes all at once now, efficiently. - Passes.run(M.get()); - - return 0; -} + // Ask the target to add backend passes as necessary. + if (Target.addPassesToEmitFile(Passes, *Out, FileType, Fast)) { + std::cerr << argv[0] << ": target '" << Target.getName() + << "' does not support generation of this file type!\n"; + if (Out != &std::cout) delete Out; + // And the Out file is empty and useless, so remove it now. + sys::Path(OutputFilename).eraseFromDisk(); + return 1; + } else { + // Run our queue of passes all at once now, efficiently. + Passes.run(*M.get()); + } + // Delete the ostream if it's not a stdout stream + if (Out != &std::cout) delete Out; + return 0; + } catch (const std::string& msg) { + std::cerr << argv[0] << ": " << msg << "\n"; + } catch (...) { + std::cerr << argv[0] << ": Unexpected unknown exception occurred.\n"; + } + return 1; +}