[mlir][Linalg] Allow linalg.copy to be vectorized with masking

[nicolasvasilache]

Add test

[nicolasvasilache]

Rebase

[qcolombet]

mlir/test/Dialect/Linalg/vectorization.mlir

2743 ↗

(On Diff #512730)

Could you check for the vector.create_mask?

[nicolasvasilache]

Check mask

mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp

if (!reduceOp || !getCombinerOpKind(reduceOp)) {
       return failure();
     }
   }
   return success();
 }
 
 static LogicalResult vectorizeDynamicLinalgOpPrecondition(linalg::LinalgOp op) {
   // TODO: Masking only supports dynamic generic ops for now.
-  if (!isa<linalg::GenericOp, linalg::FillOp>(op))
+  if (!isa<linalg::GenericOp, linalg::FillOp, linalg::CopyOp>(op))
     return failure();
 
   LDBG("Dynamically-shaped op meets vectorization pre-conditions\n");
   return success();
 }
 
 LogicalResult
 mlir::linalg::vectorizeLinalgOpPrecondition(LinalgOp linalgOp,