FIX

intel · Oct 17, 2024 · 07a94f0 · 07a94f0
1 parent 085ef35
commit 07a94f0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 13 deletions.
diff --git a/lib/gc/Analysis/MatmulConfigAnalysis.cpp b/lib/gc/Analysis/MatmulConfigAnalysis.cpp
@@ -386,13 +386,13 @@ bool readConfigFromAttrs(MatmulConfig &config, ArrayRef<NamedAttribute> attrs) {
     } else if (attr.getName() == "MThreads") {
       config.MThreads = cast<IntegerAttr>(attr.getValue()).getInt();
       cfgItemCnt++;
-    } else if (attr.getName() == "innerMostMBlock") {
+    } else if (attr.getName() == "innermostMBlock") {
       config.innerMostMBlock = cast<IntegerAttr>(attr.getValue()).getInt();
       cfgItemCnt++;
-    } else if (attr.getName() == "innerMostNBlock") {
+    } else if (attr.getName() == "innermostNBlock") {
       config.innerMostNBlock = cast<IntegerAttr>(attr.getValue()).getInt();
       cfgItemCnt++;
-    } else if (attr.getName() == "innerMostKBlock") {
+    } else if (attr.getName() == "innermostKBlock") {
       config.innerMostKBlock = cast<IntegerAttr>(attr.getValue()).getInt();
       cfgItemCnt++;
     }

diff --git a/test/benchgc/src/benchgc/tuner/op_config.py b/test/benchgc/src/benchgc/tuner/op_config.py
@@ -169,9 +169,9 @@ def attach_to_ir(self, op: OpView):
             "MBlock": self.m_block,
             "KBlock": self.k_block,
             "NBlock": self.n_block,
-            "innerMostMBlock": self.innermost_m_block,
-            "innerMostKBlock": self.innermost_k_block,
-            "innerMostNBlock": self.innermost_n_block,
+            "innermostMBlock": self.innermost_m_block,
+            "innermostKBlock": self.innermost_k_block,
+            "innermostNBlock": self.innermost_n_block,
         }
         for name, value in attr_to_field.items():
             op.attributes[name] = IntegerAttr.get(T.i32(), value)
@@ -180,29 +180,30 @@ def __repr__(self) -> str:
         return str(
             [
                 self.m_threads,
-                self.k_threads,
                 self.n_threads,
+                self.k_threads,
                 self.m_block,
-                self.k_block,
                 self.n_block,
+                self.k_block,
                 self.innermost_m_block,
-                self.innermost_k_block,
                 self.innermost_n_block,
+                self.innermost_k_block,
             ]
         )
 
     def __str__(self) -> str:
         obj_dict = {
             "MatMulConfig": {
-                "MThreads": self.m_threads,
-                "KThreads": self.k_threads,
+                "MThreads": self.m_threads,          
                 "NThreads": self.n_threads,
+                "KThreads": self.k_threads,
                 "MBlock": self.m_block,
-                "KBlock": self.k_block,
                 "NBlock": self.n_block,
+                "KBlock": self.k_block,            
                 "innerMostMBlock": self.innermost_m_block,
-                "innerMostKBlock": self.innermost_k_block,
                 "innerMostNBlock": self.innermost_n_block,
+                "innerMostKBlock": self.innermost_k_block,
+
             }
         }
         return json.dumps(obj_dict, indent=4)