[Mlir-commits] [mlir] [mlir][core|ptr] Add `PtrLikeTypeInterface` and casting ops to the `ptr` dialect (PR #137469)
Fabian Mora
llvmlistbot at llvm.org
Mon Jun 16 12:22:03 PDT 2025
================
@@ -41,6 +41,54 @@ void PtrDialect::initialize() {
>();
}
+//===----------------------------------------------------------------------===//
+// FromPtrOp
+//===----------------------------------------------------------------------===//
+
+OpFoldResult FromPtrOp::fold(FoldAdaptor adaptor) {
+ // Fold the pattern:
+ // %ptr = ptr.to_ptr %v : type -> ptr
+ // (%mda = ptr.get_metadata %v : type)?
+ // %val = ptr.from_ptr %ptr (metadata %mda)? : ptr -> type
+ // To:
+ // %val -> %v
+ auto toPtr = dyn_cast_or_null<ToPtrOp>(getPtr().getDefiningOp());
+ // Cannot fold if it's not a `to_ptr` op or the initial and final types are
+ // different.
+ if (!toPtr || toPtr.getPtr().getType() != getType())
+ return nullptr;
+ Value md = getMetadata();
+ if (!md)
+ return toPtr.getPtr();
----------------
fabianmcg wrote:
Apologies, missed this comment revision. Fixed, I updated the condition to:
```C++
// If the type has trivial metadata fold.
if (!fromPtr.getType().hasPtrMetadata()) {
ptrLike = toPtr.getPtr();
} else if (md) {
// Fold if the metadata can be verified to be equal.
if (auto mdOp = dyn_cast_or_null<GetMetadataOp>(md.getDefiningOp());
mdOp && mdOp.getPtr() == toPtr.getPtr())
ptrLike = toPtr.getPtr();
}
```
to reflect that information loss cannot be folded.
https://github.com/llvm/llvm-project/pull/137469
More information about the Mlir-commits
mailing list