File tree Expand file tree Collapse file tree 2 files changed +7
-3
lines changed
lib/Dialect/XeGPU/Transforms Expand file tree Collapse file tree 2 files changed +7
-3
lines changed Original file line number Diff line number Diff line change @@ -303,7 +303,8 @@ void XeGPUBlockingPass::runOnOperation() {
303303 // If the encoding is a ScatterTensorDescAttr, we need to
304304 // potentially adjust the chunk size based on the inst_data.
305305 if (tdescTy.isScattered ()) {
306- auto scatterAttr = llvm::dyn_cast_if_present<xegpu::ScatterTensorDescAttr>(encoding);
306+ auto scatterAttr =
307+ llvm::dyn_cast_if_present<xegpu::ScatterTensorDescAttr>(encoding);
307308 int64_t chunkSize = scatterAttr.getChunkSize ().getInt ();
308309
309310 if (chunkSize > 1 ) {
Original file line number Diff line number Diff line change @@ -107,7 +107,9 @@ struct TestXeGPUUnrollingPatterns
107107 // If the encoding is a ScatterTensorDescAttr, we need to
108108 // potentially adjust the chunk size based on the inst_data.
109109 if (tdescTy.isScattered ()) {
110- auto scatterAttr = llvm::dyn_cast_if_present<xegpu::ScatterTensorDescAttr>(encoding);
110+ auto scatterAttr =
111+ llvm::dyn_cast_if_present<xegpu::ScatterTensorDescAttr>(
112+ encoding);
111113 int64_t chunkSize = scatterAttr.getChunkSize ().getInt ();
112114
113115 if (chunkSize > 1 ) {
@@ -118,7 +120,8 @@ struct TestXeGPUUnrollingPatterns
118120
119121 // To create a new attribute with a different chunk_size:
120122 auto newEncoding = xegpu::ScatterTensorDescAttr::get (
121- ctx, scatterAttr.getMemorySpace ().getValue (), blockedChunkSize);
123+ ctx, scatterAttr.getMemorySpace ().getValue (),
124+ blockedChunkSize);
122125
123126 encoding = newEncoding;
124127 }
You can’t perform that action at this time.
0 commit comments