[llvm] 3358838 - [ADT] Add APIntOps::abds signed absolute difference and rename absdiff -> abdu (#84791)
via llvm-commits
llvm-commits at lists.llvm.org
Tue Mar 12 03:42:02 PDT 2024
Author: Simon Pilgrim
Date: 2024-03-12T10:41:59Z
New Revision: 3358838446428976a41390fde98fe5b04b08a132
URL: https://github.com/llvm/llvm-project/commit/3358838446428976a41390fde98fe5b04b08a132
DIFF: https://github.com/llvm/llvm-project/commit/3358838446428976a41390fde98fe5b04b08a132.diff
LOG: [ADT] Add APIntOps::abds signed absolute difference and rename absdiff -> abdu (#84791)
When I created APIntOps::absdiff, I totally missed that we already have ISD::ABDS/ABDU nodes, and we use this term in other places/targets as well.
I've added the APIntOps::abds implementation and renamed APIntOps::absdiff to APIntOps::abdu.
Given that APIntOps::absdiff is so young I don't think we need to create a deprecation wrapper, but I can if anyone thinks it important.
I'll do a KnownBits rename patch after this.
Added:
Modified:
llvm/include/llvm/ADT/APInt.h
llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp
llvm/unittests/ADT/APIntTest.cpp
llvm/unittests/Support/KnownBitsTest.cpp
Removed:
################################################################################
diff --git a/llvm/include/llvm/ADT/APInt.h b/llvm/include/llvm/ADT/APInt.h
index 1fc3c7b2236a17..bea3e28adf308f 100644
--- a/llvm/include/llvm/ADT/APInt.h
+++ b/llvm/include/llvm/ADT/APInt.h
@@ -2188,8 +2188,13 @@ inline const APInt &umax(const APInt &A, const APInt &B) {
return A.ugt(B) ? A : B;
}
+/// Determine the absolute
diff erence of two APInts considered to be signed.
+inline const APInt abds(const APInt &A, const APInt &B) {
+ return A.sge(B) ? (A - B) : (B - A);
+}
+
/// Determine the absolute
diff erence of two APInts considered to be unsigned.
-inline const APInt abs
diff (const APInt &A, const APInt &B) {
+inline const APInt abdu(const APInt &A, const APInt &B) {
return A.uge(B) ? (A - B) : (B - A);
}
diff --git a/llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp b/llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp
index 7a0c1c328df1fb..c24303592769a0 100644
--- a/llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp
+++ b/llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp
@@ -6068,9 +6068,9 @@ static std::optional<APInt> FoldValue(unsigned Opcode, const APInt &C1,
return (C1Ext + C2Ext + 1).extractBits(C1.getBitWidth(), 1);
}
case ISD::ABDS:
- return APIntOps::smax(C1, C2) - APIntOps::smin(C1, C2);
+ return APIntOps::abds(C1, C2);
case ISD::ABDU:
- return APIntOps::umax(C1, C2) - APIntOps::umin(C1, C2);
+ return APIntOps::abdu(C1, C2);
}
return std::nullopt;
}
diff --git a/llvm/unittests/ADT/APIntTest.cpp b/llvm/unittests/ADT/APIntTest.cpp
index 24324822356bf6..11237d2e1602dc 100644
--- a/llvm/unittests/ADT/APIntTest.cpp
+++ b/llvm/unittests/ADT/APIntTest.cpp
@@ -2532,38 +2532,72 @@ TEST(APIntTest, clearLowBits) {
EXPECT_EQ(16u, i32hi16.popcount());
}
-TEST(APIntTest, AbsDiff) {
- using APIntOps::abs
diff ;
+TEST(APIntTest, abds) {
+ using APIntOps::abds;
APInt MaxU1(1, 1, false);
APInt MinU1(1, 0, false);
- EXPECT_EQ(1u, abs
diff (MaxU1, MinU1).getZExtValue());
- EXPECT_EQ(1u, abs
diff (MinU1, MaxU1).getZExtValue());
+ EXPECT_EQ(1u, abds(MaxU1, MinU1).getZExtValue());
+ EXPECT_EQ(1u, abds(MinU1, MaxU1).getZExtValue());
APInt MaxU4(4, 15, false);
APInt MinU4(4, 0, false);
- EXPECT_EQ(15u, abs
diff (MaxU4, MinU4).getZExtValue());
- EXPECT_EQ(15u, abs
diff (MinU4, MaxU4).getZExtValue());
+ EXPECT_EQ(1, abds(MaxU4, MinU4).getSExtValue());
+ EXPECT_EQ(1, abds(MinU4, MaxU4).getSExtValue());
APInt MaxS8(8, 127, true);
APInt MinS8(8, -128, true);
- EXPECT_EQ(1u, abs
diff (MaxS8, MinS8).getZExtValue());
- EXPECT_EQ(1u, abs
diff (MinS8, MaxS8).getZExtValue());
+ EXPECT_EQ(-1, abds(MaxS8, MinS8).getSExtValue());
+ EXPECT_EQ(-1, abds(MinS8, MaxS8).getSExtValue());
APInt MaxU16(16, 65535, false);
APInt MinU16(16, 0, false);
- EXPECT_EQ(65535u, abs
diff (MaxU16, MinU16).getZExtValue());
- EXPECT_EQ(65535u, abs
diff (MinU16, MaxU16).getZExtValue());
+ EXPECT_EQ(1, abds(MaxU16, MinU16).getSExtValue());
+ EXPECT_EQ(1, abds(MinU16, MaxU16).getSExtValue());
APInt MaxS16(16, 32767, true);
APInt MinS16(16, -32768, true);
APInt ZeroS16(16, 0, true);
- EXPECT_EQ(1u, abs
diff (MaxS16, MinS16).getZExtValue());
- EXPECT_EQ(1u, abs
diff (MinS16, MaxS16).getZExtValue());
- EXPECT_EQ(32768u, abs
diff (ZeroS16, MinS16));
- EXPECT_EQ(32768u, abs
diff (MinS16, ZeroS16));
- EXPECT_EQ(32767u, abs
diff (ZeroS16, MaxS16));
- EXPECT_EQ(32767u, abs
diff (MaxS16, ZeroS16));
+ EXPECT_EQ(-1, abds(MaxS16, MinS16).getSExtValue());
+ EXPECT_EQ(-1, abds(MinS16, MaxS16).getSExtValue());
+ EXPECT_EQ(32768u, abds(ZeroS16, MinS16));
+ EXPECT_EQ(32768u, abds(MinS16, ZeroS16));
+ EXPECT_EQ(32767u, abds(ZeroS16, MaxS16));
+ EXPECT_EQ(32767u, abds(MaxS16, ZeroS16));
+}
+
+TEST(APIntTest, abdu) {
+ using APIntOps::abdu;
+
+ APInt MaxU1(1, 1, false);
+ APInt MinU1(1, 0, false);
+ EXPECT_EQ(1u, abdu(MaxU1, MinU1).getZExtValue());
+ EXPECT_EQ(1u, abdu(MinU1, MaxU1).getZExtValue());
+
+ APInt MaxU4(4, 15, false);
+ APInt MinU4(4, 0, false);
+ EXPECT_EQ(15u, abdu(MaxU4, MinU4).getZExtValue());
+ EXPECT_EQ(15u, abdu(MinU4, MaxU4).getZExtValue());
+
+ APInt MaxS8(8, 127, true);
+ APInt MinS8(8, -128, true);
+ EXPECT_EQ(1u, abdu(MaxS8, MinS8).getZExtValue());
+ EXPECT_EQ(1u, abdu(MinS8, MaxS8).getZExtValue());
+
+ APInt MaxU16(16, 65535, false);
+ APInt MinU16(16, 0, false);
+ EXPECT_EQ(65535u, abdu(MaxU16, MinU16).getZExtValue());
+ EXPECT_EQ(65535u, abdu(MinU16, MaxU16).getZExtValue());
+
+ APInt MaxS16(16, 32767, true);
+ APInt MinS16(16, -32768, true);
+ APInt ZeroS16(16, 0, true);
+ EXPECT_EQ(1u, abdu(MaxS16, MinS16).getZExtValue());
+ EXPECT_EQ(1u, abdu(MinS16, MaxS16).getZExtValue());
+ EXPECT_EQ(32768u, abdu(ZeroS16, MinS16));
+ EXPECT_EQ(32768u, abdu(MinS16, ZeroS16));
+ EXPECT_EQ(32767u, abdu(ZeroS16, MaxS16));
+ EXPECT_EQ(32767u, abdu(MaxS16, ZeroS16));
}
TEST(APIntTest, GCD) {
diff --git a/llvm/unittests/Support/KnownBitsTest.cpp b/llvm/unittests/Support/KnownBitsTest.cpp
index 7c183e9626f985..2ac25f0b2801ba 100644
--- a/llvm/unittests/Support/KnownBitsTest.cpp
+++ b/llvm/unittests/Support/KnownBitsTest.cpp
@@ -361,9 +361,7 @@ TEST(KnownBitsTest, BinaryExhaustive) {
[](const KnownBits &Known1, const KnownBits &Known2) {
return KnownBits::abs
diff (Known1, Known2);
},
- [](const APInt &N1, const APInt &N2) {
- return APIntOps::abs
diff (N1, N2);
- },
+ [](const APInt &N1, const APInt &N2) { return APIntOps::abdu(N1, N2); },
checkCorrectnessOnlyBinary);
testBinaryOpExhaustive(
[](const KnownBits &Known1, const KnownBits &Known2) {
More information about the llvm-commits
mailing list