From 53c4429f370dcae2b8b66a0fac03f97358ad274f Mon Sep 17 00:00:00 2001
From: Yanxing-Shi <yanxishi@amd.com>
Date: Wed, 14 May 2025 15:46:50 +0000
Subject: [PATCH] format

---
 tile_engine/ops/gemm/benchmark_gemm.hpp       |  6 ++++--
 tile_engine/ops/gemm/codegen_utils.py         | 11 ++++++-----
 tile_engine/ops/gemm/gemm_instance_builder.py |  3 ++-
 tile_engine/ops/gemm/json_config.py           |  2 +-
 4 files changed, 13 insertions(+), 9 deletions(-)

diff --git a/tile_engine/ops/gemm/benchmark_gemm.hpp b/tile_engine/ops/gemm/benchmark_gemm.hpp
index f0ce51a896..38e967e8d0 100644
--- a/tile_engine/ops/gemm/benchmark_gemm.hpp
+++ b/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"
            << "}";
diff --git a/tile_engine/ops/gemm/codegen_utils.py b/tile_engine/ops/gemm/codegen_utils.py
index 67dfa0570b..b7cea4abb7 100644
--- a/tile_engine/ops/gemm/codegen_utils.py
+++ b/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()
diff --git a/tile_engine/ops/gemm/gemm_instance_builder.py b/tile_engine/ops/gemm/gemm_instance_builder.py
index 8589d910c8..4368a6b5ca 100755
--- a/tile_engine/ops/gemm/gemm_instance_builder.py
+++ b/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(
diff --git a/tile_engine/ops/gemm/json_config.py b/tile_engine/ops/gemm/json_config.py
index 0408b2dfb5..4c5c1074c9 100644
--- a/tile_engine/ops/gemm/json_config.py
+++ b/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: