[llvm] [VPlan] Unroll VPReplicateRecipe by VF. (PR #142433)

Florian Hahn via llvm-commits llvm-commits at lists.llvm.org
Sat Jun 21 14:27:14 PDT 2025


================
@@ -608,6 +608,35 @@ Value *VPInstruction::generate(VPTransformState &State) {
     return Builder.CreateVectorSplat(
         State.VF, State.get(getOperand(0), /*IsScalar*/ true), "broadcast");
   }
+  case VPInstruction::BuildStructVector: {
+    // For struct types, we need to build a new 'wide' struct type, where each
+    // element is widened.
+    auto *StructTy =
+        cast<StructType>(State.TypeAnalysis.inferScalarType(getOperand(0)));
+    auto NumOfElements = ElementCount::getFixed(getNumOperands());
+    Value *Res = PoisonValue::get(toVectorizedTy(StructTy, NumOfElements));
+    assert(NumOfElements.getKnownMinValue() == StructTy->getNumElements() &&
+           "number of operands must match number of elements in StructTy");
+    for (const auto &[Idx, Op] : enumerate(operands())) {
+      for (unsigned I = 0; I != NumOfElements.getKnownMinValue(); I++) {
----------------
fhahn wrote:

yep the inner loop needs to process StructTy->getNumElements, which may be different than the number of operands.

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


More information about the llvm-commits mailing list