Post-commit changes to c57b6a4d6bd (#11766)

diff --git a/compiler/src/iree/compiler/Codegen/Sandbox/LinalgTensorCodegenDriver.cpp b/compiler/src/iree/compiler/Codegen/Sandbox/LinalgTensorCodegenDriver.cpp
index a21ae16..89b164f 100644
--- a/compiler/src/iree/compiler/Codegen/Sandbox/LinalgTensorCodegenDriver.cpp
+++ b/compiler/src/iree/compiler/Codegen/Sandbox/LinalgTensorCodegenDriver.cpp
@@ -147,7 +147,7 @@
     return {};
   }
 
-  // TODO: Not easy to infer the tiles sizes for an op that is not the root op.
+  // TODO: Infer the tiles sizes for an op that is not the root op.
   if (*rootOp != linalgOp.getOperation()) {
     return {};
   }
diff --git a/llvm-external-projects/iree-dialects/include/iree-dialects/Dialect/LinalgExt/Passes/Passes.h b/llvm-external-projects/iree-dialects/include/iree-dialects/Dialect/LinalgExt/Passes/Passes.h
index d495a23..9a84f10 100644
--- a/llvm-external-projects/iree-dialects/include/iree-dialects/Dialect/LinalgExt/Passes/Passes.h
+++ b/llvm-external-projects/iree-dialects/include/iree-dialects/Dialect/LinalgExt/Passes/Passes.h
@@ -247,7 +247,7 @@
 
 /// Create a LinalgStrategyVectorizePass.
 using VectorSizeComputationFunction =
-    std::function<SmallVector<int64_t, 4>(linalg::LinalgOp, ArrayRef<int64_t>)>;
+    std::function<SmallVector<int64_t>(linalg::LinalgOp, ArrayRef<int64_t>)>;
 
 struct LinalgVectorizationOptions {
   /// Canonical vector sizes for the vector iteration space (i.e., vectorization