Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===- llvm/CodeGen/GCStrategy.h - Garbage collection -----------*- C++ -*-===// |
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 | // |
||
9 | // GCStrategy coordinates code generation algorithms and implements some itself |
||
10 | // in order to generate code compatible with a target code generator as |
||
11 | // specified in a function's 'gc' attribute. Algorithms are enabled by setting |
||
12 | // flags in a subclass's constructor, and some virtual methods can be |
||
13 | // overridden. |
||
14 | // |
||
15 | // GCStrategy is relevant for implementations using either gc.root or |
||
16 | // gc.statepoint based lowering strategies, but is currently focused mostly on |
||
17 | // options for gc.root. This will change over time. |
||
18 | // |
||
19 | // When requested by a subclass of GCStrategy, the gc.root implementation will |
||
20 | // populate GCModuleInfo and GCFunctionInfo with that about each Function in |
||
21 | // the Module that opts in to garbage collection. Specifically: |
||
22 | // |
||
23 | // - Safe points |
||
24 | // Garbage collection is generally only possible at certain points in code. |
||
25 | // GCStrategy can request that the collector insert such points: |
||
26 | // |
||
27 | // - At and after any call to a subroutine |
||
28 | // - Before returning from the current function |
||
29 | // - Before backwards branches (loops) |
||
30 | // |
||
31 | // - Roots |
||
32 | // When a reference to a GC-allocated object exists on the stack, it must be |
||
33 | // stored in an alloca registered with llvm.gcoot. |
||
34 | // |
||
35 | // This information can used to emit the metadata tables which are required by |
||
36 | // the target garbage collector runtime. |
||
37 | // |
||
38 | // When used with gc.statepoint, information about safepoint and roots can be |
||
39 | // found in the binary StackMap section after code generation. Safepoint |
||
40 | // placement is currently the responsibility of the frontend, though late |
||
41 | // insertion support is planned. |
||
42 | // |
||
43 | // The read and write barrier support can be used with either implementation. |
||
44 | // |
||
45 | //===----------------------------------------------------------------------===// |
||
46 | |||
47 | #ifndef LLVM_IR_GCSTRATEGY_H |
||
48 | #define LLVM_IR_GCSTRATEGY_H |
||
49 | |||
50 | #include "llvm/Support/Registry.h" |
||
51 | #include <optional> |
||
52 | #include <string> |
||
53 | |||
54 | namespace llvm { |
||
55 | |||
56 | class Type; |
||
57 | |||
58 | /// GCStrategy describes a garbage collector algorithm's code generation |
||
59 | /// requirements, and provides overridable hooks for those needs which cannot |
||
60 | /// be abstractly described. GCStrategy objects must be looked up through |
||
61 | /// the Function. The objects themselves are owned by the Context and must |
||
62 | /// be immutable. |
||
63 | class GCStrategy { |
||
64 | private: |
||
65 | friend class GCModuleInfo; |
||
66 | |||
67 | std::string Name; |
||
68 | |||
69 | protected: |
||
70 | bool UseStatepoints = false; /// Uses gc.statepoints as opposed to gc.roots, |
||
71 | /// if set, NeededSafePoints and UsesMetadata |
||
72 | /// should be left at their default values. |
||
73 | |||
74 | bool UseRS4GC = false; /// If UseStatepoints is set, this determines whether |
||
75 | /// the RewriteStatepointsForGC pass should rewrite |
||
76 | /// this function's calls. |
||
77 | /// This should only be set if UseStatepoints is set. |
||
78 | |||
79 | bool NeededSafePoints = false; ///< if set, calls are inferred to be safepoints |
||
80 | bool UsesMetadata = false; ///< If set, backend must emit metadata tables. |
||
81 | |||
82 | public: |
||
83 | GCStrategy(); |
||
84 | virtual ~GCStrategy() = default; |
||
85 | |||
86 | /// Return the name of the GC strategy. This is the value of the collector |
||
87 | /// name string specified on functions which use this strategy. |
||
88 | const std::string &getName() const { return Name; } |
||
89 | |||
90 | /// Returns true if this strategy is expecting the use of gc.statepoints, |
||
91 | /// and false otherwise. |
||
92 | bool useStatepoints() const { return UseStatepoints; } |
||
93 | |||
94 | /** @name Statepoint Specific Properties */ |
||
95 | ///@{ |
||
96 | |||
97 | /// If the type specified can be reliably distinguished, returns true for |
||
98 | /// pointers to GC managed locations and false for pointers to non-GC |
||
99 | /// managed locations. Note a GCStrategy can always return 'std::nullopt' |
||
100 | /// (i.e. an empty optional indicating it can't reliably distinguish. |
||
101 | virtual std::optional<bool> isGCManagedPointer(const Type *Ty) const { |
||
102 | return std::nullopt; |
||
103 | } |
||
104 | |||
105 | /// Returns true if the RewriteStatepointsForGC pass should run on functions |
||
106 | /// using this GC. |
||
107 | bool useRS4GC() const { |
||
108 | assert(useStatepoints() && |
||
109 | "GC strategy has useRS4GC but not useStatepoints set"); |
||
110 | return UseRS4GC; |
||
111 | } |
||
112 | |||
113 | ///@} |
||
114 | |||
115 | /// If set, appropriate metadata tables must be emitted by the back-end |
||
116 | /// (assembler, JIT, or otherwise). The default stackmap information can be |
||
117 | /// found in the StackMap section as described in the documentation. |
||
118 | bool usesMetadata() const { return UsesMetadata; } |
||
119 | |||
120 | /** @name GCRoot Specific Properties |
||
121 | * These properties and overrides only apply to collector strategies using |
||
122 | * GCRoot. |
||
123 | */ |
||
124 | ///@{ |
||
125 | |||
126 | /// True if safe points need to be inferred on call sites |
||
127 | bool needsSafePoints() const { return NeededSafePoints; } |
||
128 | |||
129 | ///@} |
||
130 | }; |
||
131 | |||
132 | /// Subclasses of GCStrategy are made available for use during compilation by |
||
133 | /// adding them to the global GCRegistry. This can done either within the |
||
134 | /// LLVM source tree or via a loadable plugin. An example registeration |
||
135 | /// would be: |
||
136 | /// static GCRegistry::Add<CustomGC> X("custom-name", |
||
137 | /// "my custom supper fancy gc strategy"); |
||
138 | /// |
||
139 | /// Note that to use a custom GCMetadataPrinter, you must also |
||
140 | /// register your GCMetadataPrinter subclass with the |
||
141 | /// GCMetadataPrinterRegistery as well. |
||
142 | using GCRegistry = Registry<GCStrategy>; |
||
143 | |||
144 | /// Lookup the GCStrategy object associated with the given gc name. |
||
145 | std::unique_ptr<GCStrategy> getGCStrategy(const StringRef Name); |
||
146 | |||
147 | } // end namespace llvm |
||
148 | |||
149 | #endif // LLVM_IR_GCSTRATEGY_H |