Skip to content

Commit 07a94f0

Browse files
committed
FIX
1 parent 085ef35 commit 07a94f0

File tree

2 files changed

+14
-13
lines changed

2 files changed

+14
-13
lines changed

lib/gc/Analysis/MatmulConfigAnalysis.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -386,13 +386,13 @@ bool readConfigFromAttrs(MatmulConfig &config, ArrayRef<NamedAttribute> attrs) {
386386
} else if (attr.getName() == "MThreads") {
387387
config.MThreads = cast<IntegerAttr>(attr.getValue()).getInt();
388388
cfgItemCnt++;
389-
} else if (attr.getName() == "innerMostMBlock") {
389+
} else if (attr.getName() == "innermostMBlock") {
390390
config.innerMostMBlock = cast<IntegerAttr>(attr.getValue()).getInt();
391391
cfgItemCnt++;
392-
} else if (attr.getName() == "innerMostNBlock") {
392+
} else if (attr.getName() == "innermostNBlock") {
393393
config.innerMostNBlock = cast<IntegerAttr>(attr.getValue()).getInt();
394394
cfgItemCnt++;
395-
} else if (attr.getName() == "innerMostKBlock") {
395+
} else if (attr.getName() == "innermostKBlock") {
396396
config.innerMostKBlock = cast<IntegerAttr>(attr.getValue()).getInt();
397397
cfgItemCnt++;
398398
}

test/benchgc/src/benchgc/tuner/op_config.py

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)