1dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines//===---- llvm/MDBuilder.cpp - Builder for LLVM metadata ------------------===// 2dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines// 3dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines// The LLVM Compiler Infrastructure 4dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines// 5dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines// This file is distributed under the University of Illinois Open Source 6dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines// License. See LICENSE.TXT for details. 7dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines// 8dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines//===----------------------------------------------------------------------===// 9dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines// 10dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines// This file defines the MDBuilder class, which is used as a convenient way to 11dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines// create LLVM metadata with a consistent and simplified interface. 12dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines// 13dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines//===----------------------------------------------------------------------===// 14dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 15dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines#include "llvm/IR/MDBuilder.h" 16dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines#include "llvm/IR/Constants.h" 17dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines#include "llvm/IR/Metadata.h" 18dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hinesusing namespace llvm; 19dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 20dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen HinesMDString *MDBuilder::createString(StringRef Str) { 21dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return MDString::get(Context, Str); 22dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines} 23dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 24ebe69fe11e48d322045d5949c83283927a0d790bStephen HinesConstantAsMetadata *MDBuilder::createConstant(Constant *C) { 25ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines return ConstantAsMetadata::get(C); 26ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines} 27ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines 28dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen HinesMDNode *MDBuilder::createFPMath(float Accuracy) { 29dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (Accuracy == 0.0) 30dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 31dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines assert(Accuracy > 0.0 && "Invalid fpmath accuracy!"); 32ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines auto *Op = 33ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines createConstant(ConstantFP::get(Type::getFloatTy(Context), Accuracy)); 34dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return MDNode::get(Context, Op); 35dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines} 36dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 37dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen HinesMDNode *MDBuilder::createBranchWeights(uint32_t TrueWeight, 38dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines uint32_t FalseWeight) { 39dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines uint32_t Weights[] = {TrueWeight, FalseWeight}; 40dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return createBranchWeights(Weights); 41dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines} 42dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 43dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen HinesMDNode *MDBuilder::createBranchWeights(ArrayRef<uint32_t> Weights) { 44dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines assert(Weights.size() >= 2 && "Need at least two branch weights!"); 45dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 46ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines SmallVector<Metadata *, 4> Vals(Weights.size() + 1); 47dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Vals[0] = createString("branch_weights"); 48dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 49dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Type *Int32Ty = Type::getInt32Ty(Context); 50dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines for (unsigned i = 0, e = Weights.size(); i != e; ++i) 51ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines Vals[i + 1] = createConstant(ConstantInt::get(Int32Ty, Weights[i])); 52dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 53dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return MDNode::get(Context, Vals); 54dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines} 55dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 56dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen HinesMDNode *MDBuilder::createRange(const APInt &Lo, const APInt &Hi) { 57dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines assert(Lo.getBitWidth() == Hi.getBitWidth() && "Mismatched bitwidths!"); 58ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines 59ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines Type *Ty = IntegerType::get(Context, Lo.getBitWidth()); 60ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines return createRange(ConstantInt::get(Ty, Lo), ConstantInt::get(Ty, Hi)); 61ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines} 62ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines 63ebe69fe11e48d322045d5949c83283927a0d790bStephen HinesMDNode *MDBuilder::createRange(Constant *Lo, Constant *Hi) { 64dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // If the range is everything then it is useless. 65dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (Hi == Lo) 66dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 67dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 68dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // Return the range [Lo, Hi). 69ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines Metadata *Range[2] = {createConstant(Lo), createConstant(Hi)}; 70dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return MDNode::get(Context, Range); 71dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines} 72dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 7337ed9c199ca639565f6ce88105f9e39e898d82d0Stephen HinesMDNode *MDBuilder::createAnonymousAARoot(StringRef Name, MDNode *Extra) { 74dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // To ensure uniqueness the root node is self-referential. 75ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines auto Dummy = MDNode::getTemporary(Context, None); 7637ed9c199ca639565f6ce88105f9e39e898d82d0Stephen Hines 77ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines SmallVector<Metadata *, 3> Args(1, Dummy.get()); 7837ed9c199ca639565f6ce88105f9e39e898d82d0Stephen Hines if (Extra) 7937ed9c199ca639565f6ce88105f9e39e898d82d0Stephen Hines Args.push_back(Extra); 8037ed9c199ca639565f6ce88105f9e39e898d82d0Stephen Hines if (!Name.empty()) 8137ed9c199ca639565f6ce88105f9e39e898d82d0Stephen Hines Args.push_back(createString(Name)); 8237ed9c199ca639565f6ce88105f9e39e898d82d0Stephen Hines MDNode *Root = MDNode::get(Context, Args); 8337ed9c199ca639565f6ce88105f9e39e898d82d0Stephen Hines 84dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // At this point we have 85dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // !0 = metadata !{} <- dummy 86dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // !1 = metadata !{metadata !0} <- root 87dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // Replace the dummy operand with the root node itself and delete the dummy. 88dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Root->replaceOperandWith(0, Root); 89ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines 90dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // We now have 91dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // !1 = metadata !{metadata !1} <- self-referential root 92dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return Root; 93dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines} 94dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 95dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen HinesMDNode *MDBuilder::createTBAARoot(StringRef Name) { 96dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return MDNode::get(Context, createString(Name)); 97dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines} 98dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 99dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines/// \brief Return metadata for a non-root TBAA node with the given name, 100dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines/// parent in the TBAA tree, and value for 'pointsToConstantMemory'. 101dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen HinesMDNode *MDBuilder::createTBAANode(StringRef Name, MDNode *Parent, 102dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines bool isConstant) { 103dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (isConstant) { 104dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Constant *Flags = ConstantInt::get(Type::getInt64Ty(Context), 1); 105ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines Metadata *Ops[3] = {createString(Name), Parent, createConstant(Flags)}; 106dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return MDNode::get(Context, Ops); 107dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } else { 108ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines Metadata *Ops[2] = {createString(Name), Parent}; 109dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return MDNode::get(Context, Ops); 110dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 111dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines} 112dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 11337ed9c199ca639565f6ce88105f9e39e898d82d0Stephen HinesMDNode *MDBuilder::createAliasScopeDomain(StringRef Name) { 11437ed9c199ca639565f6ce88105f9e39e898d82d0Stephen Hines return MDNode::get(Context, createString(Name)); 11537ed9c199ca639565f6ce88105f9e39e898d82d0Stephen Hines} 11637ed9c199ca639565f6ce88105f9e39e898d82d0Stephen Hines 11737ed9c199ca639565f6ce88105f9e39e898d82d0Stephen HinesMDNode *MDBuilder::createAliasScope(StringRef Name, MDNode *Domain) { 118ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines Metadata *Ops[2] = {createString(Name), Domain}; 11937ed9c199ca639565f6ce88105f9e39e898d82d0Stephen Hines return MDNode::get(Context, Ops); 12037ed9c199ca639565f6ce88105f9e39e898d82d0Stephen Hines} 12137ed9c199ca639565f6ce88105f9e39e898d82d0Stephen Hines 122dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines/// \brief Return metadata for a tbaa.struct node with the given 123dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines/// struct field descriptions. 124dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen HinesMDNode *MDBuilder::createTBAAStructNode(ArrayRef<TBAAStructField> Fields) { 125ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines SmallVector<Metadata *, 4> Vals(Fields.size() * 3); 126dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Type *Int64 = Type::getInt64Ty(Context); 127dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines for (unsigned i = 0, e = Fields.size(); i != e; ++i) { 128ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines Vals[i * 3 + 0] = createConstant(ConstantInt::get(Int64, Fields[i].Offset)); 129ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines Vals[i * 3 + 1] = createConstant(ConstantInt::get(Int64, Fields[i].Size)); 130dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Vals[i * 3 + 2] = Fields[i].TBAA; 131dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 132dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return MDNode::get(Context, Vals); 133dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines} 134dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 135dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines/// \brief Return metadata for a TBAA struct node in the type DAG 136dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines/// with the given name, a list of pairs (offset, field type in the type DAG). 137dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen HinesMDNode *MDBuilder::createTBAAStructTypeNode( 138dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines StringRef Name, ArrayRef<std::pair<MDNode *, uint64_t>> Fields) { 139ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines SmallVector<Metadata *, 4> Ops(Fields.size() * 2 + 1); 140dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Type *Int64 = Type::getInt64Ty(Context); 141dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Ops[0] = createString(Name); 142dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines for (unsigned i = 0, e = Fields.size(); i != e; ++i) { 143dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Ops[i * 2 + 1] = Fields[i].first; 144ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines Ops[i * 2 + 2] = createConstant(ConstantInt::get(Int64, Fields[i].second)); 145dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 146dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return MDNode::get(Context, Ops); 147dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines} 148dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 149dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines/// \brief Return metadata for a TBAA scalar type node with the 150dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines/// given name, an offset and a parent in the TBAA type DAG. 151dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen HinesMDNode *MDBuilder::createTBAAScalarTypeNode(StringRef Name, MDNode *Parent, 152dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines uint64_t Offset) { 153dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines ConstantInt *Off = ConstantInt::get(Type::getInt64Ty(Context), Offset); 154ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines Metadata *Ops[3] = {createString(Name), Parent, createConstant(Off)}; 155dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return MDNode::get(Context, Ops); 156dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines} 157dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 158dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines/// \brief Return metadata for a TBAA tag node with the given 159dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines/// base type, access type and offset relative to the base type. 160dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen HinesMDNode *MDBuilder::createTBAAStructTagNode(MDNode *BaseType, MDNode *AccessType, 161dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines uint64_t Offset) { 162dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Type *Int64 = Type::getInt64Ty(Context); 163ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines Metadata *Ops[3] = {BaseType, AccessType, 164ebe69fe11e48d322045d5949c83283927a0d790bStephen Hines createConstant(ConstantInt::get(Int64, Offset))}; 165dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return MDNode::get(Context, Ops); 166dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines} 167