From 0a1d315f9f166207a95ce5d1e124e1a290807f98 Mon Sep 17 00:00:00 2001
From: Anton Afanasyev <anton.a.afanasyev@gmail.com>
Date: Tue, 17 Nov 2020 18:46:31 +0300
Subject: [PATCH] [SLPVectorizer] Fix assert

---
 llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp b/llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp
index 63efbbd672a59..4e4b6a8dd1ce6 100644
--- a/llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp
+++ b/llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp
@@ -1765,7 +1765,7 @@ class BoUpSLP {
                           ArrayRef<unsigned> ReorderIndices = None) {
     assert(!(Bundle && EntryState == TreeEntry::NeedToGather) &&
            "Need to gather vectorized entry?");
-    assert(!Bundle && EntryState != TreeEntry::NeedToGather &&
+    assert((Bundle || EntryState == TreeEntry::NeedToGather) &&
            "Need to vectorize gather entry?");
     VectorizableTree.push_back(std::make_unique<TreeEntry>(VectorizableTree));
     TreeEntry *Last = VectorizableTree.back().get();