[Codegen] Add vector transfer + slice foldings in GenericVectorization

Signed-off-by: Max Dawkins <[email protected]>

compiler/src/iree/compiler/Codegen/Common/GenericVectorization.cpp

@@ -364,6 +364,16 @@ void GenericVectorizationPass::runOnOperation() {
                             vectorizeGatherAccesses);
   };
 
+  {
+    RewritePatternSet transferSliceFoldingPatterns(funcOp.getContext());
+    tensor::populateFoldTensorSubsetIntoVectorTransferPatterns(
+        transferSliceFoldingPatterns);
+    if (failed(applyPatternsAndFoldGreedily(funcOp,
+            std::move(transferSliceFoldingPatterns)))) {
+      return signalPassFailure();
+    }
+  }
+
   {
     // Canonicalize mask related ops before we lower them.
     RewritePatternSet maskCanonPatterns(funcOp.getContext());