@@ -169,9 +169,9 @@ def attach_to_ir(self, op: OpView):
169169 "MBlock" : self .m_block ,
170170 "KBlock" : self .k_block ,
171171 "NBlock" : self .n_block ,
172- "innerMostMBlock " : self .innermost_m_block ,
173- "innerMostKBlock " : self .innermost_k_block ,
174- "innerMostNBlock " : self .innermost_n_block ,
172+ "innermostMBlock " : self .innermost_m_block ,
173+ "innermostKBlock " : self .innermost_k_block ,
174+ "innermostNBlock " : self .innermost_n_block ,
175175 }
176176 for name , value in attr_to_field .items ():
177177 op .attributes [name ] = IntegerAttr .get (T .i32 (), value )
@@ -180,29 +180,30 @@ def __repr__(self) -> str:
180180 return str (
181181 [
182182 self .m_threads ,
183- self .k_threads ,
184183 self .n_threads ,
184+ self .k_threads ,
185185 self .m_block ,
186- self .k_block ,
187186 self .n_block ,
187+ self .k_block ,
188188 self .innermost_m_block ,
189- self .innermost_k_block ,
190189 self .innermost_n_block ,
190+ self .innermost_k_block ,
191191 ]
192192 )
193193
194194 def __str__ (self ) -> str :
195195 obj_dict = {
196196 "MatMulConfig" : {
197- "MThreads" : self .m_threads ,
198- "KThreads" : self .k_threads ,
197+ "MThreads" : self .m_threads ,
199198 "NThreads" : self .n_threads ,
199+ "KThreads" : self .k_threads ,
200200 "MBlock" : self .m_block ,
201- "KBlock" : self .k_block ,
202201 "NBlock" : self .n_block ,
202+ "KBlock" : self .k_block ,
203203 "innerMostMBlock" : self .innermost_m_block ,
204- "innerMostKBlock" : self .innermost_k_block ,
205204 "innerMostNBlock" : self .innermost_n_block ,
205+ "innerMostKBlock" : self .innermost_k_block ,
206+
206207 }
207208 }
208209 return json .dumps (obj_dict , indent = 4 )
0 commit comments