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 | |
15 | namespace mlir { |
16 | namespace LLVM { |
17 | namespace 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. |
21 | static llvm::ArrayRef<std::pair<llvm::Attribute::AttrKind, llvm::StringRef>> |
22 | getAttrKindToNameMapping() { |
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> |
69 | getAttrNameToKindMapping() { |
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 | |