1//===- AttrKindDetail.h - AttrKind conversion details -----------*- C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#ifndef ATTRKINDDETAIL_H_
10#define ATTRKINDDETAIL_H_
11
12#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
13#include "llvm/IR/Attributes.h"
14
15namespace mlir {
16namespace LLVM {
17namespace detail {
18
19/// Returns a list of pairs that each hold a mapping from LLVM attribute kinds
20/// to their corresponding string name in LLVM IR dialect.
21static llvm::ArrayRef<std::pair<llvm::Attribute::AttrKind, llvm::StringRef>>
22getAttrKindToNameMapping() {
23 using ElemTy = std::pair<llvm::Attribute::AttrKind, llvm::StringRef>;
24 // Mapping from llvm attribute kinds to their corresponding MLIR name.
25 static const llvm::SmallVector<ElemTy> kindNamePairs = {
26 {llvm::Attribute::AttrKind::Alignment, LLVMDialect::getAlignAttrName()},
27 {llvm::Attribute::AttrKind::AllocAlign,
28 LLVMDialect::getAllocAlignAttrName()},
29 {llvm::Attribute::AttrKind::AllocatedPointer,
30 LLVMDialect::getAllocatedPointerAttrName()},
31 {llvm::Attribute::AttrKind::ByVal, LLVMDialect::getByValAttrName()},
32 {llvm::Attribute::AttrKind::ByRef, LLVMDialect::getByRefAttrName()},
33 {llvm::Attribute::AttrKind::NoUndef, LLVMDialect::getNoUndefAttrName()},
34 {llvm::Attribute::AttrKind::Dereferenceable,
35 LLVMDialect::getDereferenceableAttrName()},
36 {llvm::Attribute::AttrKind::DereferenceableOrNull,
37 LLVMDialect::getDereferenceableOrNullAttrName()},
38 {llvm::Attribute::AttrKind::ElementType,
39 LLVMDialect::getElementTypeAttrName()},
40 {llvm::Attribute::AttrKind::InAlloca, LLVMDialect::getInAllocaAttrName()},
41 {llvm::Attribute::AttrKind::InReg, LLVMDialect::getInRegAttrName()},
42 {llvm::Attribute::AttrKind::Nest, LLVMDialect::getNestAttrName()},
43 {llvm::Attribute::AttrKind::NoAlias, LLVMDialect::getNoAliasAttrName()},
44 {llvm::Attribute::AttrKind::Captures,
45 LLVMDialect::getNoCaptureAttrName()},
46 {llvm::Attribute::AttrKind::NoFree, LLVMDialect::getNoFreeAttrName()},
47 {llvm::Attribute::AttrKind::NonNull, LLVMDialect::getNonNullAttrName()},
48 {llvm::Attribute::AttrKind::Preallocated,
49 LLVMDialect::getPreallocatedAttrName()},
50 {llvm::Attribute::AttrKind::Range, LLVMDialect::getRangeAttrName()},
51 {llvm::Attribute::AttrKind::ReadOnly, LLVMDialect::getReadonlyAttrName()},
52 {llvm::Attribute::AttrKind::ReadNone, LLVMDialect::getReadnoneAttrName()},
53 {llvm::Attribute::AttrKind::Returned, LLVMDialect::getReturnedAttrName()},
54 {llvm::Attribute::AttrKind::SExt, LLVMDialect::getSExtAttrName()},
55 {llvm::Attribute::AttrKind::StackAlignment,
56 LLVMDialect::getStackAlignmentAttrName()},
57 {llvm::Attribute::AttrKind::StructRet,
58 LLVMDialect::getStructRetAttrName()},
59 {llvm::Attribute::AttrKind::WriteOnly,
60 LLVMDialect::getWriteOnlyAttrName()},
61 {llvm::Attribute::AttrKind::ZExt, LLVMDialect::getZExtAttrName()}};
62 return kindNamePairs;
63}
64
65/// Returns a dense map from LLVM attribute name to their kind in LLVM IR
66/// dialect.
67[[maybe_unused]] static llvm::DenseMap<llvm::StringRef,
68 llvm::Attribute::AttrKind>
69getAttrNameToKindMapping() {
70 static auto attrNameToKindMapping = []() {
71 llvm::DenseMap<llvm::StringRef, llvm::Attribute::AttrKind> nameKindMap;
72 for (auto kindNamePair : getAttrKindToNameMapping()) {
73 nameKindMap.insert(KV: {kindNamePair.second, kindNamePair.first});
74 }
75 return nameKindMap;
76 }();
77 return attrNameToKindMapping;
78}
79
80} // namespace detail
81} // namespace LLVM
82} // namespace mlir
83
84#endif // ATTRKINDDETAIL_H_
85

source code of mlir/lib/Target/LLVMIR/AttrKindDetail.h