diff options
| -rw-r--r-- | lib/Module/RaiseAsm.cpp | 23 |
1 files changed, 13 insertions, 10 deletions
diff --git a/lib/Module/RaiseAsm.cpp b/lib/Module/RaiseAsm.cpp index ff4c2467..cd8f7934 100644 --- a/lib/Module/RaiseAsm.cpp +++ b/lib/Module/RaiseAsm.cpp @@ -86,28 +86,31 @@ bool RaiseAsmPass::runOnInstruction(Module &M, Instruction *I) { bool RaiseAsmPass::runOnModule(Module &M) { bool changed = false; - std::string Err; - std::string HostTriple = llvm::sys::getDefaultTargetTriple(); - const Target *NativeTarget = TargetRegistry::lookupTarget(HostTriple, Err); + + // Use target triple from the module if possible. + std::string TargetTriple = M.getTargetTriple(); + if (TargetTriple.empty()) + TargetTriple = llvm::sys::getDefaultTargetTriple(); + const Target *Target = TargetRegistry::lookupTarget(TargetTriple, Err); TargetMachine * TM = 0; - if (NativeTarget == 0) { - klee_warning("Warning: unable to select native target: %s", Err.c_str()); + if (Target == 0) { + klee_warning("Warning: unable to select target: %s", Err.c_str()); TLI = 0; } else { #if LLVM_VERSION_CODE >= LLVM_VERSION(3, 9) - TM = NativeTarget->createTargetMachine(HostTriple, "", "", TargetOptions(), - None); + TM = Target->createTargetMachine(TargetTriple, "", "", TargetOptions(), + None); TLI = TM->getSubtargetImpl(*(M.begin()))->getTargetLowering(); #else - TM = NativeTarget->createTargetMachine(HostTriple, "", "", TargetOptions()); + TM = Target->createTargetMachine(TargetTriple, "", "", TargetOptions()); TLI = TM->getSubtargetImpl(*(M.begin()))->getTargetLowering(); #endif - triple = llvm::Triple(HostTriple); + triple = llvm::Triple(TargetTriple); } - + for (Module::iterator fi = M.begin(), fe = M.end(); fi != fe; ++fi) { for (Function::iterator bi = fi->begin(), be = fi->end(); bi != be; ++bi) { for (BasicBlock::iterator ii = bi->begin(), ie = bi->end(); ii != ie;) { |