llvm/lib/Transforms/Vectorize/VPlanAnalysis.h

   1 //===- VPlanAnalysis.h - Various Analyses working on VPlan ------*- C++ -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #ifndef LLVM_TRANSFORMS_VECTORIZE_VPLANANALYSIS_H
  10 #define LLVM_TRANSFORMS_VECTORIZE_VPLANANALYSIS_H
  11
  12 #include "llvm/ADT/DenseMap.h"
  13
  14 namespace llvm {
  15
  16 class LLVMContext;
  17 class VPValue;
  18 class VPBlendRecipe;
  19 class VPInterleaveRecipe;
  20 class VPInstruction;
  21 class VPReductionPHIRecipe;
  22 class VPWidenRecipe;
  23 class VPWidenCallRecipe;
  24 class VPWidenCastRecipe;
  25 class VPWidenIntOrFpInductionRecipe;
  26 class VPWidenMemoryInstructionRecipe;
  27 struct VPWidenSelectRecipe;
  28 class VPReplicateRecipe;
  29 class Type;
  30
  31 /// An analysis for type-inference for VPValues.
  32 /// It infers the scalar type for a given VPValue by bottom-up traversing
  33 /// through defining recipes until root nodes with known types are reached (e.g.
  34 /// live-ins or load recipes). The types are then propagated top down through
  35 /// operations.
  36 /// Note that the analysis caches the inferred types. A new analysis object must
  37 /// be constructed once a VPlan has been modified in a way that invalidates any
  38 /// of the previously inferred types.
  39 class VPTypeAnalysis {
  40   DenseMap<const VPValue *, Type *> CachedTypes;
  41   LLVMContext &Ctx;
  42
  43   Type *inferScalarTypeForRecipe(const VPBlendRecipe *R);
  44   Type *inferScalarTypeForRecipe(const VPInstruction *R);
  45   Type *inferScalarTypeForRecipe(const VPWidenCallRecipe *R);
  46   Type *inferScalarTypeForRecipe(const VPWidenRecipe *R);
  47   Type *inferScalarTypeForRecipe(const VPWidenIntOrFpInductionRecipe *R);
  48   Type *inferScalarTypeForRecipe(const VPWidenMemoryInstructionRecipe *R);
  49   Type *inferScalarTypeForRecipe(const VPWidenSelectRecipe *R);
  50   Type *inferScalarTypeForRecipe(const VPReplicateRecipe *R);
  51
  52 public:
  53   VPTypeAnalysis(LLVMContext &Ctx) : Ctx(Ctx) {}
  54
  55   /// Infer the type of \p V. Returns the scalar type of \p V.
  56   Type *inferScalarType(const VPValue *V);
  57 };
  58
  59 } // end namespace llvm
  60
  61 #endif // LLVM_TRANSFORMS_VECTORIZE_VPLANANALYSIS_H