Fix black formatting in driver.py and padded_matmul.py

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: erwei-xilinx

amd_triton_npu/backend/driver.py

@@ -1141,8 +1141,9 @@ def _generate_elf_launcher(constants, signature, kernel_name):
 """
 
 
-def compile_module(launcher_src, kernel_placeholder_name, output_format="xclbin",
-                   actual_sizes=None):
+def compile_module(
+    launcher_src, kernel_placeholder_name, output_format="xclbin", actual_sizes=None
+):
     py_version = sys.version_info
     if platform.system() == "Windows":
         py_include_dir = os.path.join(sys.base_prefix, "include")
@@ -1186,8 +1187,9 @@ def launch(
         air_proj_path = _get_air_project_path()
         os.makedirs(air_proj_path, exist_ok=True)
         Path(os.path.join(air_proj_path, "asm_src.mlir")).write_bytes(asm_src)
-        air_output = _ttshared_to_air(asm_src, gridX, gridY, gridZ,
-                                      actual_sizes=actual_sizes)
+        air_output = _ttshared_to_air(
+            asm_src, gridX, gridY, gridZ, actual_sizes=actual_sizes
+        )
         with open(Path(os.path.join(air_proj_path, "asm_air_output.mlir")), "w") as f:
             f.write(str(air_output))
 
@@ -1408,10 +1410,16 @@ def __init__(self, src, metadata):
                 n_val = raw_constants.get((n_idx,))
                 if m_val is not None and n_val is not None:
                     # Check if BLOCK_SIZE_M/N are available to determine alignment
-                    bsm_idx = (arg_names.index("BLOCK_SIZE_M")
-                               if "BLOCK_SIZE_M" in arg_names else None)
-                    bsn_idx = (arg_names.index("BLOCK_SIZE_N")
-                               if "BLOCK_SIZE_N" in arg_names else None)
+                    bsm_idx = (
+                        arg_names.index("BLOCK_SIZE_M")
+                        if "BLOCK_SIZE_M" in arg_names
+                        else None
+                    )
+                    bsn_idx = (
+                        arg_names.index("BLOCK_SIZE_N")
+                        if "BLOCK_SIZE_N" in arg_names
+                        else None
+                    )
                     bsm = raw_constants.get((bsm_idx,)) if bsm_idx is not None else None
                     bsn = raw_constants.get((bsn_idx,)) if bsn_idx is not None else None
                     needs_padding = True
@@ -1423,7 +1431,9 @@ def __init__(self, src, metadata):
         # Later KERNEL_NAME_PLACEHOLDER will be used to assign the kernel name
         # in the following launch function.
         self.launch = compile_module(
-            launcher_src, kernel_placeholder_name, self.output_format,
+            launcher_src,
+            kernel_placeholder_name,
+            self.output_format,
             actual_sizes=actual_sizes,
         )
 

examples/padded_matmul/padded_matmul.py

@@ -41,9 +41,7 @@
 N_actual = 500
 K_val = 1024
 
-assert K_val % K_L2_TILE == 0, (
-    f"K={K_val} must be divisible by K_L2_TILE={K_L2_TILE}"
-)
+assert K_val % K_L2_TILE == 0, f"K={K_val} must be divisible by K_L2_TILE={K_L2_TILE}"
 
 # === Padded/allocated dimensions ===
 M_padded = math.ceil(M_actual / LAUNCH_TILE_M) * LAUNCH_TILE_M  # 512
@@ -175,9 +173,7 @@ def run_padded_matmul():
         if not np.isclose(actual, expected, rtol=0.1, atol=10.0):
             errors += 1
             if errors <= 5:
-                print(
-                    f"Mismatch at ({m}, {n}): actual={actual}, expected={expected}"
-                )
+                print(f"Mismatch at ({m}, {n}): actual={actual}, expected={expected}")
 
     total = len(sample_m)
     if errors == 0: