#ifndef LLVM_CLANG_BASIC_TARGETINFO_H
#define LLVM_CLANG_BASIC_TARGETINFO_H
+#include "clang/AST/Attr.h"
#include "clang/Basic/AddressSpaces.h"
+#include "clang/Basic/Attributes.h"
#include "clang/Basic/LLVM.h"
#include "clang/Basic/Specifiers.h"
#include "clang/Basic/TargetCXXABI.h"
/// language options which change the target configuration.
virtual void adjust(const LangOptions &Opts);
+ /// \brief Parse a __target__ attribute and get the cpu/feature strings
+ /// out of it for later use.
+ typedef std::pair<StringRef, std::vector<std::string>> ParsedTargetAttr;
+ ParsedTargetAttr parseTargetAttr(const TargetAttr *TA) const;
+
/// \brief Initialize the map with the default set of target features for the
/// CPU this should include all legal feature strings on the target.
///
/// \return False on error (invalid features).
virtual bool initFeatureMap(llvm::StringMap<bool> &Features,
DiagnosticsEngine &Diags, StringRef CPU,
- std::vector<std::string> &FeatureVec) const {
- for (const auto &F : FeatureVec) {
- const char *Name = F.c_str();
- // Apply the feature via the target.
- bool Enabled = Name[0] == '+';
- setFeatureEnabled(Features, Name + 1, Enabled);
- }
- return true;
- }
+ std::vector<std::string> &FeatureVec) const;
/// \brief Get the ABI currently in use.
virtual StringRef getABI() const { return StringRef(); }
return true;
}
+
+bool TargetInfo::initFeatureMap(llvm::StringMap<bool> &Features,
+ DiagnosticsEngine &Diags, StringRef CPU,
+ std::vector<std::string> &FeatureVec) const {
+ for (const auto &F : FeatureVec) {
+ const char *Name = F.c_str();
+ // Apply the feature via the target.
+ bool Enabled = Name[0] == '+';
+ setFeatureEnabled(Features, Name + 1, Enabled);
+ }
+ return true;
+}
+
+TargetInfo::ParsedTargetAttr
+TargetInfo::parseTargetAttr(const TargetAttr *TA) const {
+ std::pair<StringRef, std::vector<std::string>> RetPair;
+
+ // Grab the target attribute string.
+ StringRef FeaturesStr = TA->getFeatures();
+ SmallVector<StringRef, 1> AttrFeatures;
+ FeaturesStr.split(AttrFeatures, ",");
+
+ // Grab the various features and prepend a "+" to turn on the feature to
+ // the backend and add them to our existing set of features.
+ for (auto &Feature : AttrFeatures) {
+ // Go ahead and trim whitespace rather than either erroring or
+ // accepting it weirdly.
+ Feature = Feature.trim();
+
+ // While we're here iterating check for a different target cpu.
+ if (Feature.startswith("arch="))
+ RetPair.first = Feature.split("=").second.trim();
+ else if (Feature.startswith("tune="))
+ // We don't support cpu tuning this way currently.
+ ;
+ else if (Feature.startswith("fpmath="))
+ // TODO: Support the fpmath option this way. It will require checking
+ // overall feature validity for the function with the rest of the
+ // attributes on the function.
+ ;
+ else if (Feature.startswith("no-"))
+ RetPair.second.push_back("-" + Feature.split("-").second.str());
+ else
+ RetPair.second.push_back("+" + Feature.str());
+ }
+ return RetPair;
+}
const FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(TargetDecl);
if (FD && FD->getAttr<TargetAttr>()) {
llvm::StringMap<bool> FeatureMap;
- const auto *TD = FD->getAttr<TargetAttr>();
-
- // Make a copy of the features as passed on the command line.
- std::vector<std::string> FnFeatures =
- getTarget().getTargetOpts().FeaturesAsWritten;
-
- // Grab the target attribute string.
- StringRef FeaturesStr = TD->getFeatures();
- SmallVector<StringRef, 1> AttrFeatures;
- FeaturesStr.split(AttrFeatures, ",");
-
- // Grab the various features and prepend a "+" to turn on the feature to
- // the backend and add them to our existing set of features.
- for (auto &Feature : AttrFeatures) {
- // Go ahead and trim whitespace rather than either erroring or
- // accepting it weirdly.
- Feature = Feature.trim();
-
- // While we're here iterating check for a different target cpu.
- if (Feature.startswith("arch="))
- TargetCPU = Feature.split("=").second.trim();
- else if (Feature.startswith("tune="))
- // We don't support cpu tuning this way currently.
- ;
- else if (Feature.startswith("fpmath="))
- // TODO: Support the fpmath option this way. It will require checking
- // overall feature validity for the function with the rest of the
- // attributes on the function.
- ;
- else if (Feature.startswith("no-"))
- FnFeatures.push_back("-" + Feature.split("-").second.str());
- else
- FnFeatures.push_back("+" + Feature.str());
- }
+
// Now populate the feature map, first with the TargetCPU which is either
// the default or a new one from the target attribute string. Then we'll
// use the passed in features (FeaturesAsWritten) along with the new ones
// from the attribute.
- getTarget().initFeatureMap(FeatureMap, Diags, TargetCPU, FnFeatures);
+ TargetInfo::ParsedTargetAttr PTA =
+ getTarget().parseTargetAttr(FD->getAttr<TargetAttr>());
+ if (PTA.first != "")
+ TargetCPU = PTA.first;
+ PTA.second.insert(PTA.second.begin(),
+ getTarget().getTargetOpts().FeaturesAsWritten.begin(),
+ getTarget().getTargetOpts().FeaturesAsWritten.end());
+ getTarget().initFeatureMap(FeatureMap, Diags, TargetCPU, PTA.second);
// Produce the canonical string for this set of features.
std::vector<std::string> Features;