File tree Expand file tree Collapse file tree 1 file changed +3
-4
lines changed
mlir/lib/Dialect/XeGPU/Transforms Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Original file line number Diff line number Diff line change @@ -205,20 +205,19 @@ XeGPUBlockingPass::getTileShape(Operation *op) const {
205205
206206bool 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 }
You can’t perform that action at this time.
0 commit comments