10 #ifndef LLVM_CLANG_DRIVER_COMPILATION_H 11 #define LLVM_CLANG_DRIVER_COMPILATION_H 16 #include "llvm/ADT/DenseMap.h" 43 unsigned ActiveOffloadMask;
49 std::multimap<Action::OffloadKind, const ToolChain *>
50 OrderedOffloadingToolchains;
53 llvm::opt::InputArgList *Args;
57 llvm::opt::DerivedArgList *TranslatedArgs;
61 std::vector<std::unique_ptr<Action>> AllActions;
72 struct TCArgsKey final {
76 bool operator<(
const TCArgsKey &K)
const {
79 else if (TC == K.TC && BoundArch < K.BoundArch)
81 else if (TC == K.TC && BoundArch == K.BoundArch &&
82 DeviceOffloadKind < K.DeviceOffloadKind)
86 TCArgsKey(
const ToolChain *TC, StringRef BoundArch,
88 : TC(TC), BoundArch(BoundArch), DeviceOffloadKind(DeviceOffloadKind) {}
90 std::map<TCArgsKey, llvm::opt::DerivedArgList *> TCArgs;
93 llvm::opt::ArgStringList TempFiles;
103 std::vector<Optional<StringRef>> Redirects;
113 llvm::opt::InputArgList *Args,
114 llvm::opt::DerivedArgList *TranslatedArgs,
bool ContainsError);
122 return ActiveOffloadMask &
Kind;
130 const_offload_toolchains_iterator>
133 template <Action::OffloadKind Kind>
135 return OrderedOffloadingToolchains.equal_range(
Kind);
140 return OrderedOffloadingToolchains.find(
Kind) !=
141 OrderedOffloadingToolchains.end();
146 template <Action::OffloadKind Kind>
148 auto TCs = getOffloadToolChains<Kind>();
150 assert(TCs.first != TCs.second &&
151 "No tool chains of the selected kind exist!");
152 assert(std::next(TCs.first) == TCs.second &&
153 "More than one tool chain of the this kind exist.");
154 return TCs.first->second;
158 Action::OffloadKind OffloadKind) {
159 assert(OffloadKind != Action::OFK_Host && OffloadKind != Action::OFK_None &&
160 "This is not a device tool chain!");
163 ActiveOffloadMask |= OffloadKind;
164 OrderedOffloadingToolchains.insert(
165 std::make_pair(OffloadKind, DeviceToolChain));
170 const llvm::opt::DerivedArgList &
getArgs()
const {
return *TranslatedArgs; }
172 llvm::opt::DerivedArgList &
getArgs() {
return *TranslatedArgs; }
180 template <
typename T,
typename... Args>
T *
MakeAction(Args &&... Arg) {
181 T *RawPtr =
new T(std::forward<Args>(Arg)...);
182 AllActions.push_back(std::unique_ptr<Action>(RawPtr));
191 const llvm::opt::ArgStringList &
getTempFiles()
const {
return TempFiles; }
196 return FailureResultFiles;
200 StringRef getSysRoot()
const;
210 const llvm::opt::DerivedArgList &
211 getArgsForToolChain(
const ToolChain *TC, StringRef BoundArch,
212 Action::OffloadKind DeviceOffloadKind);
217 TempFiles.push_back(Name);
224 ResultFiles[JA] = Name;
231 FailureResultFiles[JA] = Name;
239 bool CleanupFile(
const char *File,
bool IssueErrors =
false)
const;
245 bool CleanupFileList(
const llvm::opt::ArgStringList &Files,
246 bool IssueErrors =
false)
const;
256 bool IssueErrors =
false)
const;
263 int ExecuteCommand(
const Command &C,
const Command *&FailingCommand)
const;
271 SmallVectorImpl<std::pair<int, const Command *>> &FailingCommands)
const;
276 void initCompilationForDiagnostics();
const JobList & getJobs() const
const llvm::opt::ArgStringList & getTempFiles() const
DominatorTree GraphTraits specialization so the DominatorTree can be iterable by generic graph iterat...
T * MakeAction(Args &&... Arg)
Creates a new Action owned by this Compilation.
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)
const std::multimap< Action::OffloadKind, const ToolChain * >::const_iterator const_offload_toolchains_iterator
Iterator that visits device toolchains of a given kind.
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.
const FunctionProtoType * T
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.
std::pair< const_offload_toolchains_iterator, const_offload_toolchains_iterator > const_offload_toolchains_range
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.