Fix support of SPV_INTEL_vector_compute (#1176)

The previous patch has fixed only case with 1-element vectors, however
after closer look at the spec it became clear that
SPV_INTEL_vector_compute actually allows any number of vector elements
(capability VectorAnyINTEL), so this is the proper fix.

Author: Fznamznon

lib/SPIRV/SPIRVLowerBitCastToNonStandardType.cpp

@@ -150,6 +150,13 @@ class SPIRVLowerBitCastToNonStandardTypePass
     // parameter, since it added by an optimization.
     bool Changed = false;
 
+    // SPV_INTEL_vector_compute allows to use vectors with any number of
+    // components. Since this method only lowers vectors with non-standard
+    // in pure SPIR-V number of components, there is no need to do anything in
+    // case SPV_INTEL_vector_compute is enabled.
+    if (Opts.isAllowedToUseExtension(ExtensionID::SPV_INTEL_vector_compute))
+      return PreservedAnalyses::all();
+
     std::vector<Instruction *> BCastsToNonStdVec;
     std::vector<Instruction *> InstsToErase;
     for (auto &BB : F)
@@ -160,23 +167,15 @@ class SPIRVLowerBitCastToNonStandardTypePass
         VectorType *SrcVecTy = getVectorType(BC->getSrcTy());
         if (SrcVecTy) {
           uint64_t NumElemsInSrcVec = SrcVecTy->getElementCount().getValue();
-          // SPV_INTEL_vector_compute allows 1-element vectors
-          if (!isValidVectorSize(NumElemsInSrcVec) &&
-              !(Opts.isAllowedToUseExtension(
-                    ExtensionID::SPV_INTEL_vector_compute) &&
-                NumElemsInSrcVec == 1))
+          if (!isValidVectorSize(NumElemsInSrcVec))
             report_fatal_error("Unsupported vector type with the size of: " +
                                    std::to_string(NumElemsInSrcVec),
                                false);
         }
         VectorType *DestVecTy = getVectorType(BC->getDestTy());
         if (DestVecTy) {
           uint64_t NumElemsInDestVec = DestVecTy->getElementCount().getValue();
-          // SPV_INTEL_vector_compute allows 1-element vectors
-          if (!isValidVectorSize(NumElemsInDestVec) &&
-              !(Opts.isAllowedToUseExtension(
-                    ExtensionID::SPV_INTEL_vector_compute) &&
-                NumElemsInDestVec == 1))
+          if (!isValidVectorSize(NumElemsInDestVec))
             BCastsToNonStdVec.push_back(&I);
         }
       }

test/lower-non-standard-vec-with-ext.ll

@@ -15,6 +15,7 @@ entry:
   %1 = extractelement <4 x i8> %0, i32 0
   %2 = bitcast <1 x i32> <i32 131586> to <4 x i8>
   %3 = extractelement <4 x i8> %2, i32 0
+  %4 = bitcast <5 x i32> <i32 1, i32 1, i32 1, i32 1, i32 1> to <20 x i8>
   ret void
 }