[LV][EVL] Support fpext/fptrunc/fptosi/fptoui/sitofp/uitofp/inttoptr/ptrtoint of cast instruction with EVL-vectorization

Author: LiqinWeng

llvm/lib/Transforms/Vectorize/VPlan.h

@@ -1594,14 +1594,14 @@ class VPWidenCastEVLRecipe : public VPWidenCastRecipe {
 
 public:
   VPWidenCastEVLRecipe(Instruction::CastOps Opcode, VPValue *Op, Type *ResultTy,
-                       VPValue &EVL)
-      : VPWidenCastRecipe(VPDef::VPWidenCastEVLSC, Opcode, Op, ResultTy) {
+                       CastInst &UI, VPValue &EVL)
+      : VPWidenCastRecipe(VPDef::VPWidenCastEVLSC, Opcode, Op, ResultTy, UI) {
     addOperand(&EVL);
   }
 
   VPWidenCastEVLRecipe(VPWidenCastRecipe &W, VPValue &EVL)
       : VPWidenCastEVLRecipe(W.getOpcode(), W.getOperand(0), W.getResultType(),
-                             EVL) {
+                             *cast<CastInst>(W.getUnderlyingInstr()), EVL) {
     transferFlags(W);
   }
 

llvm/lib/Transforms/Vectorize/VPlanRecipes.cpp

@@ -1345,13 +1345,12 @@ void VPWidenCastRecipe::execute(VPTransformState &State) {
 
 void VPWidenCastEVLRecipe::execute(VPTransformState &State) {
   unsigned Opcode = getOpcode();
+  auto Inst = cast<CastInst>(getUnderlyingInstr());
   State.setDebugLocFrom(getDebugLoc());
   assert(State.UF == 1 && "Expected only UF == 1 when vectorizing with "
                           "explicit vector length.");
 
-  // TODO: add more cast instruction, eg: fptoint/inttofp/inttoptr/fptofp
-  if (Opcode == Instruction::SExt || Opcode == Instruction::ZExt ||
-      Opcode == Instruction::Trunc) {
+  if (Inst->isCast()) {
     Value *SrcVal = State.get(getOperand(0), 0);
     VectorType *DsType = VectorType::get(getResultType(), State.VF);
 

llvm/lib/Transforms/Vectorize/VPlanTransforms.cpp

@@ -1381,10 +1381,8 @@ static void transformRecipestoEVLRecipes(VPlan &Plan, VPValue &EVL) {
               })
               .Case<VPWidenCastRecipe>(
                   [&](VPWidenCastRecipe *W) -> VPRecipeBase * {
-                    unsigned Opcode = W->getOpcode();
-                    if (Opcode != Instruction::SExt &&
-                        Opcode != Instruction::ZExt &&
-                        Opcode != Instruction::Trunc)
+                    auto Inst = cast<CastInst>(W->getUnderlyingInstr());
+                    if (!Inst->isCast())
                       return nullptr;
                     return new VPWidenCastEVLRecipe(*W, EVL);
                   })