[llvm] [ExpandLargeDivRem] Scalarize vector types. (PR #86959)

Matt Arsenault via llvm-commits llvm-commits at lists.llvm.org
Thu Mar 28 11:01:41 PDT 2024


================
@@ -54,8 +54,34 @@ static bool isSigned(unsigned int Opcode) {
   return Opcode == Instruction::SDiv || Opcode == Instruction::SRem;
 }
 
+static void scalarize(BinaryOperator *BO,
+                      SmallVectorImpl<BinaryOperator *> &Replace) {
+  VectorType *VTy = cast<VectorType>(BO->getType());
+  assert(!VTy->isScalableTy() && "Tried to scalarize scalable vector!");
+
+  IRBuilder<> Builder(BO);
+
+  unsigned NumElements = VTy->getElementCount().getKnownMinValue();
+  Value *Result = nullptr;
+  for (unsigned Idx = 0; Idx < NumElements; ++Idx) {
+    Value *LHS = Builder.CreateExtractElement(BO->getOperand(0), Idx);
----------------
arsenm wrote:

We should probably create a utility for this somewhere. Scalarizer and AMDGPUCodeGenPrepare both do essentially the same thing already 

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


More information about the llvm-commits mailing list