[clang] [Clang] Add __builtin_vectorelements to get number of elements in vector (PR #69010)
Lawrence Benson via cfe-commits
cfe-commits at lists.llvm.org
Mon Oct 16 09:58:41 PDT 2023
================
@@ -3083,6 +3083,19 @@ ScalarExprEmitter::VisitUnaryExprOrTypeTraitExpr(
E->getTypeOfArgument()->getPointeeType()))
.getQuantity();
return llvm::ConstantInt::get(CGF.SizeTy, Alignment);
+ } else if (E->getKind() == UETT_VectorElements) {
+ // For scalable vectors, we don't know the size at compile time. We can use
+ // @llvm.vscale to calculate it at runtime.
+ if (E->getTypeOfArgument()->isSizelessVectorType()) {
+ auto *VecTy = dyn_cast<llvm::ScalableVectorType>(
+ ConvertType(E->getTypeOfArgument()));
+ uint64_t NumUnscaledElements = VecTy->getMinNumElements();
+
+ llvm::Value *VScale =
+ Builder.CreateVScale(llvm::ConstantInt::get(CGF.SizeTy, 1));
+ return Builder.CreateMul(
+ VScale, llvm::ConstantInt::get(CGF.SizeTy, NumUnscaledElements));
----------------
lawben wrote:
Neat, thanks for the pointer. that is a lot nicer. @erichkeane this should fix the comments above, as we now handle both types of vectors here and the code is more clear.
https://github.com/llvm/llvm-project/pull/69010
More information about the cfe-commits
mailing list