DynamicAPInt: Support APInt constructor. (#146301)

This PR introduces a constructor for `DynamicAPInt` that takes an
`APInt` object as input to simplifies the creation of a large numbers.
This commit is contained in:
Zakk Chen
2025-07-01 15:44:12 +08:00
committed by GitHub
parent 65cb0eae58
commit 3cc200f46b
2 changed files with 19 additions and 0 deletions

View File

@@ -16,6 +16,7 @@
#ifndef LLVM_ADT_DYNAMICAPINT_H
#define LLVM_ADT_DYNAMICAPINT_H
#include "llvm/ADT/APInt.h"
#include "llvm/ADT/SlowDynamicAPInt.h"
#include "llvm/Support/Compiler.h"
#include "llvm/Support/MathExtras.h"
@@ -116,6 +117,14 @@ public:
: ValSmall(Val) {
ValLarge.Val.BitWidth = 0;
}
LLVM_ATTRIBUTE_ALWAYS_INLINE explicit DynamicAPInt(const APInt &Val) {
if (Val.getBitWidth() <= 64) {
ValSmall = Val.getSExtValue();
ValLarge.Val.BitWidth = 0;
} else {
new (&ValLarge) detail::SlowDynamicAPInt(Val);
}
}
LLVM_ATTRIBUTE_ALWAYS_INLINE DynamicAPInt() : DynamicAPInt(0) {}
LLVM_ATTRIBUTE_ALWAYS_INLINE ~DynamicAPInt() {
if (LLVM_UNLIKELY(isLarge()))

View File

@@ -31,6 +31,16 @@ public:
};
TYPED_TEST_SUITE(IntTest, TypeList, TypeNames);
TYPED_TEST(IntTest, ValueInit) {
APInt Large(65, 0, true);
Large.setBit(64);
TypeParam DynLarge(1ll << 63);
EXPECT_EQ(TypeParam(Large), DynLarge + DynLarge);
APInt Small(64, -1, true);
TypeParam DynSmall(Small.getSExtValue());
EXPECT_EQ(TypeParam(Small), DynSmall);
}
TYPED_TEST(IntTest, ops) {
TypeParam Two(2), Five(5), Seven(7), Ten(10);
EXPECT_EQ(Five + Five, Ten);