diff --git a/llvm/include/llvm/ADT/EquivalenceClasses.h b/llvm/include/llvm/ADT/EquivalenceClasses.h index 4f98b84cf97d2..c375d6e77b12a 100644 --- a/llvm/include/llvm/ADT/EquivalenceClasses.h +++ b/llvm/include/llvm/ADT/EquivalenceClasses.h @@ -144,7 +144,7 @@ class EquivalenceClasses { operator=(RHS); } - const EquivalenceClasses &operator=(const EquivalenceClasses &RHS) { + EquivalenceClasses &operator=(const EquivalenceClasses &RHS) { TheMapping.clear(); for (iterator I = RHS.begin(), E = RHS.end(); I != E; ++I) if (I->isLeader()) { diff --git a/llvm/unittests/ADT/EquivalenceClassesTest.cpp b/llvm/unittests/ADT/EquivalenceClassesTest.cpp index 70e161a03d988..bfb7c8d185fc8 100644 --- a/llvm/unittests/ADT/EquivalenceClassesTest.cpp +++ b/llvm/unittests/ADT/EquivalenceClassesTest.cpp @@ -13,6 +13,15 @@ using namespace llvm; namespace llvm { +TEST(EquivalenceClassesTest, CopyAssignemnt) { + EquivalenceClasses EC, Copy; + EC.insert(1); + EC.insert(4); + EquivalenceClasses &Ref = Copy = EC; + EXPECT_EQ(Copy.getNumClasses(), 2u); + EXPECT_EQ(&Ref, &Copy); +} + TEST(EquivalenceClassesTest, NoMerges) { EquivalenceClasses EqClasses; // Until we merged any sets, check that every element is only equivalent to