File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed
third_party/intel/lib/TritonIntelGPUTransforms Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change @@ -161,12 +161,14 @@ struct TritonIntelGPUMaterializeBlockPointerPass
161
161
162
162
LDBG (" Considering tensor of pointer of memory accessing op: " << *op);
163
163
164
+ #if 0
164
165
if (auto loadOp = dyn_cast<tt::LoadOp>(*op)) {
165
166
if (loadOp.getMask()) {
166
167
LDBG("Load op has mask, skip block IO attribute");
167
168
return;
168
169
}
169
170
}
171
+ #endif
170
172
171
173
// The axis info gives the information about the value of the indices
172
174
// tensor. For example, if the indices tensor is tensor<8x16xi32> and
@@ -233,6 +235,12 @@ struct TritonIntelGPUMaterializeBlockPointerPass
233
235
StringAttr::get (context, " row_major" ));
234
236
}
235
237
238
+ if (isMajor (0 /* fastChangeDim*/ )) {
239
+ LDBG (" Setting column_major attribute\n " );
240
+ op->setAttr (ttgi::TritonIntelGPUDialect::getBlockIOAttrName (),
241
+ StringAttr::get (context, " column_major" ));
242
+ }
243
+
236
244
// TODO: set column_major attribute
237
245
}
238
246
You can’t perform that action at this time.
0 commit comments