[llvm-commits] [llvm] r142794 - in /llvm/trunk: include/llvm/Support/BranchProbability.h unittests/Support/BlockFrequencyTest.cpp
Benjamin Kramer
benny.kra at googlemail.com
Mon Oct 24 06:50:56 PDT 2011
Author: d0k
Date: Mon Oct 24 08:50:56 2011
New Revision: 142794
URL: http://llvm.org/viewvc/llvm-project?rev=142794&view=rev
Log:
Implement comparison operators for BranchProbability in a way that can't overflow INT64_MAX.
Add a test case for the edge case that triggers this. Thanks to Chandler for bringing this to my attention.
Modified:
llvm/trunk/include/llvm/Support/BranchProbability.h
llvm/trunk/unittests/Support/BlockFrequencyTest.cpp
Modified: llvm/trunk/include/llvm/Support/BranchProbability.h
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/include/llvm/Support/BranchProbability.h?rev=142794&r1=142793&r2=142794&view=diff
==============================================================================
--- llvm/trunk/include/llvm/Support/BranchProbability.h (original)
+++ llvm/trunk/include/llvm/Support/BranchProbability.h Mon Oct 24 08:50:56 2011
@@ -29,10 +29,6 @@
// Denominator
uint32_t D;
- int64_t compare(BranchProbability RHS) const {
- return (uint64_t)N * RHS.D - (uint64_t)D * RHS.N;
- }
-
public:
BranchProbability(uint32_t n, uint32_t d) : N(n), D(d) {
assert(d > 0 && "Denomiator cannot be 0!");
@@ -54,12 +50,24 @@
void dump() const;
- bool operator==(BranchProbability RHS) const { return compare(RHS) == 0; }
- bool operator!=(BranchProbability RHS) const { return compare(RHS) != 0; }
- bool operator< (BranchProbability RHS) const { return compare(RHS) < 0; }
- bool operator> (BranchProbability RHS) const { return compare(RHS) > 0; }
- bool operator<=(BranchProbability RHS) const { return compare(RHS) <= 0; }
- bool operator>=(BranchProbability RHS) const { return compare(RHS) >= 0; }
+ bool operator==(BranchProbability RHS) const {
+ return (uint64_t)N * RHS.D == (uint64_t)D * RHS.N;
+ }
+ bool operator!=(BranchProbability RHS) const {
+ return !(*this == RHS);
+ }
+ bool operator<(BranchProbability RHS) const {
+ return (uint64_t)N * RHS.D < (uint64_t)D * RHS.N;
+ }
+ bool operator>(BranchProbability RHS) const {
+ return RHS < *this;
+ }
+ bool operator<=(BranchProbability RHS) const {
+ return (uint64_t)N * RHS.D <= (uint64_t)D * RHS.N;
+ }
+ bool operator>=(BranchProbability RHS) const {
+ return RHS <= *this;
+ }
};
raw_ostream &operator<<(raw_ostream &OS, const BranchProbability &Prob);
Modified: llvm/trunk/unittests/Support/BlockFrequencyTest.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/unittests/Support/BlockFrequencyTest.cpp?rev=142794&r1=142793&r2=142794&view=diff
==============================================================================
--- llvm/trunk/unittests/Support/BlockFrequencyTest.cpp (original)
+++ llvm/trunk/unittests/Support/BlockFrequencyTest.cpp Mon Oct 24 08:50:56 2011
@@ -71,6 +71,15 @@
EXPECT_TRUE(B > C);
EXPECT_FALSE(B <= C);
EXPECT_TRUE(B >= C);
+
+ BranchProbability BigZero(0, UINT32_MAX);
+ BranchProbability BigOne(UINT32_MAX, UINT32_MAX);
+ EXPECT_FALSE(BigZero == BigOne);
+ EXPECT_TRUE(BigZero != BigOne);
+ EXPECT_TRUE(BigZero < BigOne);
+ EXPECT_FALSE(BigZero > BigOne);
+ EXPECT_TRUE(BigZero <= BigOne);
+ EXPECT_FALSE(BigZero >= BigOne);
}
}
More information about the llvm-commits
mailing list