[Mlir-commits] [mlir] [MLIR][XeGPU] Extend unrolling support for scatter ops with chunk_size (PR #144447)

[Charitha Saumya]

================
@@ -61,6 +67,16 @@ struct TestXeGPUUnrollingPatterns
               tdescTy = loadNdOp.getTensorDescType();
             } else if (auto storeNdOp = dyn_cast<xegpu::StoreNdOp>(op)) {
               tdescTy = storeNdOp.getTensorDescType();
+            } else if (auto createOp = dyn_cast<xegpu::CreateDescOp>(op)) {
----------------
charithaintc wrote:

nit: this can be improved using `TypeSwitch`.

Check example here:
https://github.com/llvm/llvm-project/blob/main/mlir/lib/Conversion/VectorToGPU/VectorToGPU.cpp#L1282

https://github.com/llvm/llvm-project/pull/144447