[Mlir-commits] [mlir] [mlir][xegpu] Patch dynamic descriptor creation (PR #93580)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Tue May 28 10:05:35 PDT 2024
llvmbot wrote:
<!--LLVM PR SUMMARY COMMENT-->
@llvm/pr-subscribers-mlir
Author: Artem Kroviakov (akroviakov)
<details>
<summary>Changes</summary>
This brief PR fixes the bug in XeGPU's `CreateNdDescOp` tensor creation with dynamic offset and strides.
---
Full diff: https://github.com/llvm/llvm-project/pull/93580.diff
1 Files Affected:
- (modified) mlir/lib/Dialect/XeGPU/IR/XeGPUOps.cpp (+1-1)
``````````diff
diff --git a/mlir/lib/Dialect/XeGPU/IR/XeGPUOps.cpp b/mlir/lib/Dialect/XeGPU/IR/XeGPUOps.cpp
index 22959224d56c2..37904f55cc23a 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, dynamicOffsets);
auto staticOffsetsAttr = builder.getDenseI64ArrayAttr(staticOffsets);
auto staticShapeAttr = builder.getDenseI64ArrayAttr(staticShape);
``````````
</details>
https://github.com/llvm/llvm-project/pull/93580
More information about the Mlir-commits
mailing list