Format and update filecheck checks

Author: ashermancinelli

mlir/python/mlir/dialects/gpu/__init__.py

@@ -87,9 +87,8 @@ def __init__(
 
         if known_block_size is not None:
             if isinstance(known_block_size, Sequence):
-                self.attributes[self.KNOWN_BLOCK_SIZE_ATTR_NAME] = (
-                    DenseI32ArrayAttr.get(known_block_size)
-                )
+                block_size = DenseI32ArrayAttr.get(known_block_size)
+                self.attributes[self.KNOWN_BLOCK_SIZE_ATTR_NAME] = block_size
             elif isinstance(known_block_size, DenseI32ArrayAttr):
                 self.attributes[self.KNOWN_BLOCK_SIZE_ATTR_NAME] = known_block_size
             else:
@@ -99,9 +98,8 @@ def __init__(
 
         if known_grid_size is not None:
             if isinstance(known_grid_size, Sequence):
-                self.attributes[self.KNOWN_GRID_SIZE_ATTR_NAME] = DenseI32ArrayAttr.get(
-                    known_grid_size
-                )
+                grid_size = DenseI32ArrayAttr.get(known_grid_size)
+                self.attributes[self.KNOWN_GRID_SIZE_ATTR_NAME] = grid_size
             elif isinstance(known_grid_size, DenseI32ArrayAttr):
                 self.attributes[self.KNOWN_GRID_SIZE_ATTR_NAME] = known_grid_size
             else:

mlir/test/python/dialects/gpu/dialect.py

@@ -148,8 +148,8 @@ def builder(func: gpu.GPUFuncOp) -> None:
     # CHECK:   gpu.return
     # CHECK: }
     # CHECK: gpu.func @kernel1() kernel attributes
-    # CHECK-SAME: gpu.known_block_size = array<i32: 1, 2, 3>
-    # CHECK-SAME: gpu.known_grid_size = array<i32: 4, 5, 6>
+    # CHECK-SAME: known_block_size = array<i32: 1, 2, 3>
+    # CHECK-SAME: known_grid_size = array<i32: 4, 5, 6>
     # CHECK:   %[[VAL_0:.*]] = gpu.global_id  x
     # CHECK:   gpu.return
     # CHECK: }