Apply code-format changes

Author: flagtree-bot

python/setup_tools/utils/spacemit.py

@@ -35,6 +35,7 @@ def _cache_asset(name):
 
 
 def _set_env_path(name):
+
     def set_path(path):
         os.environ[name] = str(path)
 
@@ -48,8 +49,7 @@ def register_cache(cache, flagtree_backend, check_env, set_llvm_env):
 
     assets = (
         ("LLVM", "llvm_installed", "LLVM_SYSPATH", set_llvm_env),
-        ("SPINE_MLIR", "spine_mlir_installed", "SPINE_MLIR_INSTALL_DIR",
-         _set_env_path("SPINE_MLIR_INSTALL_DIR")),
+        ("SPINE_MLIR", "spine_mlir_installed", "SPINE_MLIR_INSTALL_DIR", _set_env_path("SPINE_MLIR_INSTALL_DIR")),
         ("SPINE_RUNTIME", "spine_runtime_installed", "SPINE_RUNTIME_INSTALL_DIR",
          _set_env_path("SPINE_RUNTIME_INSTALL_DIR")),
         ("RPC_RUNTIME", "rpc_runtime_installed", None, None),
@@ -162,10 +162,7 @@ def install_extension(*args, **kargs):
     from build_helpers import get_cmake_dir
 
     build_temp = Path(get_cmake_dir())
-    spine_triton_opt_src = (
-        build_temp / "third_party" / "spacemit" / "tools" /
-        "spine-triton-opt" / "spine-triton-opt"
-    )
+    spine_triton_opt_src = (build_temp / "third_party" / "spacemit" / "tools" / "spine-triton-opt" / "spine-triton-opt")
     if not spine_triton_opt_src.exists():
         print(f"[spacemit] Warning: spine-triton-opt not found at {spine_triton_opt_src}")
         return

third_party/spacemit/patch/flagtree.patch

@@ -210,7 +210,7 @@ index 3d60f95c4..376bb9b75 100644
 +
  # flagtree setup print config
  YELLOW, NC = helper.get_console_colors()
- 
+
 @@ -97,22 +120,21 @@ class BackendInstaller:
 
      @staticmethod
@@ -397,7 +397,7 @@ index 3d60f95c4..376bb9b75 100644
          cmake_args += helper.get_backend_cmake_args(build_ext=self)  # flagtree
 @@ -568,7 +684,8 @@ class CMakeBuild(build_ext):
          cmake_args += helper.customize_gluon_cmake_args()
- 
+
 -        if is_offline_build():
 -            # unit test builds fetch googletests from GitHub
 +        if is_offline_build() or _is_spacemit_build():
@@ -434,8 +434,8 @@ index 3d60f95c4..376bb9b75 100644
 +    ]
 +else:
 +    backends = helper.init_backends(BackendInstaller)  # flagtree
- 
- 
+
+
 diff --git a/third_party/proton/Dialect/include/Dialect/Proton/IR/ProtonDialect.td b/third_party/proton/Dialect/include/Dialect/Proton/IR/ProtonDialect.td
 index 8d35a5ec1..e476cad96 100644
 --- a/third_party/proton/Dialect/include/Dialect/Proton/IR/ProtonDialect.td

third_party/spacemit/scripts/smoke_test.py

@@ -10,8 +10,7 @@
 
 
 def check(description, condition, detail=""):
-    print(f"[{'OK ' if condition else 'FAIL'}] {description}"
-          + (f"  ({detail})" if detail else ""))
+    print(f"[{'OK ' if condition else 'FAIL'}] {description}" + (f"  ({detail})" if detail else ""))
     return condition
 
 
@@ -20,26 +19,20 @@ def main():
     ok &= check("import triton", True, f"version={triton.__version__}")
 
     names = sorted(backends)
-    ok &= check("spacemit backend registered", "spacemit" in names,
-                f"backends={names}")
+    ok &= check("spacemit backend registered", "spacemit" in names, f"backends={names}")
 
     backend = backends["spacemit"]
     ok &= check(
         "compiler/driver",
-        backend.compiler.__name__ == "CPUBackend"
-        and backend.driver.__name__ == "CPUDriver",
+        backend.compiler.__name__ == "CPUBackend" and backend.driver.__name__ == "CPUDriver",
         f"{backend.compiler.__name__}/{backend.driver.__name__}",
     )
 
-    pads = [value for value in dir(ir.PADDING_OPTION)
-            if value.startswith("PAD")]
-    ok &= check("PADDING_OPTION adapt",
-                {"PAD_NEG_INF", "PAD_INF"} <= set(pads), f"pads={pads}")
+    pads = [value for value in dir(ir.PADDING_OPTION) if value.startswith("PAD")]
+    ok &= check("PADDING_OPTION adapt", {"PAD_NEG_INF", "PAD_INF"} <= set(pads), f"pads={pads}")
 
     submodules = [name for name in dir(spacemit) if not name.startswith("__")]
-    ok &= check("spacemit C++ submod",
-                {"load_dialects", "tle_ir", "xsmt_ir"} <= set(submodules),
-                f"sub={submodules}")
+    ok &= check("spacemit C++ submod", {"load_dialects", "tle_ir", "xsmt_ir"} <= set(submodules), f"sub={submodules}")
 
     host = os.environ.get("SPINE_TRITON_RPC_HOST", "127.0.0.1")
     port = int(os.environ.get("SPINE_TRITON_RPC_PORT", "9999"))