1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
//===- 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
});
|
