Format

Author: ashermancinelli

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

@@ -80,9 +80,7 @@ def __init__(
         elif isinstance(sym_name, StringAttr):
             self.attributes[self.SYM_NAME_ATTR_NAME] = sym_name
         else:
-            raise ValueError(
-                "sym_name must be a string or a StringAttr"
-            )
+            raise ValueError("sym_name must be a string or a StringAttr")
 
         if kernel:
             self.attributes[self.KERNEL_ATTR_NAME] = UnitAttr.get()
@@ -137,8 +135,8 @@ def add_entry_block(self) -> Block:
     def entry_block(self) -> Block:
         if len(self.body.blocks) == 0:
             raise RuntimeError(
-                f"Entry block does not exist for {self.name.value}." +
-                " Do you need to call the add_entry_block() method on this GPUFuncOp?"
+                f"Entry block does not exist for {self.name.value}."
+                + " Do you need to call the add_entry_block() method on this GPUFuncOp?"
             )
         return self.body.blocks[0]
 

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

@@ -93,7 +93,10 @@ def builder(func: gpu.GPUFuncOp) -> None:
                 func.entry_block
                 assert False, "Expected RuntimeError"
             except RuntimeError as e:
-                assert str(e) == "Entry block does not exist for kernel0. Do you need to call the add_entry_block() method on this GPUFuncOp?"
+                assert (
+                    str(e)
+                    == "Entry block does not exist for kernel0. Do you need to call the add_entry_block() method on this GPUFuncOp?"
+                )
 
             block = func.add_entry_block()
             with InsertionPoint(block):