Skip to content

Commit 45a3d28

Browse files
committed
renaming
1 parent 76f8761 commit 45a3d28

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ struct UnrollPattern : public OpRewritePattern<SourceOp> {
7373
}
7474

7575
// copy the layout attribte and drops the inst_data field.
76-
xegpu::LayoutAttr getLaneLayoutAttr(Attribute attr) const {
76+
xegpu::LayoutAttr getLaneLevelAttrsOnly(Attribute attr) const {
7777
auto layout = dyn_cast_if_present<xegpu::LayoutAttr>(attr);
7878
if (!layout || layout.getLaneLayout() == nullptr)
7979
return xegpu::LayoutAttr();
@@ -90,7 +90,7 @@ struct UnrollPattern : public OpRewritePattern<SourceOp> {
9090
auto encoding = tdescTy.getEncoding();
9191
auto layout = tdescTy.getLayout();
9292
newTy = xegpu::TensorDescType::get(ctx, blockSize, elemTy, encoding,
93-
getLaneLayoutAttr(layout));
93+
getLaneLevelAttrsOnly(layout));
9494
} else {
9595
newTy = type.clone(blockSize, elemTy);
9696
}
@@ -205,7 +205,7 @@ struct UnrollCreateNdOp : public UnrollPattern<xegpu::CreateNdDescOp> {
205205
return failure();
206206

207207
auto encoding = tdescTy.getEncoding();
208-
auto newLayout = getLaneLayoutAttr(layout);
208+
auto newLayout = getLaneLevelAttrsOnly(layout);
209209
auto newTdescTy = xegpu::TensorDescType::get(
210210
ctx, targetShape, tdescTy.getElementType(), encoding, newLayout);
211211

0 commit comments

Comments
 (0)