[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