[llvm] [LoopVectorizer] Allow partial reductions to be made in predicated loops (PR #124268)

Florian Hahn via llvm-commits llvm-commits at lists.llvm.org
Fri Jan 31 12:50:09 PST 2025


================
@@ -2,6 +2,7 @@
 ; RUN: opt -passes=loop-vectorize -enable-epilogue-vectorization=false -mattr=+neon,+dotprod -force-vector-interleave=1 -S < %s | FileCheck %s --check-prefixes=CHECK-INTERLEAVE1
 ; RUN: opt -passes=loop-vectorize -enable-epilogue-vectorization=false -mattr=+neon,+dotprod -S < %s | FileCheck %s --check-prefixes=CHECK-INTERLEAVED
 ; RUN: opt -passes=loop-vectorize -enable-epilogue-vectorization=false -mattr=+neon,+dotprod -force-vector-interleave=1 -vectorizer-maximize-bandwidth -S < %s | FileCheck %s --check-prefixes=CHECK-MAXBW
+; RUN: opt -passes=loop-vectorize -enable-epilogue-vectorization=true -mattr=+neon,+dotprod -S < %s | FileCheck %s --check-prefixes=CHECK-EPILOGUE
----------------
fhahn wrote:

There is already a separate test file to check epilogue codegen for partial reductions: `llvm/test/Transforms/LoopVectorize/AArch64/partial-reduce-dot-product-epilogue.ll`.

Can you add the predicated case there instead?

https://github.com/llvm/llvm-project/pull/124268


More information about the llvm-commits mailing list