Skip to content

Commit beb5c8e

Browse files
committed
minor change
1 parent 915830c commit beb5c8e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mlir/test/lib/Dialect/XeGPU/TestXeGPUTransforms.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ 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 = tdescTy.getEncodingAsScatterTensorDescAttr();
110+
auto scatterAttr = llvm::dyn_cast_if_present<xegpu::ScatterTensorDescAttr>(encoding);
111111
int64_t chunkSize = scatterAttr.getChunkSize().getInt();
112112

113113
if (chunkSize > 1) {

0 commit comments

Comments
 (0)