LLVM 19.0.0git
LLJIT.cpp
Go to the documentation of this file.
1//===--------- LLJIT.cpp - An ORC-based JIT for compiling LLVM IR ---------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
25#include "llvm/IR/IRBuilder.h"
26#include "llvm/IR/Mangler.h"
27#include "llvm/IR/Module.h"
29
30#define DEBUG_TYPE "orc"
31
32using namespace llvm;
33using namespace llvm::orc;
34
35namespace {
36
37/// Adds helper function decls and wrapper functions that call the helper with
38/// some additional prefix arguments.
39///
40/// E.g. For wrapper "foo" with type i8(i8, i64), helper "bar", and prefix
41/// args i32 4 and i16 12345, this function will add:
42///
43/// declare i8 @bar(i32, i16, i8, i64)
44///
45/// define i8 @foo(i8, i64) {
46/// entry:
47/// %2 = call i8 @bar(i32 4, i16 12345, i8 %0, i64 %1)
48/// ret i8 %2
49/// }
50///
51Function *addHelperAndWrapper(Module &M, StringRef WrapperName,
52 FunctionType *WrapperFnType,
53 GlobalValue::VisibilityTypes WrapperVisibility,
54 StringRef HelperName,
55 ArrayRef<Value *> HelperPrefixArgs) {
56 std::vector<Type *> HelperArgTypes;
57 for (auto *Arg : HelperPrefixArgs)
58 HelperArgTypes.push_back(Arg->getType());
59 for (auto *T : WrapperFnType->params())
60 HelperArgTypes.push_back(T);
61 auto *HelperFnType =
62 FunctionType::get(WrapperFnType->getReturnType(), HelperArgTypes, false);
63 auto *HelperFn = Function::Create(HelperFnType, GlobalValue::ExternalLinkage,
64 HelperName, M);
65
66 auto *WrapperFn = Function::Create(
67 WrapperFnType, GlobalValue::ExternalLinkage, WrapperName, M);
68 WrapperFn->setVisibility(WrapperVisibility);
69
70 auto *EntryBlock = BasicBlock::Create(M.getContext(), "entry", WrapperFn);
71 IRBuilder<> IB(EntryBlock);
72
73 std::vector<Value *> HelperArgs;
74 for (auto *Arg : HelperPrefixArgs)
75 HelperArgs.push_back(Arg);
76 for (auto &Arg : WrapperFn->args())
77 HelperArgs.push_back(&Arg);
78 auto *HelperResult = IB.CreateCall(HelperFn, HelperArgs);
79 if (HelperFn->getReturnType()->isVoidTy())
80 IB.CreateRetVoid();
81 else
82 IB.CreateRet(HelperResult);
83
84 return WrapperFn;
85}
86
87class GenericLLVMIRPlatformSupport;
88
89/// orc::Platform component of Generic LLVM IR Platform support.
90/// Just forwards calls to the GenericLLVMIRPlatformSupport class below.
91class GenericLLVMIRPlatform : public Platform {
92public:
93 GenericLLVMIRPlatform(GenericLLVMIRPlatformSupport &S) : S(S) {}
94 Error setupJITDylib(JITDylib &JD) override;
95 Error teardownJITDylib(JITDylib &JD) override;
97 const MaterializationUnit &MU) override;
99 // Noop -- Nothing to do (yet).
100 return Error::success();
101 }
102
103private:
104 GenericLLVMIRPlatformSupport &S;
105};
106
107/// This transform parses llvm.global_ctors to produce a single initialization
108/// function for the module, records the function, then deletes
109/// llvm.global_ctors.
110class GlobalCtorDtorScraper {
111public:
112 GlobalCtorDtorScraper(GenericLLVMIRPlatformSupport &PS,
113 StringRef InitFunctionPrefix,
114 StringRef DeInitFunctionPrefix)
115 : PS(PS), InitFunctionPrefix(InitFunctionPrefix),
116 DeInitFunctionPrefix(DeInitFunctionPrefix) {}
119
120private:
121 GenericLLVMIRPlatformSupport &PS;
122 StringRef InitFunctionPrefix;
123 StringRef DeInitFunctionPrefix;
124};
125
126/// Generic IR Platform Support
127///
128/// Scrapes llvm.global_ctors and llvm.global_dtors and replaces them with
129/// specially named 'init' and 'deinit'. Injects definitions / interposes for
130/// some runtime API, including __cxa_atexit, dlopen, and dlclose.
131class GenericLLVMIRPlatformSupport : public LLJIT::PlatformSupport {
132public:
133 GenericLLVMIRPlatformSupport(LLJIT &J, JITDylib &PlatformJD)
134 : J(J), InitFunctionPrefix(J.mangle("__orc_init_func.")),
135 DeInitFunctionPrefix(J.mangle("__orc_deinit_func.")) {
136
138 std::make_unique<GenericLLVMIRPlatform>(*this));
139
140 setInitTransform(J, GlobalCtorDtorScraper(*this, InitFunctionPrefix,
141 DeInitFunctionPrefix));
142
143 SymbolMap StdInterposes;
144
145 StdInterposes[J.mangleAndIntern("__lljit.platform_support_instance")] = {
147 StdInterposes[J.mangleAndIntern("__lljit.cxa_atexit_helper")] = {
148 ExecutorAddr::fromPtr(registerCxaAtExitHelper), JITSymbolFlags()};
149
150 cantFail(PlatformJD.define(absoluteSymbols(std::move(StdInterposes))));
151 cantFail(setupJITDylib(PlatformJD));
152 cantFail(J.addIRModule(PlatformJD, createPlatformRuntimeModule()));
153 }
154
156
157 /// Adds a module that defines the __dso_handle global.
158 Error setupJITDylib(JITDylib &JD) {
159
160 // Add per-jitdylib standard interposes.
161 SymbolMap PerJDInterposes;
162 PerJDInterposes[J.mangleAndIntern("__lljit.run_atexits_helper")] = {
163 ExecutorAddr::fromPtr(runAtExitsHelper), JITSymbolFlags()};
164 PerJDInterposes[J.mangleAndIntern("__lljit.atexit_helper")] = {
165 ExecutorAddr::fromPtr(registerAtExitHelper), JITSymbolFlags()};
166 cantFail(JD.define(absoluteSymbols(std::move(PerJDInterposes))));
167
168 auto Ctx = std::make_unique<LLVMContext>();
169 auto M = std::make_unique<Module>("__standard_lib", *Ctx);
170 M->setDataLayout(J.getDataLayout());
171
172 auto *Int64Ty = Type::getInt64Ty(*Ctx);
173 auto *DSOHandle = new GlobalVariable(
174 *M, Int64Ty, true, GlobalValue::ExternalLinkage,
175 ConstantInt::get(Int64Ty, reinterpret_cast<uintptr_t>(&JD)),
176 "__dso_handle");
177 DSOHandle->setVisibility(GlobalValue::DefaultVisibility);
178 DSOHandle->setInitializer(
179 ConstantInt::get(Int64Ty, ExecutorAddr::fromPtr(&JD).getValue()));
180
181 auto *GenericIRPlatformSupportTy =
182 StructType::create(*Ctx, "lljit.GenericLLJITIRPlatformSupport");
183
184 auto *PlatformInstanceDecl = new GlobalVariable(
185 *M, GenericIRPlatformSupportTy, true, GlobalValue::ExternalLinkage,
186 nullptr, "__lljit.platform_support_instance");
187
188 auto *VoidTy = Type::getVoidTy(*Ctx);
189 addHelperAndWrapper(
190 *M, "__lljit_run_atexits", FunctionType::get(VoidTy, {}, false),
191 GlobalValue::HiddenVisibility, "__lljit.run_atexits_helper",
192 {PlatformInstanceDecl, DSOHandle});
193
194 auto *IntTy = Type::getIntNTy(*Ctx, sizeof(int) * CHAR_BIT);
195 auto *AtExitCallbackTy = FunctionType::get(VoidTy, {}, false);
196 auto *AtExitCallbackPtrTy = PointerType::getUnqual(AtExitCallbackTy);
197 addHelperAndWrapper(*M, "atexit",
198 FunctionType::get(IntTy, {AtExitCallbackPtrTy}, false),
199 GlobalValue::HiddenVisibility, "__lljit.atexit_helper",
200 {PlatformInstanceDecl, DSOHandle});
201
202 return J.addIRModule(JD, ThreadSafeModule(std::move(M), std::move(Ctx)));
203 }
204
205 Error notifyAdding(ResourceTracker &RT, const MaterializationUnit &MU) {
206 auto &JD = RT.getJITDylib();
207 if (auto &InitSym = MU.getInitializerSymbol())
208 InitSymbols[&JD].add(InitSym, SymbolLookupFlags::WeaklyReferencedSymbol);
209 else {
210 // If there's no identified init symbol attached, but there is a symbol
211 // with the GenericIRPlatform::InitFunctionPrefix, then treat that as
212 // an init function. Add the symbol to both the InitSymbols map (which
213 // will trigger a lookup to materialize the module) and the InitFunctions
214 // map (which holds the names of the symbols to execute).
215 for (auto &KV : MU.getSymbols())
216 if ((*KV.first).starts_with(InitFunctionPrefix)) {
217 InitSymbols[&JD].add(KV.first,
218 SymbolLookupFlags::WeaklyReferencedSymbol);
219 InitFunctions[&JD].add(KV.first);
220 } else if ((*KV.first).starts_with(DeInitFunctionPrefix)) {
221 DeInitFunctions[&JD].add(KV.first);
222 }
223 }
224 return Error::success();
225 }
226
227 Error initialize(JITDylib &JD) override {
228 LLVM_DEBUG({
229 dbgs() << "GenericLLVMIRPlatformSupport getting initializers to run\n";
230 });
231 if (auto Initializers = getInitializers(JD)) {
233 { dbgs() << "GenericLLVMIRPlatformSupport running initializers\n"; });
234 for (auto InitFnAddr : *Initializers) {
235 LLVM_DEBUG({
236 dbgs() << " Running init " << formatv("{0:x16}", InitFnAddr)
237 << "...\n";
238 });
239 auto *InitFn = InitFnAddr.toPtr<void (*)()>();
240 InitFn();
241 }
242 } else
243 return Initializers.takeError();
244 return Error::success();
245 }
246
247 Error deinitialize(JITDylib &JD) override {
248 LLVM_DEBUG({
249 dbgs() << "GenericLLVMIRPlatformSupport getting deinitializers to run\n";
250 });
251 if (auto Deinitializers = getDeinitializers(JD)) {
252 LLVM_DEBUG({
253 dbgs() << "GenericLLVMIRPlatformSupport running deinitializers\n";
254 });
255 for (auto DeinitFnAddr : *Deinitializers) {
256 LLVM_DEBUG({
257 dbgs() << " Running deinit " << formatv("{0:x16}", DeinitFnAddr)
258 << "...\n";
259 });
260 auto *DeinitFn = DeinitFnAddr.toPtr<void (*)()>();
261 DeinitFn();
262 }
263 } else
264 return Deinitializers.takeError();
265
266 return Error::success();
267 }
268
269 void registerInitFunc(JITDylib &JD, SymbolStringPtr InitName) {
271 InitFunctions[&JD].add(InitName);
272 });
273 }
274
275 void registerDeInitFunc(JITDylib &JD, SymbolStringPtr DeInitName) {
277 [&]() { DeInitFunctions[&JD].add(DeInitName); });
278 }
279
280private:
281 Expected<std::vector<ExecutorAddr>> getInitializers(JITDylib &JD) {
282 if (auto Err = issueInitLookups(JD))
283 return std::move(Err);
284
286 std::vector<JITDylibSP> DFSLinkOrder;
287
288 if (auto Err = getExecutionSession().runSessionLocked([&]() -> Error {
289 if (auto DFSLinkOrderOrErr = JD.getDFSLinkOrder())
290 DFSLinkOrder = std::move(*DFSLinkOrderOrErr);
291 else
292 return DFSLinkOrderOrErr.takeError();
293
294 for (auto &NextJD : DFSLinkOrder) {
295 auto IFItr = InitFunctions.find(NextJD.get());
296 if (IFItr != InitFunctions.end()) {
297 LookupSymbols[NextJD.get()] = std::move(IFItr->second);
298 InitFunctions.erase(IFItr);
299 }
300 }
301 return Error::success();
302 }))
303 return std::move(Err);
304
305 LLVM_DEBUG({
306 dbgs() << "JITDylib init order is [ ";
307 for (auto &JD : llvm::reverse(DFSLinkOrder))
308 dbgs() << "\"" << JD->getName() << "\" ";
309 dbgs() << "]\n";
310 dbgs() << "Looking up init functions:\n";
311 for (auto &KV : LookupSymbols)
312 dbgs() << " \"" << KV.first->getName() << "\": " << KV.second << "\n";
313 });
314
315 auto &ES = getExecutionSession();
316 auto LookupResult = Platform::lookupInitSymbols(ES, LookupSymbols);
317
318 if (!LookupResult)
319 return LookupResult.takeError();
320
321 std::vector<ExecutorAddr> Initializers;
322 while (!DFSLinkOrder.empty()) {
323 auto &NextJD = *DFSLinkOrder.back();
324 DFSLinkOrder.pop_back();
325 auto InitsItr = LookupResult->find(&NextJD);
326 if (InitsItr == LookupResult->end())
327 continue;
328 for (auto &KV : InitsItr->second)
329 Initializers.push_back(KV.second.getAddress());
330 }
331
332 return Initializers;
333 }
334
335 Expected<std::vector<ExecutorAddr>> getDeinitializers(JITDylib &JD) {
336 auto &ES = getExecutionSession();
337
338 auto LLJITRunAtExits = J.mangleAndIntern("__lljit_run_atexits");
339
341 std::vector<JITDylibSP> DFSLinkOrder;
342
343 if (auto Err = ES.runSessionLocked([&]() -> Error {
344 if (auto DFSLinkOrderOrErr = JD.getDFSLinkOrder())
345 DFSLinkOrder = std::move(*DFSLinkOrderOrErr);
346 else
347 return DFSLinkOrderOrErr.takeError();
348
349 for (auto &NextJD : DFSLinkOrder) {
350 auto &JDLookupSymbols = LookupSymbols[NextJD.get()];
351 auto DIFItr = DeInitFunctions.find(NextJD.get());
352 if (DIFItr != DeInitFunctions.end()) {
353 LookupSymbols[NextJD.get()] = std::move(DIFItr->second);
354 DeInitFunctions.erase(DIFItr);
355 }
356 JDLookupSymbols.add(LLJITRunAtExits,
357 SymbolLookupFlags::WeaklyReferencedSymbol);
358 }
359 return Error::success();
360 }))
361 return std::move(Err);
362
363 LLVM_DEBUG({
364 dbgs() << "JITDylib deinit order is [ ";
365 for (auto &JD : DFSLinkOrder)
366 dbgs() << "\"" << JD->getName() << "\" ";
367 dbgs() << "]\n";
368 dbgs() << "Looking up deinit functions:\n";
369 for (auto &KV : LookupSymbols)
370 dbgs() << " \"" << KV.first->getName() << "\": " << KV.second << "\n";
371 });
372
373 auto LookupResult = Platform::lookupInitSymbols(ES, LookupSymbols);
374
375 if (!LookupResult)
376 return LookupResult.takeError();
377
378 std::vector<ExecutorAddr> DeInitializers;
379 for (auto &NextJD : DFSLinkOrder) {
380 auto DeInitsItr = LookupResult->find(NextJD.get());
381 assert(DeInitsItr != LookupResult->end() &&
382 "Every JD should have at least __lljit_run_atexits");
383
384 auto RunAtExitsItr = DeInitsItr->second.find(LLJITRunAtExits);
385 if (RunAtExitsItr != DeInitsItr->second.end())
386 DeInitializers.push_back(RunAtExitsItr->second.getAddress());
387
388 for (auto &KV : DeInitsItr->second)
389 if (KV.first != LLJITRunAtExits)
390 DeInitializers.push_back(KV.second.getAddress());
391 }
392
393 return DeInitializers;
394 }
395
396 /// Issue lookups for all init symbols required to initialize JD (and any
397 /// JITDylibs that it depends on).
398 Error issueInitLookups(JITDylib &JD) {
399 DenseMap<JITDylib *, SymbolLookupSet> RequiredInitSymbols;
400 std::vector<JITDylibSP> DFSLinkOrder;
401
402 if (auto Err = getExecutionSession().runSessionLocked([&]() -> Error {
403 if (auto DFSLinkOrderOrErr = JD.getDFSLinkOrder())
404 DFSLinkOrder = std::move(*DFSLinkOrderOrErr);
405 else
406 return DFSLinkOrderOrErr.takeError();
407
408 for (auto &NextJD : DFSLinkOrder) {
409 auto ISItr = InitSymbols.find(NextJD.get());
410 if (ISItr != InitSymbols.end()) {
411 RequiredInitSymbols[NextJD.get()] = std::move(ISItr->second);
412 InitSymbols.erase(ISItr);
413 }
414 }
415 return Error::success();
416 }))
417 return Err;
418
419 return Platform::lookupInitSymbols(getExecutionSession(),
420 RequiredInitSymbols)
421 .takeError();
422 }
423
424 static void registerCxaAtExitHelper(void *Self, void (*F)(void *), void *Ctx,
425 void *DSOHandle) {
426 LLVM_DEBUG({
427 dbgs() << "Registering cxa atexit function " << (void *)F << " for JD "
428 << (*static_cast<JITDylib **>(DSOHandle))->getName() << "\n";
429 });
430 static_cast<GenericLLVMIRPlatformSupport *>(Self)->AtExitMgr.registerAtExit(
431 F, Ctx, DSOHandle);
432 }
433
434 static void registerAtExitHelper(void *Self, void *DSOHandle, void (*F)()) {
435 LLVM_DEBUG({
436 dbgs() << "Registering atexit function " << (void *)F << " for JD "
437 << (*static_cast<JITDylib **>(DSOHandle))->getName() << "\n";
438 });
439 static_cast<GenericLLVMIRPlatformSupport *>(Self)->AtExitMgr.registerAtExit(
440 reinterpret_cast<void (*)(void *)>(F), nullptr, DSOHandle);
441 }
442
443 static void runAtExitsHelper(void *Self, void *DSOHandle) {
444 LLVM_DEBUG({
445 dbgs() << "Running atexit functions for JD "
446 << (*static_cast<JITDylib **>(DSOHandle))->getName() << "\n";
447 });
448 static_cast<GenericLLVMIRPlatformSupport *>(Self)->AtExitMgr.runAtExits(
449 DSOHandle);
450 }
451
452 // Constructs an LLVM IR module containing platform runtime globals,
453 // functions, and interposes.
454 ThreadSafeModule createPlatformRuntimeModule() {
455 auto Ctx = std::make_unique<LLVMContext>();
456 auto M = std::make_unique<Module>("__standard_lib", *Ctx);
457 M->setDataLayout(J.getDataLayout());
458
459 auto *GenericIRPlatformSupportTy =
460 StructType::create(*Ctx, "lljit.GenericLLJITIRPlatformSupport");
461
462 auto *PlatformInstanceDecl = new GlobalVariable(
463 *M, GenericIRPlatformSupportTy, true, GlobalValue::ExternalLinkage,
464 nullptr, "__lljit.platform_support_instance");
465
466 auto *Int8Ty = Type::getInt8Ty(*Ctx);
467 auto *IntTy = Type::getIntNTy(*Ctx, sizeof(int) * CHAR_BIT);
468 auto *VoidTy = Type::getVoidTy(*Ctx);
469 auto *BytePtrTy = PointerType::getUnqual(Int8Ty);
470 auto *CxaAtExitCallbackTy = FunctionType::get(VoidTy, {BytePtrTy}, false);
471 auto *CxaAtExitCallbackPtrTy = PointerType::getUnqual(CxaAtExitCallbackTy);
472
473 addHelperAndWrapper(
474 *M, "__cxa_atexit",
475 FunctionType::get(IntTy, {CxaAtExitCallbackPtrTy, BytePtrTy, BytePtrTy},
476 false),
477 GlobalValue::DefaultVisibility, "__lljit.cxa_atexit_helper",
478 {PlatformInstanceDecl});
479
480 return ThreadSafeModule(std::move(M), std::move(Ctx));
481 }
482
483 LLJIT &J;
484 std::string InitFunctionPrefix;
485 std::string DeInitFunctionPrefix;
489 ItaniumCXAAtExitSupport AtExitMgr;
490};
491
492Error GenericLLVMIRPlatform::setupJITDylib(JITDylib &JD) {
493 return S.setupJITDylib(JD);
494}
495
496Error GenericLLVMIRPlatform::teardownJITDylib(JITDylib &JD) {
497 return Error::success();
498}
499
500Error GenericLLVMIRPlatform::notifyAdding(ResourceTracker &RT,
501 const MaterializationUnit &MU) {
502 return S.notifyAdding(RT, MU);
503}
504
506GlobalCtorDtorScraper::operator()(ThreadSafeModule TSM,
508 auto Err = TSM.withModuleDo([&](Module &M) -> Error {
509 auto &Ctx = M.getContext();
510 auto *GlobalCtors = M.getNamedGlobal("llvm.global_ctors");
511 auto *GlobalDtors = M.getNamedGlobal("llvm.global_dtors");
512
513 auto RegisterCOrDtors = [&](GlobalVariable *GlobalCOrDtors,
514 bool isCtor) -> Error {
515 // If there's no llvm.global_c/dtor or it's just a decl then skip.
516 if (!GlobalCOrDtors || GlobalCOrDtors->isDeclaration())
517 return Error::success();
518 std::string InitOrDeInitFunctionName;
519 if (isCtor)
520 raw_string_ostream(InitOrDeInitFunctionName)
521 << InitFunctionPrefix << M.getModuleIdentifier();
522 else
523 raw_string_ostream(InitOrDeInitFunctionName)
524 << DeInitFunctionPrefix << M.getModuleIdentifier();
525
526 MangleAndInterner Mangle(PS.getExecutionSession(), M.getDataLayout());
527 auto InternedInitOrDeInitName = Mangle(InitOrDeInitFunctionName);
528 if (auto Err = R.defineMaterializing(
529 {{InternedInitOrDeInitName, JITSymbolFlags::Callable}}))
530 return Err;
531
532 auto *InitOrDeInitFunc = Function::Create(
533 FunctionType::get(Type::getVoidTy(Ctx), {}, false),
534 GlobalValue::ExternalLinkage, InitOrDeInitFunctionName, &M);
535 InitOrDeInitFunc->setVisibility(GlobalValue::HiddenVisibility);
536 std::vector<std::pair<Function *, unsigned>> InitsOrDeInits;
537 auto COrDtors = isCtor ? getConstructors(M) : getDestructors(M);
538
539 for (auto E : COrDtors)
540 InitsOrDeInits.push_back(std::make_pair(E.Func, E.Priority));
541 llvm::sort(InitsOrDeInits, llvm::less_second());
542
543 auto *InitOrDeInitFuncEntryBlock =
544 BasicBlock::Create(Ctx, "entry", InitOrDeInitFunc);
545 IRBuilder<> IB(InitOrDeInitFuncEntryBlock);
546 for (auto &KV : InitsOrDeInits)
547 IB.CreateCall(KV.first);
548 IB.CreateRetVoid();
549
550 if (isCtor)
551 PS.registerInitFunc(R.getTargetJITDylib(), InternedInitOrDeInitName);
552 else
553 PS.registerDeInitFunc(R.getTargetJITDylib(), InternedInitOrDeInitName);
554
555 GlobalCOrDtors->eraseFromParent();
556 return Error::success();
557 };
558
559 if (auto Err = RegisterCOrDtors(GlobalCtors, true))
560 return Err;
561 if (auto Err = RegisterCOrDtors(GlobalDtors, false))
562 return Err;
563
564 return Error::success();
565 });
566
567 if (Err)
568 return std::move(Err);
569
570 return std::move(TSM);
571}
572
573/// Inactive Platform Support
574///
575/// Explicitly disables platform support. JITDylibs are not scanned for special
576/// init/deinit symbols. No runtime API interposes are injected.
577class InactivePlatformSupport : public LLJIT::PlatformSupport {
578public:
579 InactivePlatformSupport() = default;
580
581 Error initialize(JITDylib &JD) override {
582 LLVM_DEBUG(dbgs() << "InactivePlatformSupport: no initializers running for "
583 << JD.getName() << "\n");
584 return Error::success();
585 }
586
587 Error deinitialize(JITDylib &JD) override {
589 dbgs() << "InactivePlatformSupport: no deinitializers running for "
590 << JD.getName() << "\n");
591 return Error::success();
592 }
593};
594
595} // end anonymous namespace
596
597namespace llvm {
598namespace orc {
599
603 using SPSDLOpenSig = SPSExecutorAddr(SPSString, int32_t);
604 enum dlopen_mode : int32_t {
605 ORC_RT_RTLD_LAZY = 0x1,
606 ORC_RT_RTLD_NOW = 0x2,
607 ORC_RT_RTLD_LOCAL = 0x4,
608 ORC_RT_RTLD_GLOBAL = 0x8
609 };
610
611 auto &ES = J.getExecutionSession();
612 auto MainSearchOrder = J.getMainJITDylib().withLinkOrderDo(
613 [](const JITDylibSearchOrder &SO) { return SO; });
614
615 if (auto WrapperAddr = ES.lookup(
616 MainSearchOrder, J.mangleAndIntern("__orc_rt_jit_dlopen_wrapper"))) {
617 return ES.callSPSWrapper<SPSDLOpenSig>(WrapperAddr->getAddress(),
618 DSOHandles[&JD], JD.getName(),
619 int32_t(ORC_RT_RTLD_LAZY));
620 } else
621 return WrapperAddr.takeError();
622}
623
626 using SPSDLCloseSig = int32_t(SPSExecutorAddr);
627
628 auto &ES = J.getExecutionSession();
629 auto MainSearchOrder = J.getMainJITDylib().withLinkOrderDo(
630 [](const JITDylibSearchOrder &SO) { return SO; });
631
632 if (auto WrapperAddr = ES.lookup(
633 MainSearchOrder, J.mangleAndIntern("__orc_rt_jit_dlclose_wrapper"))) {
634 int32_t result;
635 auto E = J.getExecutionSession().callSPSWrapper<SPSDLCloseSig>(
636 WrapperAddr->getAddress(), result, DSOHandles[&JD]);
637 if (E)
638 return E;
639 else if (result)
640 return make_error<StringError>("dlclose failed",
642 DSOHandles.erase(&JD);
643 } else
644 return WrapperAddr.takeError();
645 return Error::success();
646}
647
650 J.InitHelperTransformLayer->setTransform(std::move(T));
651}
652
654
656
657 LLVM_DEBUG(dbgs() << "Preparing to create LLJIT instance...\n");
658
659 if (!JTMB) {
660 LLVM_DEBUG({
661 dbgs() << " No explicitly set JITTargetMachineBuilder. "
662 "Detecting host...\n";
663 });
664 if (auto JTMBOrErr = JITTargetMachineBuilder::detectHost())
665 JTMB = std::move(*JTMBOrErr);
666 else
667 return JTMBOrErr.takeError();
668 }
669
670 if ((ES || EPC) && NumCompileThreads)
671 return make_error<StringError>(
672 "NumCompileThreads cannot be used with a custom ExecutionSession or "
673 "ExecutorProcessControl",
675
676#if !LLVM_ENABLE_THREADS
677 if (NumCompileThreads)
678 return make_error<StringError>(
679 "LLJIT num-compile-threads is " + Twine(NumCompileThreads) +
680 " but LLVM was compiled with LLVM_ENABLE_THREADS=Off",
682#endif // !LLVM_ENABLE_THREADS
683
684 // Only used in debug builds.
685 [[maybe_unused]] bool ConcurrentCompilationSettingDefaulted =
686 !SupportConcurrentCompilation;
687
688 if (!SupportConcurrentCompilation) {
689#if LLVM_ENABLE_THREADS
690 SupportConcurrentCompilation = NumCompileThreads || ES || EPC;
691#else
692 SupportConcurrentCompilation = false;
693#endif // LLVM_ENABLE_THREADS
694 } else {
695#if !LLVM_ENABLE_THREADS
696 if (*SupportConcurrentCompilation)
697 return make_error<StringError>(
698 "LLJIT concurrent compilation support requested, but LLVM was built "
699 "with LLVM_ENABLE_THREADS=Off",
701#endif // !LLVM_ENABLE_THREADS
702 }
703
704 LLVM_DEBUG({
705 dbgs() << " JITTargetMachineBuilder is "
706 << JITTargetMachineBuilderPrinter(*JTMB, " ")
707 << " Pre-constructed ExecutionSession: " << (ES ? "Yes" : "No")
708 << "\n"
709 << " DataLayout: ";
710 if (DL)
711 dbgs() << DL->getStringRepresentation() << "\n";
712 else
713 dbgs() << "None (will be created by JITTargetMachineBuilder)\n";
714
715 dbgs() << " Custom object-linking-layer creator: "
716 << (CreateObjectLinkingLayer ? "Yes" : "No") << "\n"
717 << " Custom compile-function creator: "
718 << (CreateCompileFunction ? "Yes" : "No") << "\n"
719 << " Custom platform-setup function: "
720 << (SetUpPlatform ? "Yes" : "No") << "\n"
721 << " Support concurrent compilation: "
722 << (*SupportConcurrentCompilation ? "Yes" : "No");
723 if (ConcurrentCompilationSettingDefaulted)
724 dbgs() << " (defaulted based on ES / EPC / NumCompileThreads)\n";
725 else
726 dbgs() << "\n";
727 dbgs() << " Number of compile threads: " << NumCompileThreads << "\n";
728 });
729
730 // Create DL if not specified.
731 if (!DL) {
732 if (auto DLOrErr = JTMB->getDefaultDataLayoutForTarget())
733 DL = std::move(*DLOrErr);
734 else
735 return DLOrErr.takeError();
736 }
737
738 // If neither ES nor EPC has been set then create an EPC instance.
739 if (!ES && !EPC) {
740 LLVM_DEBUG({
741 dbgs() << "ExecutorProcessControl not specified, "
742 "Creating SelfExecutorProcessControl instance\n";
743 });
744
745 std::unique_ptr<TaskDispatcher> D = nullptr;
746#if LLVM_ENABLE_THREADS
747 if (*SupportConcurrentCompilation) {
748 std::optional<size_t> NumThreads = std ::nullopt;
749 if (NumCompileThreads)
750 NumThreads = NumCompileThreads;
751 D = std::make_unique<DynamicThreadPoolTaskDispatcher>(NumThreads);
752 } else
753 D = std::make_unique<InPlaceTaskDispatcher>();
754#endif // LLVM_ENABLE_THREADS
755 if (auto EPCOrErr =
756 SelfExecutorProcessControl::Create(nullptr, std::move(D), nullptr))
757 EPC = std::move(*EPCOrErr);
758 else
759 return EPCOrErr.takeError();
760 } else if (EPC) {
761 LLVM_DEBUG({
762 dbgs() << "Using explicitly specified ExecutorProcessControl instance "
763 << EPC.get() << "\n";
764 });
765 } else {
766 LLVM_DEBUG({
767 dbgs() << "Using explicitly specified ExecutionSession instance "
768 << ES.get() << "\n";
769 });
770 }
771
772 // If the client didn't configure any linker options then auto-configure the
773 // JIT linker.
774 if (!CreateObjectLinkingLayer) {
775 auto &TT = JTMB->getTargetTriple();
776 bool UseJITLink = false;
777 switch (TT.getArch()) {
778 case Triple::riscv64:
780 UseJITLink = true;
781 break;
782 case Triple::aarch64:
783 UseJITLink = !TT.isOSBinFormatCOFF();
784 break;
785 case Triple::arm:
786 case Triple::armeb:
787 case Triple::thumb:
788 case Triple::thumbeb:
789 UseJITLink = TT.isOSBinFormatELF();
790 break;
791 case Triple::x86_64:
792 UseJITLink = !TT.isOSBinFormatCOFF();
793 break;
794 case Triple::ppc64:
795 UseJITLink = TT.isPPC64ELFv2ABI();
796 break;
797 case Triple::ppc64le:
798 UseJITLink = TT.isOSBinFormatELF();
799 break;
800 default:
801 break;
802 }
803 if (UseJITLink) {
804 JTMB->setRelocationModel(Reloc::PIC_);
805 JTMB->setCodeModel(CodeModel::Small);
806 CreateObjectLinkingLayer =
808 const Triple &) -> Expected<std::unique_ptr<ObjectLayer>> {
809 auto ObjLinkingLayer = std::make_unique<ObjectLinkingLayer>(ES);
810 if (auto EHFrameRegistrar = EPCEHFrameRegistrar::Create(ES))
811 ObjLinkingLayer->addPlugin(
812 std::make_unique<EHFrameRegistrationPlugin>(
813 ES, std::move(*EHFrameRegistrar)));
814 else
815 return EHFrameRegistrar.takeError();
816 return std::move(ObjLinkingLayer);
817 };
818 }
819 }
820
821 // If we need a process JITDylib but no setup function has been given then
822 // create a default one.
823 if (!SetupProcessSymbolsJITDylib && LinkProcessSymbolsByDefault) {
824 LLVM_DEBUG(dbgs() << "Creating default Process JD setup function\n");
825 SetupProcessSymbolsJITDylib = [](LLJIT &J) -> Expected<JITDylibSP> {
826 auto &JD =
827 J.getExecutionSession().createBareJITDylib("<Process Symbols>");
829 J.getExecutionSession());
830 if (!G)
831 return G.takeError();
832 JD.addGenerator(std::move(*G));
833 return &JD;
834 };
835 }
836
837 return Error::success();
838}
839
841 if (auto Err = ES->endSession())
842 ES->reportError(std::move(Err));
843}
844
846
848
850 auto JD = ES->createJITDylib(std::move(Name));
851 if (!JD)
852 return JD.takeError();
853
855 return JD;
856}
857
860 if (!G)
861 return G.takeError();
862
863 if (auto *ExistingJD = ES->getJITDylibByName(Path))
864 return *ExistingJD;
865
866 auto &JD = ES->createBareJITDylib(Path);
867 JD.addGenerator(std::move(*G));
868 return JD;
869}
870
872 std::unique_ptr<MemoryBuffer> LibBuffer) {
874 std::move(LibBuffer));
875 if (!G)
876 return G.takeError();
877
878 JD.addGenerator(std::move(*G));
879
880 return Error::success();
881}
882
885 if (!G)
886 return G.takeError();
887
888 JD.addGenerator(std::move(*G));
889
890 return Error::success();
891}
892
894 assert(TSM && "Can not add null module");
895
896 if (auto Err =
897 TSM.withModuleDo([&](Module &M) { return applyDataLayout(M); }))
898 return Err;
899
900 return InitHelperTransformLayer->add(std::move(RT), std::move(TSM));
901}
902
904 return addIRModule(JD.getDefaultResourceTracker(), std::move(TSM));
905}
906
908 std::unique_ptr<MemoryBuffer> Obj) {
909 assert(Obj && "Can not add null object");
910
911 return ObjTransformLayer->add(std::move(RT), std::move(Obj));
912}
913
914Error LLJIT::addObjectFile(JITDylib &JD, std::unique_ptr<MemoryBuffer> Obj) {
915 return addObjectFile(JD.getDefaultResourceTracker(), std::move(Obj));
916}
917
920 if (auto Sym = ES->lookup(
922 Name))
923 return Sym->getAddress();
924 else
925 return Sym.takeError();
926}
927
930
931 // If the config state provided an ObjectLinkingLayer factory then use it.
933 return S.CreateObjectLinkingLayer(ES, S.JTMB->getTargetTriple());
934
935 // Otherwise default to creating an RTDyldObjectLinkingLayer that constructs
936 // a new SectionMemoryManager for each object.
937 auto GetMemMgr = []() { return std::make_unique<SectionMemoryManager>(); };
938 auto Layer =
939 std::make_unique<RTDyldObjectLinkingLayer>(ES, std::move(GetMemMgr));
940
941 if (S.JTMB->getTargetTriple().isOSBinFormatCOFF()) {
942 Layer->setOverrideObjectFlagsWithResponsibilityFlags(true);
943 Layer->setAutoClaimResponsibilityForObjectSymbols(true);
944 }
945
946 if (S.JTMB->getTargetTriple().isOSBinFormatELF() &&
947 (S.JTMB->getTargetTriple().getArch() == Triple::ArchType::ppc64 ||
948 S.JTMB->getTargetTriple().getArch() == Triple::ArchType::ppc64le))
949 Layer->setAutoClaimResponsibilityForObjectSymbols(true);
950
951 // FIXME: Explicit conversion to std::unique_ptr<ObjectLayer> added to silence
952 // errors from some GCC / libstdc++ bots. Remove this conversion (i.e.
953 // just return ObjLinkingLayer) once those bots are upgraded.
954 return std::unique_ptr<ObjectLayer>(std::move(Layer));
955}
956
960
961 /// If there is a custom compile function creator set then use it.
963 return S.CreateCompileFunction(std::move(JTMB));
964
965 // If using a custom EPC then use a ConcurrentIRCompiler by default.
967 return std::make_unique<ConcurrentIRCompiler>(std::move(JTMB));
968
969 auto TM = JTMB.createTargetMachine();
970 if (!TM)
971 return TM.takeError();
972
973 return std::make_unique<TMOwningSimpleCompiler>(std::move(*TM));
974}
975
977 : DL(std::move(*S.DL)), TT(S.JTMB->getTargetTriple()) {
978
980
981 assert(!(S.EPC && S.ES) && "EPC and ES should not both be set");
982
983 if (S.EPC) {
984 ES = std::make_unique<ExecutionSession>(std::move(S.EPC));
985 } else if (S.ES)
986 ES = std::move(S.ES);
987 else {
988 if (auto EPC = SelfExecutorProcessControl::Create()) {
989 ES = std::make_unique<ExecutionSession>(std::move(*EPC));
990 } else {
991 Err = EPC.takeError();
992 return;
993 }
994 }
995
996 auto ObjLayer = createObjectLinkingLayer(S, *ES);
997 if (!ObjLayer) {
998 Err = ObjLayer.takeError();
999 return;
1000 }
1001 ObjLinkingLayer = std::move(*ObjLayer);
1003 std::make_unique<ObjectTransformLayer>(*ES, *ObjLinkingLayer);
1004
1005 {
1006 auto CompileFunction = createCompileFunction(S, std::move(*S.JTMB));
1007 if (!CompileFunction) {
1008 Err = CompileFunction.takeError();
1009 return;
1010 }
1011 CompileLayer = std::make_unique<IRCompileLayer>(
1012 *ES, *ObjTransformLayer, std::move(*CompileFunction));
1013 TransformLayer = std::make_unique<IRTransformLayer>(*ES, *CompileLayer);
1015 std::make_unique<IRTransformLayer>(*ES, *TransformLayer);
1016 }
1017
1019 InitHelperTransformLayer->setCloneToNewContextOnEmit(true);
1020
1022 if (auto ProcSymsJD = S.SetupProcessSymbolsJITDylib(*this)) {
1023 ProcessSymbols = ProcSymsJD->get();
1024 } else {
1025 Err = ProcSymsJD.takeError();
1026 return;
1027 }
1028 }
1029
1030 if (S.PrePlatformSetup) {
1031 if (auto Err2 = S.PrePlatformSetup(*this)) {
1032 Err = std::move(Err2);
1033 return;
1034 }
1035 }
1036
1037 if (!S.SetUpPlatform)
1039
1040 if (auto PlatformJDOrErr = S.SetUpPlatform(*this)) {
1041 Platform = PlatformJDOrErr->get();
1042 if (Platform)
1043 DefaultLinks.push_back(
1045 } else {
1046 Err = PlatformJDOrErr.takeError();
1047 return;
1048 }
1049
1051 DefaultLinks.push_back(
1053
1054 if (auto MainOrErr = createJITDylib("main"))
1055 Main = &*MainOrErr;
1056 else {
1057 Err = MainOrErr.takeError();
1058 return;
1059 }
1060}
1061
1062std::string LLJIT::mangle(StringRef UnmangledName) const {
1063 std::string MangledName;
1064 {
1065 raw_string_ostream MangledNameStream(MangledName);
1066 Mangler::getNameWithPrefix(MangledNameStream, UnmangledName, DL);
1067 }
1068 return MangledName;
1069}
1070
1072 if (M.getDataLayout().isDefault())
1073 M.setDataLayout(DL);
1074
1075 if (M.getDataLayout() != DL)
1076 return make_error<StringError>(
1077 "Added modules have incompatible data layouts: " +
1078 M.getDataLayout().getStringRepresentation() + " (module) vs " +
1079 DL.getStringRepresentation() + " (jit)",
1081
1082 return Error::success();
1083}
1084
1086 LLVM_DEBUG({ dbgs() << "Setting up orc platform support for LLJIT\n"; });
1087 J.setPlatformSupport(std::make_unique<ORCPlatformSupport>(J));
1088 return Error::success();
1089}
1090
1092public:
1095 if (!DLLName.ends_with_insensitive(".dll"))
1096 return make_error<StringError>("DLLName not ending with .dll",
1098 auto DLLNameStr = DLLName.str(); // Guarantees null-termination.
1099 auto DLLJD = J.loadPlatformDynamicLibrary(DLLNameStr.c_str());
1100 if (!DLLJD)
1101 return DLLJD.takeError();
1102 JD.addToLinkOrder(*DLLJD);
1103 return Error::success();
1104 }
1105
1106private:
1107 LLJIT &J;
1108};
1109
1111 auto ProcessSymbolsJD = J.getProcessSymbolsJITDylib();
1112 if (!ProcessSymbolsJD)
1113 return make_error<StringError>(
1114 "Native platforms require a process symbols JITDylib",
1116
1117 const Triple &TT = J.getTargetTriple();
1118 ObjectLinkingLayer *ObjLinkingLayer =
1119 dyn_cast<ObjectLinkingLayer>(&J.getObjLinkingLayer());
1120
1121 if (!ObjLinkingLayer)
1122 return make_error<StringError>(
1123 "ExecutorNativePlatform requires ObjectLinkingLayer",
1125
1126 std::unique_ptr<MemoryBuffer> RuntimeArchiveBuffer;
1127 if (OrcRuntime.index() == 0) {
1128 auto A = errorOrToExpected(MemoryBuffer::getFile(std::get<0>(OrcRuntime)));
1129 if (!A)
1130 return A.takeError();
1131 RuntimeArchiveBuffer = std::move(*A);
1132 } else
1133 RuntimeArchiveBuffer = std::move(std::get<1>(OrcRuntime));
1134
1135 auto &ES = J.getExecutionSession();
1136 auto &PlatformJD = ES.createBareJITDylib("<Platform>");
1137 PlatformJD.addToLinkOrder(*ProcessSymbolsJD);
1138
1139 J.setPlatformSupport(std::make_unique<ORCPlatformSupport>(J));
1140
1141 switch (TT.getObjectFormat()) {
1142 case Triple::COFF: {
1143 const char *VCRuntimePath = nullptr;
1144 bool StaticVCRuntime = false;
1145 if (VCRuntime) {
1146 VCRuntimePath = VCRuntime->first.c_str();
1147 StaticVCRuntime = VCRuntime->second;
1148 }
1149 if (auto P = COFFPlatform::Create(
1150 ES, *ObjLinkingLayer, PlatformJD, std::move(RuntimeArchiveBuffer),
1151 LoadAndLinkDynLibrary(J), StaticVCRuntime, VCRuntimePath))
1152 J.getExecutionSession().setPlatform(std::move(*P));
1153 else
1154 return P.takeError();
1155 break;
1156 }
1157 case Triple::ELF: {
1159 *ObjLinkingLayer, std::move(RuntimeArchiveBuffer));
1160 if (!G)
1161 return G.takeError();
1162
1163 if (auto P = ELFNixPlatform::Create(ES, *ObjLinkingLayer, PlatformJD,
1164 std::move(*G)))
1165 J.getExecutionSession().setPlatform(std::move(*P));
1166 else
1167 return P.takeError();
1168 break;
1169 }
1170 case Triple::MachO: {
1172 *ObjLinkingLayer, std::move(RuntimeArchiveBuffer));
1173 if (!G)
1174 return G.takeError();
1175
1176 if (auto P = MachOPlatform::Create(ES, *ObjLinkingLayer, PlatformJD,
1177 std::move(*G)))
1178 ES.setPlatform(std::move(*P));
1179 else
1180 return P.takeError();
1181 break;
1182 }
1183 default:
1184 return make_error<StringError>("Unsupported object format in triple " +
1185 TT.str(),
1187 }
1188
1189 return &PlatformJD;
1190}
1191
1193 LLVM_DEBUG(
1194 { dbgs() << "Setting up GenericLLVMIRPlatform support for LLJIT\n"; });
1195 auto ProcessSymbolsJD = J.getProcessSymbolsJITDylib();
1196 if (!ProcessSymbolsJD)
1197 return make_error<StringError>(
1198 "Native platforms require a process symbols JITDylib",
1200
1201 auto &PlatformJD = J.getExecutionSession().createBareJITDylib("<Platform>");
1202 PlatformJD.addToLinkOrder(*ProcessSymbolsJD);
1203
1205 std::make_unique<GenericLLVMIRPlatformSupport>(J, PlatformJD));
1206
1207 return &PlatformJD;
1208}
1209
1211 LLVM_DEBUG(
1212 { dbgs() << "Explicitly deactivated platform support for LLJIT\n"; });
1213 J.setPlatformSupport(std::make_unique<InactivePlatformSupport>());
1214 return nullptr;
1215}
1216
1219 return Err;
1220 TT = JTMB->getTargetTriple();
1221 return Error::success();
1222}
1223
1225 assert(TSM && "Can not add null module");
1226
1227 if (auto Err = TSM.withModuleDo(
1228 [&](Module &M) -> Error { return applyDataLayout(M); }))
1229 return Err;
1230
1231 return CODLayer->add(JD, std::move(TSM));
1232}
1233
1234LLLazyJIT::LLLazyJIT(LLLazyJITBuilderState &S, Error &Err) : LLJIT(S, Err) {
1235
1236 // If LLJIT construction failed then bail out.
1237 if (Err)
1238 return;
1239
1240 ErrorAsOutParameter _(&Err);
1241
1242 /// Take/Create the lazy-compile callthrough manager.
1243 if (S.LCTMgr)
1244 LCTMgr = std::move(S.LCTMgr);
1245 else {
1246 if (auto LCTMgrOrErr = createLocalLazyCallThroughManager(
1248 LCTMgr = std::move(*LCTMgrOrErr);
1249 else {
1250 Err = LCTMgrOrErr.takeError();
1251 return;
1252 }
1253 }
1254
1255 // Take/Create the indirect stubs manager builder.
1256 auto ISMBuilder = std::move(S.ISMBuilder);
1257
1258 // If none was provided, try to build one.
1259 if (!ISMBuilder)
1261
1262 // No luck. Bail out.
1263 if (!ISMBuilder) {
1264 Err = make_error<StringError>("Could not construct "
1265 "IndirectStubsManagerBuilder for target " +
1266 S.TT.str(),
1268 return;
1269 }
1270
1271 // Create the COD layer.
1272 CODLayer = std::make_unique<CompileOnDemandLayer>(
1273 *ES, *InitHelperTransformLayer, *LCTMgr, std::move(ISMBuilder));
1274
1276 CODLayer->setCloneToNewContextOnEmit(true);
1277}
1278
1279// In-process LLJIT uses eh-frame section wrappers via EPC, so we need to force
1280// them to be linked in.
1284}
1285
1286} // End namespace orc.
1287} // End namespace llvm.
for(const MachineOperand &MO :llvm::drop_begin(OldMI.operands(), Desc.getNumOperands()))
static GCRegistry::Add< ErlangGC > A("erlang", "erlang-compatible garbage collector")
static GCRegistry::Add< StatepointGC > D("statepoint-example", "an example strategy for statepoint")
#define LLVM_ATTRIBUTE_USED
Definition: Compiler.h:151
#define LLVM_DEBUG(X)
Definition: Debug.h:101
std::string Name
Symbol * Sym
Definition: ELF_riscv.cpp:479
#define _
#define F(x, y, z)
Definition: MD5.cpp:55
#define G(x, y, z)
Definition: MD5.cpp:56
Module.h This file contains the declarations for the Module class.
#define P(N)
if(VerifyEach)
const char LLVMTargetMachineRef TM
static StringRef getName(Value *V)
llvm::orc::shared::CWrapperFunctionResult llvm_orc_deregisterEHFrameSectionWrapper(const char *Data, uint64_t Size)
llvm::orc::shared::CWrapperFunctionResult llvm_orc_registerEHFrameSectionWrapper(const char *Data, uint64_t Size)
assert(ImpDefSCC.getReg()==AMDGPU::SCC &&ImpDefSCC.isDef())
ArrayRef - Represent a constant reference to an array (0 or more elements consecutively in memory),...
Definition: ArrayRef.h:41
static BasicBlock * Create(LLVMContext &Context, const Twine &Name="", Function *Parent=nullptr, BasicBlock *InsertBefore=nullptr)
Creates a new BasicBlock.
Definition: BasicBlock.h:199
const std::string & getStringRepresentation() const
Returns the string representation of the DataLayout.
Definition: DataLayout.h:246
Helper for Errors used as out-parameters.
Definition: Error.h:1102
Lightweight error class with error context and mandatory checking.
Definition: Error.h:160
static ErrorSuccess success()
Create a success value.
Definition: Error.h:334
Tagged union holding either a T or a Error.
Definition: Error.h:474
static FunctionType * get(Type *Result, ArrayRef< Type * > Params, bool isVarArg)
This static method is the primary way of constructing a FunctionType.
static Function * Create(FunctionType *Ty, LinkageTypes Linkage, unsigned AddrSpace, const Twine &N="", Module *M=nullptr)
Definition: Function.h:163
bool isDeclaration() const
Return true if the primary definition of this global value is outside of the current translation unit...
Definition: Globals.cpp:281
VisibilityTypes
An enumeration for the kinds of visibility of global values.
Definition: GlobalValue.h:66
@ DefaultVisibility
The GV is visible.
Definition: GlobalValue.h:67
@ HiddenVisibility
The GV is hidden.
Definition: GlobalValue.h:68
@ ExternalLinkage
Externally visible function.
Definition: GlobalValue.h:52
void eraseFromParent()
eraseFromParent - This method unlinks 'this' from the containing module and deletes it.
Definition: Globals.cpp:462
This provides a uniform API for creating instructions and inserting them into a basic block: either a...
Definition: IRBuilder.h:2666
Flags for symbols in the JIT.
Definition: JITSymbol.h:74
void getNameWithPrefix(raw_ostream &OS, const GlobalValue *GV, bool CannotUsePrivateLabel) const
Print the appropriate prefix and the specified global variable's name.
Definition: Mangler.cpp:120
static ErrorOr< std::unique_ptr< MemoryBuffer > > getFile(const Twine &Filename, bool IsText=false, bool RequiresNullTerminator=true, bool IsVolatile=false, std::optional< Align > Alignment=std::nullopt)
Open the specified file as a MemoryBuffer, returning a new MemoryBuffer if successful,...
A Module instance is used to store all the information related to an LLVM module.
Definition: Module.h:65
static PointerType * getUnqual(Type *ElementType)
This constructs a pointer to an object of the specified type in the default address space (address sp...
Definition: DerivedTypes.h:662
StringRef - Represent a constant reference to a string, i.e.
Definition: StringRef.h:50
std::string str() const
str - Get the contents as an std::string.
Definition: StringRef.h:222
bool ends_with_insensitive(StringRef Suffix) const
Check if this string ends with the given Suffix, ignoring case.
Definition: StringRef.cpp:50
static StructType * create(LLVMContext &Context, StringRef Name)
This creates an identified struct.
Definition: Type.cpp:513
Triple - Helper class for working with autoconf configuration names.
Definition: Triple.h:44
bool isPPC64ELFv2ABI() const
Tests whether the target 64-bit PowerPC big endian ABI is ELFv2.
Definition: Triple.h:964
@ loongarch64
Definition: Triple.h:62
ArchType getArch() const
Get the parsed architecture type of this triple.
Definition: Triple.h:372
bool isOSBinFormatCOFF() const
Tests whether the OS uses the COFF binary format.
Definition: Triple.h:719
const std::string & str() const
Definition: Triple.h:435
bool isOSBinFormatELF() const
Tests whether the OS uses the ELF binary format.
Definition: Triple.h:714
Twine - A lightweight data structure for efficiently representing the concatenation of temporary valu...
Definition: Twine.h:81
static IntegerType * getIntNTy(LLVMContext &C, unsigned N)
static Type * getVoidTy(LLVMContext &C)
static IntegerType * getInt8Ty(LLVMContext &C)
static IntegerType * getInt64Ty(LLVMContext &C)
static Expected< std::unique_ptr< COFFPlatform > > Create(ExecutionSession &ES, ObjectLinkingLayer &ObjLinkingLayer, JITDylib &PlatformJD, std::unique_ptr< MemoryBuffer > OrcRuntimeArchiveBuffer, LoadDynamicLibrary LoadDynLibrary, bool StaticVCRuntime=false, const char *VCRuntimePath=nullptr, std::optional< SymbolAliasMap > RuntimeAliases=std::nullopt)
Try to create a COFFPlatform instance, adding the ORC runtime to the given JITDylib.
static Expected< std::unique_ptr< ELFNixPlatform > > Create(ExecutionSession &ES, ObjectLinkingLayer &ObjLinkingLayer, JITDylib &PlatformJD, std::unique_ptr< DefinitionGenerator > OrcRuntime, std::optional< SymbolAliasMap > RuntimeAliases=std::nullopt)
Try to create a ELFNixPlatform instance, adding the ORC runtime to the given JITDylib.
static Expected< std::unique_ptr< EPCDynamicLibrarySearchGenerator > > Load(ExecutionSession &ES, const char *LibraryPath, SymbolPredicate Allow=SymbolPredicate(), AddAbsoluteSymbolsFn AddAbsoluteSymbols=nullptr)
Permanently loads the library at the given path and, on success, returns a DynamicLibrarySearchGenera...
static Expected< std::unique_ptr< EPCDynamicLibrarySearchGenerator > > GetForTargetProcess(ExecutionSession &ES, SymbolPredicate Allow=SymbolPredicate(), AddAbsoluteSymbolsFn AddAbsoluteSymbols=nullptr)
Creates a EPCDynamicLibrarySearchGenerator that searches for symbols in the target process.
static Expected< std::unique_ptr< EPCEHFrameRegistrar > > Create(ExecutionSession &ES)
Create from a ExecutorProcessControl instance alone.
An ExecutionSession represents a running JIT program.
Definition: Core.h:1431
void setPlatform(std::unique_ptr< Platform > P)
Set the Platform for this ExecutionSession.
Definition: Core.h:1488
Error callSPSWrapper(ExecutorAddr WrapperFnAddr, WrapperCallArgTs &&...WrapperCallArgs)
Run a wrapper function using SPS to serialize the arguments and deserialize the results.
Definition: Core.h:1685
JITDylib & createBareJITDylib(std::string Name)
Add a new bare JITDylib to this ExecutionSession.
Definition: Core.cpp:1666
decltype(auto) runSessionLocked(Func &&F)
Run the given lambda with the session mutex locked.
Definition: Core.h:1495
static ExecutorAddr fromPtr(T *Ptr, UnwrapFn &&Unwrap=UnwrapFn())
Create an ExecutorAddr from the given pointer.
Expected< JITDylibSP > operator()(LLJIT &J)
Definition: LLJIT.cpp:1110
An interface for Itanium __cxa_atexit interposer implementations.
Represents a JIT'd dynamic library.
Definition: Core.h:989
Error define(std::unique_ptr< MaterializationUnitType > &&MU, ResourceTrackerSP RT=nullptr)
Define all symbols provided by the materialization unit to be part of this JITDylib.
Definition: Core.h:1910
ExecutionSession & getExecutionSession() const
Get a reference to the ExecutionSession for this JITDylib.
Definition: Core.h:1008
void addToLinkOrder(const JITDylibSearchOrder &NewLinks)
Append the given JITDylibSearchOrder to the link order for this JITDylib (discarding any elements alr...
Definition: Core.cpp:1031
static Expected< std::vector< JITDylibSP > > getDFSLinkOrder(ArrayRef< JITDylibSP > JDs)
Returns the given JITDylibs and all of their transitive dependencies in DFS order (based on linkage r...
Definition: Core.cpp:1724
auto withLinkOrderDo(Func &&F) -> decltype(F(std::declval< const JITDylibSearchOrder & >()))
Do something with the link order (run under the session lock).
Definition: Core.h:1903
ResourceTrackerSP getDefaultResourceTracker()
Get the default resource tracker for this JITDylib.
Definition: Core.cpp:691
GeneratorT & addGenerator(std::unique_ptr< GeneratorT > DefGenerator)
Adds a definition generator to this JITDylib and returns a referenece to it.
Definition: Core.h:1893
A utility class for building TargetMachines for JITs.
static Expected< JITTargetMachineBuilder > detectHost()
Create a JITTargetMachineBuilder for the host system.
Expected< std::unique_ptr< TargetMachine > > createTargetMachine()
Create a TargetMachine.
Error prepareForConstruction()
Called prior to JIT class construcion to fix up defaults.
Definition: LLJIT.cpp:655
ProcessSymbolsJITDylibSetupFunction SetupProcessSymbolsJITDylib
Definition: LLJIT.h:320
ObjectLinkingLayerCreator CreateObjectLinkingLayer
Definition: LLJIT.h:321
std::unique_ptr< ExecutionSession > ES
Definition: LLJIT.h:316
unique_function< Error(LLJIT &)> PrePlatformSetup
Definition: LLJIT.h:323
CompileFunctionCreator CreateCompileFunction
Definition: LLJIT.h:322
std::optional< bool > SupportConcurrentCompilation
Definition: LLJIT.h:327
std::unique_ptr< ExecutorProcessControl > EPC
Definition: LLJIT.h:315
std::optional< JITTargetMachineBuilder > JTMB
Definition: LLJIT.h:317
PlatformSetupFunction SetUpPlatform
Definition: LLJIT.h:324
Initializer support for LLJIT.
Definition: LLJIT.h:45
virtual Error deinitialize(JITDylib &JD)=0
virtual Error initialize(JITDylib &JD)=0
static void setInitTransform(LLJIT &J, IRTransformLayer::TransformFunction T)
Definition: LLJIT.cpp:648
A pre-fabricated ORC JIT stack that can serve as an alternative to MCJIT.
Definition: LLJIT.h:38
static Expected< std::unique_ptr< ObjectLayer > > createObjectLinkingLayer(LLJITBuilderState &S, ExecutionSession &ES)
Definition: LLJIT.cpp:929
void setPlatformSupport(std::unique_ptr< PlatformSupport > PS)
Set the PlatformSupport instance.
Definition: LLJIT.h:185
std::unique_ptr< ExecutionSession > ES
Definition: LLJIT.h:246
LLJIT(LLJITBuilderState &S, Error &Err)
Create an LLJIT instance with a single compile thread.
Definition: LLJIT.cpp:976
Error addObjectFile(ResourceTrackerSP RT, std::unique_ptr< MemoryBuffer > Obj)
Adds an object file to the given JITDylib.
Definition: LLJIT.cpp:907
Expected< JITDylib & > createJITDylib(std::string Name)
Create a new JITDylib with the given name and return a reference to it.
Definition: LLJIT.cpp:849
JITDylib & getMainJITDylib()
Returns a reference to the JITDylib representing the JIT'd main program.
Definition: LLJIT.h:72
JITDylibSearchOrder DefaultLinks
Definition: LLJIT.h:253
const DataLayout & getDataLayout() const
Returns a reference to the DataLayout for this instance.
Definition: LLJIT.h:69
ObjectLayer & getObjLinkingLayer()
Returns a reference to the ObjLinkingLayer.
Definition: LLJIT.h:213
std::unique_ptr< ObjectTransformLayer > ObjTransformLayer
Definition: LLJIT.h:259
friend Expected< JITDylibSP > setUpGenericLLVMIRPlatform(LLJIT &J)
Configure the LLJIT instance to scrape modules for llvm.global_ctors and llvm.global_dtors variables ...
Definition: LLJIT.cpp:1192
virtual ~LLJIT()
Destruct this instance.
Definition: LLJIT.cpp:840
std::string mangle(StringRef UnmangledName) const
Returns a linker-mangled version of UnmangledName.
Definition: LLJIT.cpp:1062
JITDylib * Main
Definition: LLJIT.h:251
JITDylibSP getPlatformJITDylib()
Returns the Platform JITDylib, which will contain the ORC runtime (if given) and any platform symbols...
Definition: LLJIT.cpp:847
Expected< JITDylib & > loadPlatformDynamicLibrary(const char *Path)
Load a (real) dynamic library and make its symbols available through a new JITDylib with the same nam...
Definition: LLJIT.cpp:858
std::unique_ptr< IRTransformLayer > InitHelperTransformLayer
Definition: LLJIT.h:262
std::unique_ptr< IRCompileLayer > CompileLayer
Definition: LLJIT.h:260
const Triple & getTargetTriple() const
Returns a reference to the triple for this instance.
Definition: LLJIT.h:66
JITDylibSP getProcessSymbolsJITDylib()
Returns the ProcessSymbols JITDylib, which by default reflects non-JIT'd symbols in the host process.
Definition: LLJIT.cpp:845
Expected< ExecutorAddr > lookupLinkerMangled(JITDylib &JD, SymbolStringPtr Name)
Look up a symbol in JITDylib JD by the symbol's linker-mangled name (to look up symbols based on thei...
Definition: LLJIT.cpp:918
static Expected< std::unique_ptr< IRCompileLayer::IRCompiler > > createCompileFunction(LLJITBuilderState &S, JITTargetMachineBuilder JTMB)
Definition: LLJIT.cpp:958
JITDylib * ProcessSymbols
Definition: LLJIT.h:249
JITDylib * Platform
Definition: LLJIT.h:250
ExecutionSession & getExecutionSession()
Returns the ExecutionSession for this instance.
Definition: LLJIT.h:63
std::unique_ptr< IRTransformLayer > TransformLayer
Definition: LLJIT.h:261
SymbolStringPtr mangleAndIntern(StringRef UnmangledName) const
Returns an interned, linker-mangled version of UnmangledName.
Definition: LLJIT.h:228
DataLayout DL
Definition: LLJIT.h:255
Error linkStaticLibraryInto(JITDylib &JD, std::unique_ptr< MemoryBuffer > LibBuffer)
Link a static library into the given JITDylib.
Definition: LLJIT.cpp:871
Error applyDataLayout(Module &M)
Definition: LLJIT.cpp:1071
std::unique_ptr< ObjectLayer > ObjLinkingLayer
Definition: LLJIT.h:258
Triple TT
Definition: LLJIT.h:256
Error addIRModule(ResourceTrackerSP RT, ThreadSafeModule TSM)
Adds an IR module with the given ResourceTracker.
Definition: LLJIT.cpp:893
ExecutorAddr LazyCompileFailureAddr
Definition: LLJIT.h:522
std::unique_ptr< LazyCallThroughManager > LCTMgr
Definition: LLJIT.h:523
IndirectStubsManagerBuilderFunction ISMBuilder
Definition: LLJIT.h:524
Error addLazyIRModule(JITDylib &JD, ThreadSafeModule M)
Add a module to be lazily compiled to JITDylib JD.
Definition: LLJIT.cpp:1224
Error operator()(JITDylib &JD, StringRef DLLName)
Definition: LLJIT.cpp:1094
static Expected< std::unique_ptr< MachOPlatform > > Create(ExecutionSession &ES, ObjectLinkingLayer &ObjLinkingLayer, JITDylib &PlatformJD, std::unique_ptr< DefinitionGenerator > OrcRuntime, HeaderOptions PlatformJDOpts={}, MachOHeaderMUBuilder BuildMachOHeaderMU=buildSimpleMachOHeaderMU, std::optional< SymbolAliasMap > RuntimeAliases=std::nullopt)
Try to create a MachOPlatform instance, adding the ORC runtime to the given JITDylib.
Mangles symbol names then uniques them in the context of an ExecutionSession.
Definition: Mangling.h:26
Tracks responsibility for materialization, and mediates interactions between MaterializationUnits and...
Definition: Core.h:555
A MaterializationUnit represents a set of symbol definitions that can be materialized as a group,...
Definition: Core.h:693
const SymbolFlagsMap & getSymbols() const
Return the set of symbols that this source provides.
Definition: Core.h:723
const SymbolStringPtr & getInitializerSymbol() const
Returns the initialization symbol for this MaterializationUnit (if any).
Definition: Core.h:726
Error deinitialize(orc::JITDylib &JD) override
Definition: LLJIT.cpp:624
Error initialize(orc::JITDylib &JD) override
Definition: LLJIT.cpp:600
An ObjectLayer implementation built on JITLink.
Platforms set up standard symbols and mediate interactions between dynamic initializers (e....
Definition: Core.h:1360
virtual Error teardownJITDylib(JITDylib &JD)=0
This method will be called outside the session lock each time a JITDylib is removed to allow the Plat...
virtual Error notifyRemoving(ResourceTracker &RT)=0
This method will be called under the ExecutionSession lock when a ResourceTracker is removed.
static Expected< DenseMap< JITDylib *, SymbolMap > > lookupInitSymbols(ExecutionSession &ES, const DenseMap< JITDylib *, SymbolLookupSet > &InitSyms)
A utility function for looking up initializer symbols.
Definition: Core.cpp:1503
virtual Error notifyAdding(ResourceTracker &RT, const MaterializationUnit &MU)=0
This method will be called under the ExecutionSession lock each time a MaterializationUnit is added t...
virtual Error setupJITDylib(JITDylib &JD)=0
This method will be called outside the session lock each time a JITDylib is created (unless it is cre...
API to remove / transfer ownership of JIT resources.
Definition: Core.h:56
JITDylib & getJITDylib() const
Return the JITDylib targeted by this tracker.
Definition: Core.h:71
static Expected< std::unique_ptr< SelfExecutorProcessControl > > Create(std::shared_ptr< SymbolStringPool > SSP=nullptr, std::unique_ptr< TaskDispatcher > D=nullptr, std::unique_ptr< jitlink::JITLinkMemoryManager > MemMgr=nullptr)
Create a SelfExecutorProcessControl with the given symbol string pool and memory manager.
static Expected< std::unique_ptr< StaticLibraryDefinitionGenerator > > Load(ObjectLayer &L, const char *FileName, GetObjectFileInterface GetObjFileInterface=GetObjectFileInterface())
Try to create a StaticLibraryDefinitionGenerator from the given path.
static Expected< std::unique_ptr< StaticLibraryDefinitionGenerator > > Create(ObjectLayer &L, std::unique_ptr< MemoryBuffer > ArchiveBuffer, std::unique_ptr< object::Archive > Archive, GetObjectFileInterface GetObjFileInterface=GetObjectFileInterface())
Try to create a StaticLibrarySearchGenerator from the given memory buffer and Archive object.
Pointer to a pooled string representing a symbol name.
An LLVM Module together with a shared ThreadSafeContext.
decltype(auto) withModuleDo(Func &&F)
Locks the associated ThreadSafeContext and calls the given function on the contained Module.
A raw_ostream that writes to an std::string.
Definition: raw_ostream.h:660
JITDylibSearchOrder makeJITDylibSearchOrder(ArrayRef< JITDylib * > JDs, JITDylibLookupFlags Flags=JITDylibLookupFlags::MatchExportedSymbolsOnly)
Convenience function for creating a search order from an ArrayRef of JITDylib*, all with the same fla...
Definition: Core.h:166
iterator_range< CtorDtorIterator > getDestructors(const Module &M)
Create an iterator range over the entries of the llvm.global_ctors array.
std::vector< std::pair< JITDylib *, JITDylibLookupFlags > > JITDylibSearchOrder
A list of (JITDylib*, JITDylibLookupFlags) pairs to be used as a search order during symbol lookup.
Definition: Core.h:162
std::unique_ptr< AbsoluteSymbolsMaterializationUnit > absoluteSymbols(SymbolMap Symbols)
Create an AbsoluteSymbolsMaterializationUnit with the given symbols.
Definition: Core.h:791
iterator_range< CtorDtorIterator > getConstructors(const Module &M)
Create an iterator range over the entries of the llvm.global_ctors array.
Expected< JITDylibSP > setUpInactivePlatform(LLJIT &J)
Configure the LLJIT instance to disable platform support explicitly.
Definition: LLJIT.cpp:1210
LLVM_ATTRIBUTE_USED void linkComponents()
Definition: LLJIT.cpp:1281
std::function< std::unique_ptr< IndirectStubsManager >()> createLocalIndirectStubsManagerBuilder(const Triple &T)
Create a local indirect stubs manager builder.
Expected< std::unique_ptr< LazyCallThroughManager > > createLocalLazyCallThroughManager(const Triple &T, ExecutionSession &ES, ExecutorAddr ErrorHandlerAddr)
Create a LocalLazyCallThroughManager from the given triple and execution session.
Expected< JITDylibSP > setUpGenericLLVMIRPlatform(LLJIT &J)
Configure the LLJIT instance to scrape modules for llvm.global_ctors and llvm.global_dtors variables ...
Definition: LLJIT.cpp:1192
Error setUpOrcPlatformManually(LLJIT &J)
Configure the LLJIT instance to use orc runtime support.
Definition: LLJIT.cpp:1085
This is an optimization pass for GlobalISel generic memory operations.
Definition: AddressRanges.h:18
std::error_code inconvertibleErrorCode()
The value returned by this function can be returned from convertToErrorCode for Error values where no...
Definition: Error.cpp:90
auto formatv(const char *Fmt, Ts &&...Vals) -> formatv_object< decltype(std::make_tuple(support::detail::build_format_adapter(std::forward< Ts >(Vals))...))>
auto reverse(ContainerTy &&C)
Definition: STLExtras.h:419
void sort(IteratorTy Start, IteratorTy End)
Definition: STLExtras.h:1647
raw_ostream & dbgs()
dbgs() - This returns a reference to a raw_ostream for debugging messages.
Definition: Debug.cpp:163
raw_fd_ostream & errs()
This returns a reference to a raw_ostream for standard error.
void cantFail(Error Err, const char *Msg=nullptr)
Report a fatal error if Err is a failure value.
Definition: Error.h:749
Expected< T > errorOrToExpected(ErrorOr< T > &&EO)
Convert an ErrorOr<T> to an Expected<T>.
Definition: Error.h:1198
OutputIt move(R &&Range, OutputIt Out)
Provide wrappers to std::move which take ranges instead of having to pass begin/end explicitly.
Definition: STLExtras.h:1849
Implement std::hash so that hash_code can be used in STL containers.
Definition: BitVector.h:858
Function object to check whether the second component of a container supported by std::get (like std:...
Definition: STLExtras.h:1459