[Mlir-commits] [mlir] [MLIR] Fix issues with XeGPU to XeVM pass. (PR #155946)
Charitha Saumya
llvmlistbot at llvm.org
Thu Aug 28 16:37:42 PDT 2025
================
@@ -259,23 +259,23 @@ class UpdateNdOffsetToXeVMPattern
// Only 2D offsets are supported for now.
if (mixedOffsets.size() != 2)
return rewriter.notifyMatchFailure(op, "Expected 2D offsets.");
- auto tdesc = adaptor.getTensorDesc();
+ auto payload = adaptor.getTensorDesc();
// Utility for updating payload offset values from op fold result.
auto updateOffset = [&](unsigned idx, int payloadPos) -> Value {
Value offset =
getValueOrCreateConstantIntOp(rewriter, loc, mixedOffsets[idx]);
offset = getValueOrCreateCastToIndexLike(rewriter, loc,
rewriter.getI32Type(), offset);
Value oldOffset =
- vector::ExtractOp::create(rewriter, loc, tdesc, payloadPos);
+ vector::ExtractOp::create(rewriter, loc, payload, payloadPos);
Value newOffset = arith::AddIOp::create(rewriter, loc, oldOffset, offset);
- return vector::InsertOp::create(rewriter, loc, newOffset, tdesc,
+ return vector::InsertOp::create(rewriter, loc, newOffset, payload,
payloadPos);
};
// Update offsets in the payload.
- auto val = updateOffset(0, static_cast<int>(NdTdescOffset::TensorOffsetH));
- val = updateOffset(1, static_cast<int>(NdTdescOffset::TensorOffsetW));
- rewriter.replaceOp(op, val);
+ payload = updateOffset(0, static_cast<int>(NdTdescOffset::TensorOffsetH));
----------------
charithaintc wrote:
what diff removing `auto` makes here?
https://github.com/llvm/llvm-project/pull/155946
More information about the Mlir-commits
mailing list