format

Signed-off-by: Qidong Su <soodoshll@gmail.com>

Author: soodoshll

examples/hopper_matmul/matmul_v0.py

@@ -10,7 +10,6 @@
 from tilus.utils import benchmark_func, cdiv
 
 
-
 @tilus.autotune(
     "block_m, block_n", [(64, 128), (128, 128), (128, 256), (256, 128), (256, 256)]
 )

python/tilus/lang/instructions/wgmma.py

@@ -34,7 +34,9 @@ def wait_group(self, n: Union[Expr, int]) -> None:
 
     def mma(self, a: SharedTensor | RegisterTensor, b: SharedTensor, d: RegisterTensor) -> None:
         if any(len(tensor.shape) != 2 for tensor in (a, b, d)):
-            raise InstructionError("mma requires 2D tensors, got shapes {}".format([tensor.shape for tensor in (a, b, d)]))
+            raise InstructionError(
+                "mma requires 2D tensors, got shapes {}".format([tensor.shape for tensor in (a, b, d)])
+            )
         if isinstance(a, SharedTensor):
             self._builder.wgmma_mma_ss(a, b, d)
         elif isinstance(a, RegisterTensor):