@@ -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