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
161161
162162 LDBG (" Considering tensor of pointer of memory accessing op: " << *op);
163163
164+ #if 0
164165 if (auto loadOp = dyn_cast<tt::LoadOp>(*op)) {
165166 if (loadOp.getMask()) {
166167 LDBG("Load op has mask, skip block IO attribute");
167168 return;
168169 }
169170 }
171+ #endif
170172
171173 // The axis info gives the information about the value of the indices
172174 // tensor. For example, if the indices tensor is tensor<8x16xi32> and
@@ -233,6 +235,12 @@ struct TritonIntelGPUMaterializeBlockPointerPass
233235 StringAttr::get (context, " row_major" ));
234236 }
235237
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+
236244 // TODO: set column_major attribute
237245 }
238246
You can’t perform that action at this time.
0 commit comments