Fixups and minor changes from code review

Co-authored-by: Jakub Kuderski <[email protected]>

Author: krzysz00

mlir/include/mlir/Dialect/Affine/IR/AffineOps.td

@@ -1109,7 +1109,7 @@ def AffineLinearizeIndexOp : Affine_Op<"linearize_index",
     The `affine.linearize_index` operation takes a sequence of index values and a
     basis of the same length and linearizes the indices using that basis.
 
-    That is, for indices %idx_1 through %i_N and basis elements b_1 through b_N,
+    That is, for indices `%idx_1` through `%i_N` and basis elements `b_1` through `b_N`,
     it computes
 
     ```
@@ -1124,13 +1124,13 @@ def AffineLinearizeIndexOp : Affine_Op<"linearize_index",
 
     Example:
 
-    ```
+    ```mlir
     %linear_index = affine.delinearize_index [%index_0, %index_1, %index_2] (16, 224, 224) : index
     ```
 
     In the above example, `%linear_index` conceptually holds the following:
 
-    ```
+    ```mlir
     #map = affine_map<()[s0, s1, s2] -> (s0 * 50176 + s1 * 224 + s2)>
     %linear_index = affine.apply #map()[%index_0, %index_1, %index_2]
     ```

mlir/lib/Dialect/Affine/IR/AffineOps.cpp

@@ -4723,8 +4723,8 @@ namespace {
 ///
 /// Alternatively, if `%x` in the above is a known constant 0, remove it even if
 /// the operation isn't asserted to be `disjoint`.
-struct DropLinearizeUnitComponentsIfDisjointOrZero
-    : public OpRewritePattern<affine::AffineLinearizeIndexOp> {
+struct DropLinearizeUnitComponentsIfDisjointOrZero final
+    : OpRewritePattern<affine::AffineLinearizeIndexOp> {
   using OpRewritePattern::OpRewritePattern;
 
   LogicalResult matchAndRewrite(affine::AffineLinearizeIndexOp op,

mlir/lib/Dialect/Affine/Transforms/AffineExpandIndexOps.cpp

@@ -48,9 +48,9 @@ struct LowerDelinearizeIndexOps
 
 /// Lowers `affine.linearize_index` into a sequence of multiplications and
 /// additions.
-struct LowerLinearizeIndexOps
-    : public OpRewritePattern<AffineLinearizeIndexOp> {
-  using OpRewritePattern<AffineLinearizeIndexOp>::OpRewritePattern;
+struct LowerLinearizeIndexOps final
+    : OpRewritePattern<AffineLinearizeIndexOp> {
+  using OpRewritePattern::OpRewritePattern;
   LogicalResult matchAndRewrite(AffineLinearizeIndexOp op,
                                 PatternRewriter &rewriter) const override {
     SmallVector<OpFoldResult> multiIndex =