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
|
//
// Copyright 2016 Francisco Jerez
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the "Software"),
// to deal in the Software without restriction, including without limitation
// the rights to use, copy, modify, merge, publish, distribute, sublicense,
// and/or sell copies of the Software, and to permit persons to whom the
// Software is furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
// THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
// OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
// OTHER DEALINGS IN THE SOFTWARE.
//
///
/// \file
/// Utility functions for LLVM IR metadata introspection.
///
#ifndef CLOVER_LLVM_METADATA_HPP
#define CLOVER_LLVM_METADATA_HPP
#include "llvm/compat.hpp"
#include "util/algorithm.hpp"
#include <vector>
#include <llvm/Config/llvm-config.h>
#include <llvm/IR/Module.h>
#include <llvm/IR/Metadata.h>
namespace clover {
namespace llvm {
namespace detail {
inline std::vector<const ::llvm::MDNode *>
get_kernel_nodes(const ::llvm::Module &mod) {
if (const ::llvm::NamedMDNode *n =
mod.getNamedMetadata("opencl.kernels"))
return { n->op_begin(), n->op_end() };
else
return {};
}
inline std::function<bool (const ::llvm::MDNode *n)>
is_kernel_node_for(const ::llvm::Function &f) {
return [&](const ::llvm::MDNode *n) {
using ::llvm::mdconst::dyn_extract;
return &f == dyn_extract< ::llvm::Function>(n->getOperand(0));
};
}
inline bool
is_kernel(const ::llvm::Function &f) {
return f.getMetadata("kernel_arg_type");
}
inline iterator_range< ::llvm::MDNode::op_iterator>
get_kernel_metadata_operands(const ::llvm::Function &f,
const std::string &name) {
const auto data_node = f.getMetadata(name);
return range(data_node->op_begin(), data_node->op_end());
}
}
///
/// Extract the string metadata node \p name corresponding to the kernel
/// argument given by \p arg.
///
inline std::string
get_argument_metadata(const ::llvm::Function &f,
const ::llvm::Argument &arg,
const std::string &name) {
return ::llvm::cast< ::llvm::MDString>(
detail::get_kernel_metadata_operands(f, name)[arg.getArgNo()])
->getString();
}
///
/// Return a vector with all CL kernel functions found in the LLVM
/// module \p mod.
///
inline std::vector<const ::llvm::Function *>
get_kernels(const ::llvm::Module &mod) {
std::vector<const ::llvm::Function *> fs;
for (auto &f : mod.getFunctionList()) {
if (detail::is_kernel(f))
fs.push_back(&f);
}
return fs;
}
}
}
#endif
|
