Skip to content

Commit a368430

Browse files
committed
roll back unnecessary change
1 parent 442c18a commit a368430

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

mlir/lib/Dialect/XeGPU/Transforms/XeGPUBlocking.cpp

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -205,20 +205,19 @@ XeGPUBlockingPass::getTileShape(Operation *op) const {
205205

206206
bool XeGPUBlockingPass::needsUnroll(Operation *op) const {
207207
// skip the op if any of its operands or results has workgroup level layouts
208-
bool hasSgLayoutOperands =
208+
bool hasWgLayoutOperands =
209209
llvm::any_of(op->getOpOperands(), [](OpOperand &opr) {
210210
xegpu::DistributeLayoutAttr layout =
211211
xegpu::getDistributeLayoutAttr(opr);
212212
return layout && layout.isForWorkgroup();
213213
});
214-
bool hasSgLayoutResults =
214+
bool hasWgLayoutResults =
215215
llvm::any_of(op->getOpResults(), [](OpResult result) {
216216
xegpu::DistributeLayoutAttr layout =
217217
xegpu::getDistributeLayoutAttr(result);
218218
return layout && layout.isForWorkgroup();
219219
});
220-
221-
if (hasSgLayoutOperands || hasSgLayoutResults) {
220+
if (hasWgLayoutOperands || hasWgLayoutResults) {
222221
LDBG() << "skip unrolling for op with workgroup level layout: " << *op;
223222
return false;
224223
}

0 commit comments

Comments
 (0)