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
|
/*========================== begin_copyright_notice ============================
Copyright (C) 2022 Intel Corporation
SPDX-License-Identifier: MIT
============================= end_copyright_notice ===========================*/
#ifndef IGCLLVM_ADT_APINT_H
#define IGCLLVM_ADT_APINT_H
#include "Probe/Assertion.h"
#include <llvm/ADT/APInt.h>
#if LLVM_VERSION_MAJOR >= 14
#include "llvm/Support/DivisionByConstantInfo.h"
#endif
namespace IGCLLVM {
#if LLVM_VERSION_MAJOR >= 14
using SignedDivisionByConstantInfo = llvm::SignedDivisionByConstantInfo;
using UnsignedDivisionByConstantInfo =
#if (LLVM_VERSION_MAJOR == 14) || defined(IGC_LLVM_TRUNK_REVISION)
// Account for a typo
llvm::UnsignedDivisonByConstantInfo;
#else // LLVM_VERSION_MAJOR == 14
llvm::UnsignedDivisionByConstantInfo;
#endif
#else
using SignedDivisionByConstantInfo = llvm::APInt::ms;
using UnsignedDivisionByConstantInfo = llvm::APInt::mu;
#endif
inline SignedDivisionByConstantInfo getAPIntMagic(const llvm::APInt &value) {
#if LLVM_VERSION_MAJOR >= 14
return llvm::SignedDivisionByConstantInfo::get(value);
#else
return value.magic();
#endif
}
inline UnsignedDivisionByConstantInfo getAPIntMagicUnsigned(const llvm::APInt &value, const unsigned LeadingZeros = 0) {
#if LLVM_VERSION_MAJOR >= 16
// Basing on this: [https://reviews.llvm.org/D140924]
return UnsignedDivisionByConstantInfo::get(value, LeadingZeros, false);
#elif LLVM_VERSION_MAJOR >= 14
return UnsignedDivisionByConstantInfo::get(value, LeadingZeros);
#else
return value.magicu(LeadingZeros);
#endif
}
inline bool IsAddition(const UnsignedDivisionByConstantInfo &mu) {
#if LLVM_VERSION_MAJOR >= 14
return mu.IsAdd;
#else
return mu.a;
#endif
}
inline unsigned ShiftAmount(const UnsignedDivisionByConstantInfo &mu) {
#if LLVM_VERSION_MAJOR >= 16
// Basing on this: https://reviews.llvm.org/D141014
return IsAddition(mu) ? mu.PostShift + 1 : mu.PostShift;
#elif LLVM_VERSION_MAJOR >= 14
return mu.ShiftAmount;
#else
return mu.s;
#endif
}
inline unsigned ShiftAmount(const SignedDivisionByConstantInfo &ms) {
#if LLVM_VERSION_MAJOR >= 14
return ms.ShiftAmount;
#else
return ms.s;
#endif
}
inline llvm::APInt MagicNumber(const UnsignedDivisionByConstantInfo &mu) {
#if LLVM_VERSION_MAJOR >= 14
return mu.Magic;
#else
return mu.m;
#endif
}
inline llvm::APInt MagicNumber(const SignedDivisionByConstantInfo &ms) {
#if LLVM_VERSION_MAJOR >= 14
return ms.Magic;
#else
return ms.m;
#endif
}
} // namespace IGCLLVM
#endif
|
