lib/ExecutionEngine/JIT/TargetSelect.cpp

   1 //===-- TargetSelect.cpp - Target Chooser Code ----------------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the LLVM research group and is distributed under
   6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file contains the hideously gross code that is currently used to select
  11 // a particular TargetMachine for the JIT to use.  This should obviously be
  12 // improved in the future, probably by having the TargetMachines register
  13 // themselves with the runtime, and then have them choose themselves if they
  14 // match the current machine.
  15 //
  16 //===----------------------------------------------------------------------===//
  17
  18 #include "JIT.h"
  19 #include "llvm/Module.h"
  20 #include "llvm/ModuleProvider.h"
  21 #include "llvm/Target/TargetMachine.h"
  22 #include "llvm/Target/TargetMachineImpls.h"
  23 #include "Support/CommandLine.h"
  24 using namespace llvm;
  25
  26 #if !defined(ENABLE_X86_JIT) && !defined(ENABLE_SPARC_JIT)
  27 #define NO_JITS_ENABLED
  28 #endif
  29
  30 namespace {
  31   enum ArchName { x86, Sparc };
  32
  33 #ifndef NO_JITS_ENABLED
  34   cl::opt<ArchName>
  35   Arch("march", cl::desc("Architecture to JIT to:"), cl::Prefix,
  36        cl::values(
  37 #ifdef ENABLE_X86_JIT
  38                   clEnumVal(x86, "  IA-32 (Pentium and above)"),
  39 #endif
  40 #ifdef ENABLE_SPARC_JIT
  41                   clEnumValN(Sparc, "sparc", "  Sparc-V9"),
  42 #endif
  43                   0),
  44 #if defined(ENABLE_X86_JIT)
  45   cl::init(x86)
  46 #elif defined(ENABLE_SPARC_JIT)
  47   cl::init(Sparc)
  48 #endif
  49        );
  50 #endif /* NO_JITS_ENABLED */
  51 }
  52
  53 /// create - Create an return a new JIT compiler if there is one available
  54 /// for the current target.  Otherwise, return null.
  55 ///
  56 ExecutionEngine *JIT::create(ModuleProvider *MP) {
  57   TargetMachine* (*TargetMachineAllocator)(const Module &) = 0;
  58
  59   // Allow a command-line switch to override what *should* be the default target
  60   // machine for this platform. This allows for debugging a Sparc JIT on X86 --
  61   // our X86 machines are much faster at recompiling LLVM and linking LLI.
  62 #ifndef NO_JITS_ENABLED
  63
  64   switch (Arch) {
  65 #ifdef ENABLE_X86_JIT
  66   case x86:
  67     TargetMachineAllocator = allocateX86TargetMachine;
  68     break;
  69 #endif
  70 #ifdef ENABLE_SPARC_JIT
  71   case Sparc:
  72     TargetMachineAllocator = allocateSparcTargetMachine;
  73     break;
  74 #endif
  75   default:
  76     assert(0 && "-march flag not supported on this host!");
  77   }
  78 #else
  79   return 0;
  80 #endif
  81
  82   // Allocate a target...
  83   TargetMachine *Target = TargetMachineAllocator(*MP->getModule());
  84   assert(Target && "Could not allocate target machine!");
  85
  86   // If the target supports JIT code generation, return a new JIT now.
  87   if (TargetJITInfo *TJ = Target->getJITInfo())
  88     return new JIT(MP, *Target, *TJ);
  89   return 0;
  90 }
  91
  92