fixup! fix customIf

llvm · Sep 30, 2024 · 3509d6b · 3509d6b
1 parent fd57581
commit 3509d6b
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/llvm/lib/Target/RISCV/GISel/RISCVLegalizerInfo.cpp b/llvm/lib/Target/RISCV/GISel/RISCVLegalizerInfo.cpp
@@ -616,9 +616,11 @@ RISCVLegalizerInfo::RISCVLegalizerInfo(const RISCVSubtarget &ST)
                 Query.Types[0].getElementCount().divideCoefficientBy(8), 8);
             return std::pair(0, CastTy);
           })
-      .customIf(
-          LegalityPredicates::any(typeIsLegalBoolVec(0, BoolVecTys, ST),
-                                  typeIsLegalIntOrFPVec(0, IntOrFPVecTys, ST)));
+      .customIf(LegalityPredicates::any(
+          all(typeIsLegalBoolVec(0, BoolVecTys, ST),
+              typeIsLegalBoolVec(1, BoolVecTys, ST)),
+          all(typeIsLegalIntOrFPVec(0, IntOrFPVecTys, ST),
+              typeIsLegalIntOrFPVec(1, IntOrFPVecTys, ST))));
 
   getLegacyLegalizerInfo().computeTables();
 }