message: Use ForLoopSpecializationPass to help with vectorization of kernels.
diff: @@ -505,6 +505,11 @@ Status LowerLHLOToGPU(mlir::ModuleOp module, LowerLHLOToGPUOptions options) {
  // Some basic cleanup.
  pm.addNestedPass<::mlir::FuncOp>(::mlir::createCanonicalizerPass());
  pm.addNestedPass<::mlir::FuncOp>(::mlir::createCSEPass());
  + // Make loops with min bounds into a conditional plus static bounds.
  + // Only do this if we unrolled in the first place.
  + if (!options.unroll_factors.empty()) {
  + pm.addNestedPass<::mlir::FuncOp>(mlir::createForLoopSpecializationPass());
  + }
  // Approximate of requested.
  if (options.use_approximations) {
  pm.addNestedPass<::mlir::FuncOp>(
  