[RISCV][VP] support vp.reduce.mul by ExpandVectorPredication

Most of VP intrinsics are implemented in RISC-V backends, but vp.reduce.mul (element length > 1) does not yet. Legalizes vp.reduce.mul using ExpandVectorPredication Pass. Reviewed By: reames Differential Revision: https://reviews.llvm.org/D139721
llvm · Dec 17, 2022 · 0359c19 · 0359c19
1 parent 8086323
commit 0359c19
Show file tree

Hide file tree

Showing 2 changed files with 467 additions and 0 deletions.
diff --git a/llvm/lib/Target/RISCV/RISCVTargetTransformInfo.h b/llvm/lib/Target/RISCV/RISCVTargetTransformInfo.h
@@ -242,6 +242,11 @@ class RISCVTTIImpl : public BasicTTIImplBase<RISCVTTIImpl> {
   TargetTransformInfo::VPLegalization
   getVPLegalizationStrategy(const VPIntrinsic &PI) const {
     using VPLegalization = TargetTransformInfo::VPLegalization;
+    if (PI.getIntrinsicID() == Intrinsic::vp_reduce_mul &&
+        cast<VectorType>(PI.getArgOperand(1)->getType())
+                ->getElementType()
+                ->getIntegerBitWidth() != 1)
+      return VPLegalization(VPLegalization::Discard, VPLegalization::Convert);
     return VPLegalization(VPLegalization::Legal, VPLegalization::Legal);
   }