[Mlir-commits] [mlir] [mlir][xegpu] Relax rank restriction of TensorDescType (PR #145916)

Charitha Saumya llvmlistbot at llvm.org
Thu Jun 26 14:59:32 PDT 2025


================
@@ -303,9 +303,7 @@ void XeGPUBlockingPass::runOnOperation() {
       // If the encoding is a ScatterTensorDescAttr, we need to
       // potentially adjust the chunk size based on the inst_data.
       if (tdescTy.isScattered()) {
-        auto scatterAttr =
-            llvm::dyn_cast_if_present<xegpu::ScatterTensorDescAttr>(encoding);
-        int64_t chunkSize = scatterAttr.getChunkSize().getInt();
+        int64_t chunkSize = tdescTy.getChunkSize();
----------------
charithaintc wrote:

nit: safer to not directly expose getChunkSize to tensor_desc 

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


More information about the Mlir-commits mailing list