fix code style

Author: Galaxy1458

src/flag_gems/runtime/backend/__init__.py

@@ -9,6 +9,7 @@
 from ..common import vendors
 from . import backend_utils
 
+
 class BackendState:
     """Singleton class to manage backend state variables."""
 
@@ -37,7 +38,7 @@ def __init__(self):
         self.customized_ops = None
 
 
-# Global singleton instance 
+# Global singleton instance
 _state = BackendState()
 
 
@@ -149,14 +150,13 @@ def _import_module_safe(module_name, vendor_name, module_type):
         return importlib.import_module(module_name)
     except ModuleNotFoundError:
         print(
-           f"[Note] No specialized {module_type} operators were found for "
-           f"the {vendor_name}, generic {module_type} operators will be used by default."
+            f"[Note] No specialized {module_type} operators were found for "
+            f"the {vendor_name}, generic {module_type} operators will be used by default."
         )
     except Exception as e:
         raise RuntimeError(f"Failed to import vendor extra lib: {e}")
 
 
-
 def import_vendor_extra_lib(vendor_name=None):
     if _state.vendor_extra_lib_imported:
         return
@@ -239,7 +239,7 @@ def get_module(vendor_name):
         return get_module(vendor_name)
 
     if _state.vendor_module is None:
-        _state.vendor_module = get_module("_" + vendor_name) 
+        _state.vendor_module = get_module("_" + vendor_name)
     return _state.vendor_module
 
 
@@ -290,7 +290,7 @@ def get_heuristic_config(vendor_name=None):
         try:
             _state.heuristic_config_module = importlib.import_module(mod_name)
         except Exception:
-            continue 
+            continue
     return getattr(_state.heuristic_config_module, "HEURISTICS_CONFIGS", None)
 
 
@@ -303,8 +303,10 @@ def get_tune_config(vendor_name=None):
 def get_expand_config(op_name=None, file_path=None):
     return backend_utils.get_expand_config(op_name=op_name, file_path=file_path)
 
+
 def get_backend_state() -> BackendState:
     """Get the global BackendState singleton instance."""
     return _state
 
+
 __all__ = ["*"]

src/flag_gems/runtime/backend/device.py

@@ -6,7 +6,6 @@
 import torch  # noqa: F401
 
 from .. import backend, error
-
 from ..common import (
     _VENDOR_TORCH_ATTR,
     UNSUPPORT_BF16,
@@ -19,6 +18,7 @@
 # A singleton class to manage device context.
 class DeviceDetector:
     """Singleton class to manage device context."""
+
     _instance = None
 
     def __new__(cls, *args, **kargs):
@@ -57,7 +57,7 @@ def get_vendor(self, vendor_name=None) -> tuple:
             return backend.get_vendor_info(vendor_from_env)
 
         vendor_name = self._get_vendor_from_quick_cmd()
-        if vendor_name :
+        if vendor_name:
             return backend.get_vendor_info(vendor_name)
         try:
             # Obtaining a vendor_info from the methods provided by torch or triton, but is not currently implemented.
@@ -75,7 +75,7 @@ def _get_vendor_from_quick_cmd(self):
             for vendor_name, attr in _VENDOR_TORCH_ATTR.items():
                 if hasattr(torch_npu, attr):
                     return vendor_name
-        except ImportError: 
+        except ImportError:
             pass
         return None
 
@@ -85,24 +85,24 @@ def _get_vendor_from_env(self):
 
     def _get_vendor_from_sys(self):
         vendor_infos = backend.get_vendor_infos()
-       
+
         def check_vendor(info):
             try:
                 cmd_args = shlex.split(info.device_query_cmd)
                 result = subprocess.run(cmd_args, capture_output=True, text=True)
                 return info if result.returncode == 0 else None
             except Exception:
                 return None
-        
-        with ThreadPoolExecutor() as executor:  
+
+        with ThreadPoolExecutor() as executor:
             futures = {
                 executor.submit(check_vendor, info): info for info in vendor_infos
             }
             for future in as_completed(futures):
                 result = future.result()
                 if result:
                     return result
-                
+
         error.device_not_found()
 
     def get_vendor_name(self):

src/flag_gems/runtime/common.py

@@ -1,5 +1,5 @@
-from enum import Enum
 import os
+from enum import Enum
 
 
 class vendors(Enum):
@@ -126,4 +126,12 @@ def get_all_vendors(cls) -> dict:
     "enflame": "gcu",
 }
 
-__all__ = ["vendors", "UNSUPPORT_FP64", "UNSUPPORT_BF16", "UNSUPPORT_INT64", "DEFAULT_STRATEGIES", "OP_KEY_ORDERS", "_VENDOR_TORCH_ATTR"]
+__all__ = [
+    "vendors",
+    "UNSUPPORT_FP64",
+    "UNSUPPORT_BF16",
+    "UNSUPPORT_INT64",
+    "DEFAULT_STRATEGIES",
+    "OP_KEY_ORDERS",
+    "_VENDOR_TORCH_ATTR",
+]

src/flag_gems/runtime/configloader.py

@@ -1,12 +1,12 @@
 import copy
-import os
 import warnings
 
 import triton
 
 from . import backend, common
 from .backend.device import DeviceDetector
 
+
 class ConfigLoader(object):
     _instance = None
 
@@ -47,7 +47,7 @@ def __init__(self):
                 self.triton_config_default["num_ldmatrixes"] = 0
             self.expand_config_registry = self._build_expand_registry()
             self.load_all()
-    
+
     def update_config_from_arch(self):
         try:
             archEvent = backend.BackendArchEvent()
@@ -76,9 +76,9 @@ def _create_triton_config(self, single_config, current_config):
             "num_ctas": current_config["num_ctas"],
         }
         if self.device.vendor_name == "hygon":
-            kwargs["num_ldmatrixes"] = current_config["num_ldmatrixes"] 
+            kwargs["num_ldmatrixes"] = current_config["num_ldmatrixes"]
         return triton.Config(single_config["META"], **kwargs)
-    
+
     def _build_configs_by_op(self, op_name, ranges, pre_hook=None):
         if op_name == "bmm":
             return [