17b48de120
This patch makes the -force-attribute and -force-remove-attribute flags in the forceattrs pass additionally work over all the functions in a module rather than on specific functions. I often find myself dealing with bitcode from projects with arbitrary build systems that have applied the optnone attribute for various reasons and having the ability to do this in opt would be quite convenient. It's possible to remove the attribute by disassembling to textual IR, running sed, and then running opt over the result, but this option makes things just a bit easier. Differential Revision: https://reviews.llvm.org/D158815 Reviewed-By: nikic
92 lines
3.3 KiB
C++
92 lines
3.3 KiB
C++
//===- ForceFunctionAttrs.cpp - Force function attrs for debugging --------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Transforms/IPO/ForceFunctionAttrs.h"
|
|
#include "llvm/IR/Function.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/Debug.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
using namespace llvm;
|
|
|
|
#define DEBUG_TYPE "forceattrs"
|
|
|
|
static cl::list<std::string> ForceAttributes(
|
|
"force-attribute", cl::Hidden,
|
|
cl::desc(
|
|
"Add an attribute to a function. This can be a "
|
|
"pair of 'function-name:attribute-name', to apply an attribute to a "
|
|
"specific function. For "
|
|
"example -force-attribute=foo:noinline. Specifying only an attribute "
|
|
"will apply the attribute to every function in the module. This "
|
|
"option can be specified multiple times."));
|
|
|
|
static cl::list<std::string> ForceRemoveAttributes(
|
|
"force-remove-attribute", cl::Hidden,
|
|
cl::desc("Remove an attribute from a function. This can be a "
|
|
"pair of 'function-name:attribute-name' to remove an attribute "
|
|
"from a specific function. For "
|
|
"example -force-remove-attribute=foo:noinline. Specifying only an "
|
|
"attribute will remove the attribute from all functions in the "
|
|
"module. This "
|
|
"option can be specified multiple times."));
|
|
|
|
/// If F has any forced attributes given on the command line, add them.
|
|
/// If F has any forced remove attributes given on the command line, remove
|
|
/// them. When both force and force-remove are given to a function, the latter
|
|
/// takes precedence.
|
|
static void forceAttributes(Function &F) {
|
|
auto ParseFunctionAndAttr = [&](StringRef S) {
|
|
StringRef AttributeText;
|
|
if (S.contains(':')) {
|
|
auto KV = StringRef(S).split(':');
|
|
if (KV.first != F.getName())
|
|
return Attribute::None;
|
|
AttributeText = KV.second;
|
|
} else {
|
|
AttributeText = S;
|
|
}
|
|
auto Kind = Attribute::getAttrKindFromName(AttributeText);
|
|
if (Kind == Attribute::None || !Attribute::canUseAsFnAttr(Kind)) {
|
|
LLVM_DEBUG(dbgs() << "ForcedAttribute: " << AttributeText
|
|
<< " unknown or not a function attribute!\n");
|
|
}
|
|
return Kind;
|
|
};
|
|
|
|
for (const auto &S : ForceAttributes) {
|
|
auto Kind = ParseFunctionAndAttr(S);
|
|
if (Kind == Attribute::None || F.hasFnAttribute(Kind))
|
|
continue;
|
|
F.addFnAttr(Kind);
|
|
}
|
|
|
|
for (const auto &S : ForceRemoveAttributes) {
|
|
auto Kind = ParseFunctionAndAttr(S);
|
|
if (Kind == Attribute::None || !F.hasFnAttribute(Kind))
|
|
continue;
|
|
F.removeFnAttr(Kind);
|
|
}
|
|
}
|
|
|
|
static bool hasForceAttributes() {
|
|
return !ForceAttributes.empty() || !ForceRemoveAttributes.empty();
|
|
}
|
|
|
|
PreservedAnalyses ForceFunctionAttrsPass::run(Module &M,
|
|
ModuleAnalysisManager &) {
|
|
if (!hasForceAttributes())
|
|
return PreservedAnalyses::all();
|
|
|
|
for (Function &F : M.functions())
|
|
forceAttributes(F);
|
|
|
|
// Just conservatively invalidate analyses, this isn't likely to be important.
|
|
return PreservedAnalyses::none();
|
|
}
|