fixup! [mlir][linalg] Add support for masked vectorization of tensor.insert_slice (2/N)

Address comment from Hanhan

Author: banach-space

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

@@ -2743,29 +2743,25 @@ vectorizeAsInsertSliceOp(RewriterBase &rewriter, tensor::InsertSliceOp sliceOp,
         sliceOp.getLoc(), readMaskType, reifiedSrcSizes[0]);
   }
 
-  // 3.a. TransferReadOp
   SmallVector<Value> readIndices(
       vecType.getRank(),
       rewriter.create<arith::ConstantIndexOp>(sliceOp.getLoc(), 0));
   Operation *read = rewriter.create<vector::TransferReadOp>(
       sliceOp.getLoc(), vecType, source, readIndices, padValue,
       ArrayRef<bool>{readInBounds});
 
-  // Mask the xfer_read Op
-  if (!inputVectorSizes.empty()) {
+  if (maskOp) {
     read = mlir::vector::maskOperation(rewriter, read, maskOp);
   }
 
-  // 3.b. TransferWriteOp
   auto writeIndices = getValueOrCreateConstantIndexOp(
       rewriter, sliceOp.getLoc(), sliceOp.getMixedOffsets());
 
   Operation *write = rewriter.create<vector::TransferWriteOp>(
       sliceOp.getLoc(), read->getResult(0), sliceOp.getDest(), writeIndices,
       ArrayRef<bool>{writeInBounds});
 
-  // Mask the xfer_write Op
-  if (!inputVectorSizes.empty()) {
+  if (maskOp) {
     write = mlir::vector::maskOperation(rewriter, write, maskOp);
   }