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 const StringRef **Redirects;
113 llvm::opt::InputArgList *Args,
114 llvm::opt::DerivedArgList *TranslatedArgs,
bool ContainsError);
122 return ActiveOffloadMask &
Kind;
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;
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;
210 const llvm::opt::DerivedArgList &
217 TempFiles.push_back(Name);
224 ResultFiles[JA] =
Name;
231 FailureResultFiles[JA] =
Name;
239 bool CleanupFile(
const char *File,
bool IssueErrors =
false)
const;
246 bool IssueErrors =
false)
const;
256 bool IssueErrors =
false)
const;
271 SmallVectorImpl<std::pair<int, const Command *>> &FailingCommands)
const;
291 void Redirect(
const StringRef** Redirects);
const Driver & getDriver() const
const ArgStringMap & getResultFiles() const
bool hasOffloadToolChain() const
Return true if an offloading tool chain of a given kind exists.
T * MakeAction(Args &&...Arg)
Creates a new Action owned by this Compilation.
const llvm::opt::DerivedArgList & getArgsForToolChain(const ToolChain *TC, StringRef BoundArch, Action::OffloadKind DeviceOffloadKind)
getArgsForToolChain - Return the derived argument list for the tool chain TC (or the default tool cha...
bool CleanupFile(const char *File, bool IssueErrors=false) const
CleanupFile - Delete a given file.
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 llvm::opt::DerivedArgList & getArgs() const
Driver - Encapsulate logic for constructing compilation processes from a set of gcc-driver-like comma...
unsigned isOffloadingHostKind(Action::OffloadKind Kind) const
llvm::opt::DerivedArgList & getArgs()
bool containsError() const
Return whether an error during the parsing of the input args.
bool isForDiagnostics() const
Return true if we're compiling for diagnostics.
const ToolChain * getSingleOffloadToolChain() const
Return an offload toolchain of the provided kind.
bool CleanupFileList(const llvm::opt::ArgStringList &Files, bool IssueErrors=false) const
CleanupFileList - Remove the files in the given list.
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.
bool CleanupFileMap(const ArgStringMap &Files, const JobAction *JA, bool IssueErrors=false) const
CleanupFileMap - Remove the files in the given map.
void Redirect(const StringRef **Redirects)
Redirect - Redirect output of this compilation.
const_offload_toolchains_range getOffloadToolChains() 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.
Compilation(const Driver &D, const ToolChain &DefaultToolChain, llvm::opt::InputArgList *Args, llvm::opt::DerivedArgList *TranslatedArgs, bool ContainsError)
bool operator<(DeclarationName LHS, DeclarationName RHS)
Ordering on two declaration names.
const ToolChain & getDefaultToolChain() const
const ArgStringMap & getFailureResultFiles() const
const ActionList & getActions() const
ActionList & getActions()
void ExecuteJobs(const JobList &Jobs, SmallVectorImpl< std::pair< int, const Command * >> &FailingCommands) const
ExecuteJob - Execute a single job.
Compilation - A set of tasks to perform for a single driver invocation.
int ExecuteCommand(const Command &C, const Command *&FailingCommand) const
ExecuteCommand - Execute an actual command.
const llvm::opt::InputArgList & getInputArgs() const
StringRef getSysRoot() const
Returns the sysroot path.
const char * addFailureResultFile(const char *Name, const JobAction *JA)
addFailureResultFile - Add a file to remove if we crash, and returns its argument.
const llvm::opt::ArgStringList & getTempFiles() const
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 JobList & getJobs() const
void initCompilationForDiagnostics()
initCompilationForDiagnostics - Remove stale state and suppress output so compilation can be reexecut...