9 #ifndef LLVM_CLANG_DRIVER_COMPILATION_H 10 #define LLVM_CLANG_DRIVER_COMPILATION_H 16 #include "llvm/ADT/ArrayRef.h" 17 #include "llvm/ADT/DenseMap.h" 18 #include "llvm/ADT/Optional.h" 19 #include "llvm/ADT/StringRef.h" 20 #include "llvm/Option/Option.h" 54 unsigned ActiveOffloadMask = 0;
60 std::multimap<Action::OffloadKind, const ToolChain *>
61 OrderedOffloadingToolchains;
64 llvm::opt::InputArgList *Args;
68 llvm::opt::DerivedArgList *TranslatedArgs;
72 std::vector<std::unique_ptr<Action>> AllActions;
83 struct TCArgsKey final {
88 TCArgsKey(
const ToolChain *TC, StringRef BoundArch,
90 : TC(TC), BoundArch(BoundArch), DeviceOffloadKind(DeviceOffloadKind) {}
92 bool operator<(
const TCArgsKey &K)
const {
95 else if (TC == K.TC && BoundArch < K.BoundArch)
97 else if (TC == K.TC && BoundArch == K.BoundArch &&
98 DeviceOffloadKind < K.DeviceOffloadKind)
103 std::map<TCArgsKey, llvm::opt::DerivedArgList *> TCArgs;
106 llvm::opt::ArgStringList TempFiles;
116 std::vector<Optional<StringRef>> Redirects;
119 bool ForDiagnostics =
false;
125 bool ForceKeepTempFiles =
false;
129 llvm::opt::InputArgList *Args,
130 llvm::opt::DerivedArgList *TranslatedArgs,
bool ContainsError);
138 return ActiveOffloadMask &
Kind;
147 const_offload_toolchains_iterator>;
149 template <Action::OffloadKind Kind>
151 return OrderedOffloadingToolchains.equal_range(
Kind);
156 return OrderedOffloadingToolchains.find(
Kind) !=
157 OrderedOffloadingToolchains.end();
162 template <Action::OffloadKind Kind>
164 auto TCs = getOffloadToolChains<Kind>();
166 assert(TCs.first != TCs.second &&
167 "No tool chains of the selected kind exist!");
168 assert(std::next(TCs.first) == TCs.second &&
169 "More than one tool chain of the this kind exist.");
170 return TCs.first->second;
174 Action::OffloadKind OffloadKind) {
175 assert(OffloadKind != Action::OFK_Host && OffloadKind != Action::OFK_None &&
176 "This is not a device tool chain!");
179 ActiveOffloadMask |= OffloadKind;
180 OrderedOffloadingToolchains.insert(
181 std::make_pair(OffloadKind, DeviceToolChain));
186 const llvm::opt::DerivedArgList &
getArgs()
const {
return *TranslatedArgs; }
188 llvm::opt::DerivedArgList &
getArgs() {
return *TranslatedArgs; }
196 template <
typename T,
typename... Args> T *
MakeAction(Args &&... Arg) {
197 T *RawPtr =
new T(std::forward<Args>(Arg)...);
198 AllActions.push_back(std::unique_ptr<Action>(RawPtr));
207 const llvm::opt::ArgStringList &
getTempFiles()
const {
return TempFiles; }
212 return FailureResultFiles;
216 StringRef getSysRoot()
const;
226 const llvm::opt::DerivedArgList &
227 getArgsForToolChain(
const ToolChain *TC, StringRef BoundArch,
228 Action::OffloadKind DeviceOffloadKind);
233 TempFiles.push_back(Name);
240 ResultFiles[JA] = Name;
247 FailureResultFiles[JA] = Name;
255 bool CleanupFile(
const char *File,
bool IssueErrors =
false)
const;
261 bool CleanupFileList(
const llvm::opt::ArgStringList &Files,
262 bool IssueErrors =
false)
const;
272 bool IssueErrors =
false)
const;
279 int ExecuteCommand(
const Command &C,
const Command *&FailingCommand)
const;
287 SmallVectorImpl<std::pair<int, const Command *>> &FailingCommands)
const;
292 void initCompilationForDiagnostics();
311 #endif // LLVM_CLANG_DRIVER_COMPILATION_H const JobList & getJobs() const
const llvm::opt::ArgStringList & getTempFiles() const
Specialize PointerLikeTypeTraits to allow LazyGenerationalUpdatePtr to be placed into a PointerUnion...
T * MakeAction(Args &&... Arg)
Creates a new Action owned by this Compilation.
const std::multimap< Action::OffloadKind, const ToolChain * >::const_iterator const_offload_toolchains_iterator
Iterator that visits device toolchains of a given kind.
std::pair< const_offload_toolchains_iterator, const_offload_toolchains_iterator > const_offload_toolchains_range
const ArgStringMap & getFailureResultFiles() const
bool containsError() const
Return whether an error during the parsing of the input args.
void addOffloadDeviceToolChain(const ToolChain *DeviceToolChain, Action::OffloadKind OffloadKind)
Forward-declares and imports various common LLVM datatypes that clang wants to use unqualified...
const ActionList & getActions() const
Driver - Encapsulate logic for constructing compilation processes from a set of gcc-driver-like comma...
const llvm::opt::InputArgList & getInputArgs() const
bool hasOffloadToolChain() const
Return true if an offloading tool chain of a given kind exists.
const ToolChain & getDefaultToolChain() const
llvm::opt::DerivedArgList & getArgs()
const ArgStringMap & getResultFiles() const
bool isForDiagnostics() const
Return true if we're compiling for diagnostics.
unsigned isOffloadingHostKind(Action::OffloadKind Kind) const
void addCommand(std::unique_ptr< Command > C)
JobList - A sequence of jobs to perform.
void addJob(std::unique_ptr< Command > J)
Add a job to the list (taking ownership).
llvm::DenseMap< const JobAction *, const char * > ArgStringMap
ArgStringMap - Type used to map a JobAction to its result file.
const_offload_toolchains_range getOffloadToolChains() const
const llvm::opt::DerivedArgList & getArgs() const
Command - An executable path/name and argument vector to execute.
const char * addResultFile(const char *Name, const JobAction *JA)
addResultFile - Add a file to remove on failure, and returns its argument.
bool operator<(DeclarationName LHS, DeclarationName RHS)
Ordering on two declaration names.
Dataflow Directional Tag Classes.
ActionList & getActions()
Compilation - A set of tasks to perform for a single driver invocation.
const Driver & getDriver() const
const char * addFailureResultFile(const char *Name, const JobAction *JA)
addFailureResultFile - Add a file to remove if we crash, and returns its argument.
const char * addTempFile(const char *Name)
addTempFile - Add a file to remove on exit, and returns its argument.
const ToolChain * getSingleOffloadToolChain() const
Return an offload toolchain of the provided kind.