[Mlir-commits] [mlir] [MLIR][XeGPU] make offsets optional for create_nd_tdesc (PR #148335)
Adam Siemieniuk
llvmlistbot at llvm.org
Wed Jul 16 09:45:58 PDT 2025
================
@@ -163,7 +174,26 @@ def XeGPU_CreateNdDescOp: XeGPU_Op<"create_nd_tdesc", [Pure, ViewLikeOpInterface
}
ArrayRef<int64_t> getStaticOffsets(){
- return getConstOffsets();
+ auto attr = getConstOffsetsAttr();
+
+ if (attr)
+ return attr;
+
+ auto memrefType = llvm::dyn_cast<MemRefType>(getSourceType());
+ int rank = 0;
+ if (memrefType)
+ //use source memref's rank, as source memref rank may be higher
+ rank = memrefType.getRank();
+ else
----------------
adam-smnk wrote:
nit: use braces as comments make it a multi-line body (if you need to keep these blocks at all)
https://github.com/llvm/llvm-project/pull/148335
More information about the Mlir-commits
mailing list