b8a825b483
This patch introduces a new abstract attributor instance that propagates assumption information from functions. Conceptually, if a function is only called by functions that have certain assumptions, then we can apply the same assumptions to that function. This problem is similar to calculating the dominator set, but the assumptions are merged instead of nodes. Reviewed By: jdoerfert Differential Revision: https://reviews.llvm.org/D111054
111 lines
3.4 KiB
C++
111 lines
3.4 KiB
C++
//===- Assumptions.cpp ------ Collection of helpers for assumptions -------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements helper functions for accessing assumption infomration
|
|
// inside of the "llvm.assume" metadata.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/IR/Assumptions.h"
|
|
#include "llvm/ADT/SetOperations.h"
|
|
#include "llvm/ADT/StringExtras.h"
|
|
#include "llvm/IR/Attributes.h"
|
|
#include "llvm/IR/Function.h"
|
|
#include "llvm/IR/InstrTypes.h"
|
|
|
|
using namespace llvm;
|
|
|
|
namespace {
|
|
bool hasAssumption(const Attribute &A,
|
|
const KnownAssumptionString &AssumptionStr) {
|
|
if (!A.isValid())
|
|
return false;
|
|
assert(A.isStringAttribute() && "Expected a string attribute!");
|
|
|
|
SmallVector<StringRef, 8> Strings;
|
|
A.getValueAsString().split(Strings, ",");
|
|
|
|
return llvm::is_contained(Strings, AssumptionStr);
|
|
}
|
|
|
|
DenseSet<StringRef> getAssumptions(const Attribute &A) {
|
|
if (!A.isValid())
|
|
return DenseSet<StringRef>();
|
|
assert(A.isStringAttribute() && "Expected a string attribute!");
|
|
|
|
DenseSet<StringRef> Assumptions;
|
|
SmallVector<StringRef, 8> Strings;
|
|
A.getValueAsString().split(Strings, ",");
|
|
|
|
for (StringRef Str : Strings)
|
|
Assumptions.insert(Str);
|
|
return Assumptions;
|
|
}
|
|
|
|
template <typename AttrSite>
|
|
bool addAssumptionsImpl(AttrSite &Site,
|
|
const DenseSet<StringRef> &Assumptions) {
|
|
if (Assumptions.empty())
|
|
return false;
|
|
|
|
DenseSet<StringRef> CurAssumptions = getAssumptions(Site);
|
|
|
|
if (!set_union(CurAssumptions, Assumptions))
|
|
return false;
|
|
|
|
LLVMContext &Ctx = Site.getContext();
|
|
Site.addFnAttr(llvm::Attribute::get(
|
|
Ctx, llvm::AssumptionAttrKey,
|
|
llvm::join(CurAssumptions.begin(), CurAssumptions.end(), ",")));
|
|
|
|
return true;
|
|
}
|
|
} // namespace
|
|
|
|
bool llvm::hasAssumption(const Function &F,
|
|
const KnownAssumptionString &AssumptionStr) {
|
|
const Attribute &A = F.getFnAttribute(AssumptionAttrKey);
|
|
return ::hasAssumption(A, AssumptionStr);
|
|
}
|
|
|
|
bool llvm::hasAssumption(const CallBase &CB,
|
|
const KnownAssumptionString &AssumptionStr) {
|
|
if (Function *F = CB.getCalledFunction())
|
|
if (hasAssumption(*F, AssumptionStr))
|
|
return true;
|
|
|
|
const Attribute &A = CB.getFnAttr(AssumptionAttrKey);
|
|
return ::hasAssumption(A, AssumptionStr);
|
|
}
|
|
|
|
DenseSet<StringRef> llvm::getAssumptions(const Function &F) {
|
|
const Attribute &A = F.getFnAttribute(AssumptionAttrKey);
|
|
return ::getAssumptions(A);
|
|
}
|
|
|
|
DenseSet<StringRef> llvm::getAssumptions(const CallBase &CB) {
|
|
const Attribute &A = CB.getFnAttr(AssumptionAttrKey);
|
|
return ::getAssumptions(A);
|
|
}
|
|
|
|
bool llvm::addAssumptions(Function &F, const DenseSet<StringRef> &Assumptions) {
|
|
return ::addAssumptionsImpl(F, Assumptions);
|
|
}
|
|
|
|
bool llvm::addAssumptions(CallBase &CB,
|
|
const DenseSet<StringRef> &Assumptions) {
|
|
return ::addAssumptionsImpl(CB, Assumptions);
|
|
}
|
|
|
|
StringSet<> llvm::KnownAssumptionStrings({
|
|
"omp_no_openmp", // OpenMP 5.1
|
|
"omp_no_openmp_routines", // OpenMP 5.1
|
|
"omp_no_parallelism", // OpenMP 5.1
|
|
"ompx_spmd_amenable", // OpenMPOpt extension
|
|
});
|