[SVE] Add specialized overloads of VectorType::get

christetreault-llvm · christetreault-llvm · commit d52ca3d7e607 · 2020-04-27T15:45:57.000-07:00
Summary: Add overloads of get to FixedVectorType and ScalableVectorType that mirror VectorType::get(Type *Ty, VectorType *VTy) that take and return derived vector types Reviewers: efriedma, c-rhodes, david-arm, fpetrogalli Reviewed By: david-arm Subscribers: tschuett, rkruppe, psnobl, llvm-commits Tags: #llvm Differential Revision: https://reviews.llvm.org/D78841
diff --git a/llvm/include/llvm/IR/DerivedTypes.h b/llvm/include/llvm/IR/DerivedTypes.h
@@ -549,6 +549,10 @@ class FixedVectorType : public VectorType {
 public:
   static FixedVectorType *get(Type *ElementType, unsigned NumElts);
 
+  static FixedVectorType *get(Type *ElementType, const FixedVectorType *FVTy) {
+    return get(ElementType, FVTy->getNumElements());
+  }
+
   static bool classof(const Type *T) {
     return T->getTypeID() == FixedVectorTyID;
   }
@@ -563,6 +567,11 @@ class ScalableVectorType : public VectorType {
 public:
   static ScalableVectorType *get(Type *ElementType, unsigned MinNumElts);
 
+  static ScalableVectorType *get(Type *ElementType,
+                                 const ScalableVectorType *SVTy) {
+    return get(ElementType, SVTy->getMinNumElements());
+  }
+
   /// Get the minimum number of elements in this vector. The actual number of
   /// elements in the vector is an integer multiple of this value.
   uint64_t getMinNumElements() const { return ElementQuantity; }
