diff options
| -rw-r--r-- | tools/driver/EmbeddedRunner.cc | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/tools/driver/EmbeddedRunner.cc b/tools/driver/EmbeddedRunner.cc index cc69849..eb29bac 100644 --- a/tools/driver/EmbeddedRunner.cc +++ b/tools/driver/EmbeddedRunner.cc @@ -354,6 +354,17 @@ int runEmbedded(const cps::CodeGen &CG) { } std::unique_ptr<llvm::orc::LLJIT> J = std::move(*JITOrErr); + // LLJIT's default layer sets these for COFF; our custom JITLink layer + // must do the same or IR-added modules report Unexpected definitions + // for backend-added symbols like __ImageBase. + if (J->getTargetTriple().isOSBinFormatCOFF()) { + if (auto *COFFObjLayer = llvm::dyn_cast<llvm::orc::ObjectLinkingLayer>( + &J->getObjLinkingLayer())) { + COFFObjLayer->setOverrideObjectFlagsWithResponsibilityFlags(true); + COFFObjLayer->setAutoClaimResponsibilityForObjectSymbols(true); + } + } + #if defined(_WIN32) && defined(__x86_64__) // Register MinGW SEH tables through JITLink; failures surface as link // errors below and prevent entry to JIT code. |