[Mlir-commits] [mlir] 73a2fd4 - [mlir][xegpu] Patch dynamic descriptor creation (#93580)

[llvmlistbot at llvm.org]

Author: Artem Kroviakov
Date: 2024-06-05T14:39:05-05:00
New Revision: 73a2fd470c58eb10e2e51466a053dd30b63bbe7a

URL: https://github.com/llvm/llvm-project/commit/73a2fd470c58eb10e2e51466a053dd30b63bbe7a
DIFF: https://github.com/llvm/llvm-project/commit/73a2fd470c58eb10e2e51466a053dd30b63bbe7a.diff

LOG: [mlir][xegpu] Patch dynamic descriptor creation (#93580)

fixes the bug in XeGPU's `CreateNdDescOp` tensor creation with dynamic offset and strides.

Added: 
    

Modified: 
    mlir/lib/Dialect/XeGPU/IR/XeGPUOps.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/XeGPU/IR/XeGPUOps.cpp b/mlir/lib/Dialect/XeGPU/IR/XeGPUOps.cpp
index 22959224d56c2..5ef47fbbe1ce0 100644
--- a/mlir/lib/Dialect/XeGPU/IR/XeGPUOps.cpp
+++ b/mlir/lib/Dialect/XeGPU/IR/XeGPUOps.cpp
@@ -110,7 +110,7 @@ void CreateNdDescOp::build(OpBuilder &builder, OperationState &state,
 
   dispatchIndexOpFoldResults(offsets, dynamicOffsets, staticOffsets);
   dispatchIndexOpFoldResults(shape, dynamicShape, staticShape);
-  dispatchIndexOpFoldResults(strides, dynamicStrides, staticOffsets);
+  dispatchIndexOpFoldResults(strides, dynamicStrides, staticStrides);
 
   auto staticOffsetsAttr = builder.getDenseI64ArrayAttr(staticOffsets);
   auto staticShapeAttr = builder.getDenseI64ArrayAttr(staticShape);