Skip to content

Commit d85a2ba

Browse files
committed
Debug flex bwd
Signed-off-by: Lu,Chengjun <chengjun.lu@intel.com>
1 parent 0e06e61 commit d85a2ba

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

third_party/intel/lib/TritonIntelGPUTransforms/MaterializeBlockPointer.cpp

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)