format

tile_engine/ops/gemm/benchmark_gemm.hpp

@@ -104,8 +104,10 @@ struct KernelInstance
     friend std::ostream& operator<<(std::ostream& os, const KernelInstance& obj)
     {
         os << "{\n"
-           << " \"name\": \"" << "{\n"
-           << obj.name << "\n}" << "\",\n"
+           << " \"name\": \""
+           << "{\n"
+           << obj.name << "\n}"
+           << "\",\n"
            << " \"problem\": \"" << obj.problem << "\",\n"
            << " \"perf_result\": " << obj.perf_result << "\n"
            << "}";

tile_engine/ops/gemm/codegen_utils.py

@@ -164,11 +164,12 @@ warp_tile_supported_combinations = {
 
 # To Do: remove some unsupported combinations
 trait_unsupported_combinations = {
-            ("compv3", "cshuffle", "interwave"),
-            ("compv3", "default", "interwave"),
-            ("compv4", "cshuffle", "interwave"),
-            ("compv4", "default", "interwave")
-        }
+    ("compv3", "cshuffle", "interwave"),
+    ("compv3", "default", "interwave"),
+    ("compv4", "cshuffle", "interwave"),
+    ("compv4", "default", "interwave")
+}
+
 
 def element_size(data_type: str) -> float:
     data_type = data_type.lower()

tile_engine/ops/gemm/gemm_instance_builder.py

@@ -375,7 +375,8 @@ struct GemmKernel {{
         # Warp combination validation
         warp_tile_key = f"{self.config.problem.datatype_map['matrix_a']}_{self.config.problem.datatype_map['matrix_b']}_{self.config.problem.datatype_map['matrix_c']}"
         current_combination = [warp_tile_m, warp_tile_n, warp_tile_k]
-        allowed_combinations = warp_tile_supported_combinations.get(warp_tile_key, [])
+        allowed_combinations = warp_tile_supported_combinations.get(
+            warp_tile_key, [])
 
         if current_combination not in allowed_combinations:
             logging.warning(

tile_engine/ops/gemm/json_config.py

@@ -341,7 +341,7 @@ class GemmConfig(BaseModel):
         try:
             if not config_path.exists():
                 raise FileNotFoundError(f"Config file {filepath} not found")
-            config_path.stat()  
+            config_path.stat()
 
             with open(filepath, 'r') as f:
                 try: