[Mlir-commits] [mlir] 5ff486d - [NFC][AMDGPU] Use getMixedSize in FatRawBufferCastOp dim reification (#174548)

[llvmlistbot at llvm.org]

Author: Jorn Tuyls
Date: 2026-01-06T08:32:25-05:00
New Revision: 5ff486d08e97b0d63f9799381d407e69ec5b1e28

URL: https://github.com/llvm/llvm-project/commit/5ff486d08e97b0d63f9799381d407e69ec5b1e28
DIFF: https://github.com/llvm/llvm-project/commit/5ff486d08e97b0d63f9799381d407e69ec5b1e28.diff

LOG: [NFC][AMDGPU] Use getMixedSize in FatRawBufferCastOp dim reification (#174548)

After https://github.com/llvm/llvm-project/pull/174477, I found similar
logic that can be replaced by `memref::getMixedSize` in the
FatRawBufferCastOp dimension reification function.

Added: 
    

Modified: 
    mlir/lib/Dialect/AMDGPU/IR/AMDGPUDialect.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/AMDGPU/IR/AMDGPUDialect.cpp b/mlir/lib/Dialect/AMDGPU/IR/AMDGPUDialect.cpp
index ccaea75f8c4c1..787248f9f339e 100644
--- a/mlir/lib/Dialect/AMDGPU/IR/AMDGPUDialect.cpp
+++ b/mlir/lib/Dialect/AMDGPU/IR/AMDGPUDialect.cpp
@@ -151,12 +151,7 @@ FailureOr<OpFoldResult> FatRawBufferCastOp::reifyDimOfResult(OpBuilder &builder,
                                                              int resultIndex,
                                                              int dim) {
   assert(resultIndex == 0 && "FatRawBufferCastOp has a single result");
-  Value source = getSource();
-  auto sourceType = cast<MemRefType>(source.getType());
-  if (sourceType.isDynamicDim(dim))
-    return OpFoldResult(
-        builder.createOrFold<memref::DimOp>(getLoc(), source, dim));
-  return OpFoldResult(builder.getIndexAttr(sourceType.getDimSize(dim)));
+  return memref::getMixedSize(builder, getLoc(), getSource(), dim);
 }
 
 LogicalResult FatRawBufferCastOp::verify() {