Transforms/Vectorize/LoopIdiomVectorize.h

0001 //===----------LoopIdiomVectorize.h -----------------------------*- C++ -*-===//
0002 //
0003 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
0004 // See https://llvm.org/LICENSE.txt for license information.
0005 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
0006 //
0007 //===----------------------------------------------------------------------===//
0008
0009 #ifndef LLVM_LIB_TRANSFORMS_VECTORIZE_LOOPIDIOMVECTORIZE_H
0010 #define LLVM_LIB_TRANSFORMS_VECTORIZE_LOOPIDIOMVECTORIZE_H
0011
0012 #include "llvm/IR/PassManager.h"
0013 #include "llvm/Transforms/Scalar/LoopPassManager.h"
0014
0015 namespace llvm {
0016 enum class LoopIdiomVectorizeStyle { Masked, Predicated };
0017
0018 class LoopIdiomVectorizePass : public PassInfoMixin<LoopIdiomVectorizePass> {
0019   LoopIdiomVectorizeStyle VectorizeStyle = LoopIdiomVectorizeStyle::Masked;
0020
0021   // The VF used in vectorizing the byte compare pattern.
0022   unsigned ByteCompareVF = 16;
0023
0024 public:
0025   LoopIdiomVectorizePass() = default;
0026   explicit LoopIdiomVectorizePass(LoopIdiomVectorizeStyle S)
0027       : VectorizeStyle(S) {}
0028
0029   LoopIdiomVectorizePass(LoopIdiomVectorizeStyle S, unsigned BCVF)
0030       : VectorizeStyle(S), ByteCompareVF(BCVF) {}
0031
0032   PreservedAnalyses run(Loop &L, LoopAnalysisManager &AM,
0033                         LoopStandardAnalysisResults &AR, LPMUpdater &U);
0034 };
0035 } // namespace llvm
0036 #endif // LLVM_LIB_TRANSFORMS_VECTORIZE_LOOPIDIOMVECTORIZE_H