Support target extension types in vectors

Author: Aryan Sefidi

llvm/lib/IR/Type.cpp

@@ -791,7 +791,8 @@ VectorType *VectorType::get(Type *ElementType, ElementCount EC) {
 
 bool VectorType::isValidElementType(Type *ElemTy) {
   return ElemTy->isIntegerTy() || ElemTy->isFloatingPointTy() ||
-         ElemTy->isPointerTy() || ElemTy->getTypeID() == TypedPointerTyID;
+         ElemTy->isPointerTy() || ElemTy->getTypeID() == TypedPointerTyID || 
+         (ElemTy->isTargetExtTy() && ElemTy->isSized());
 }
 
 //===----------------------------------------------------------------------===//

llvm/test/Verifier/target-ext-vector-invalid.ll

@@ -0,0 +1,8 @@
+; RUN: not llvm-as %s -o /dev/null 2>&1 | FileCheck %s
+
+; CHECK: invalid vector element type
+
+define void @bad() {
+  %v = alloca <2 x target("spirv.IntegralConstant")>
+  ret void
+}

llvm/test/Verifier/target-ext-vector.ll

@@ -0,0 +1,11 @@
+; RUN: llvm-as -o - %s | llvm-dis | FileCheck %s
+
+; CHECK-LABEL: @vec_ops(
+define <2 x target("spirv.Image")> @vec_ops(<2 x target("spirv.Image")> %x) {
+  %a = alloca <2 x target("spirv.Image")>
+  store <2 x target("spirv.Image")> %x, ptr %a
+  %load = load <2 x target("spirv.Image")>, ptr %a
+  %elt = extractelement <2 x target("spirv.Image")> %load, i64 0
+  %res = insertelement <2 x target("spirv.Image")> undef, target("spirv.Image") %elt, i64 1
+  ret <2 x target("spirv.Image")> %res
+}