[Mlir-commits] [mlir] [mlir][python] fix memref._is_constant_int_like (PR #89447)

[Maksim Levental]

https://github.com/makslevental created https://github.com/llvm/llvm-project/pull/89447

None

>From c6dd3656c55737e802dfcbf30fbd1acc72d65be8 Mon Sep 17 00:00:00 2001
From: max <maksim.levental at gmail.com>
Date: Fri, 19 Apr 2024 14:58:55 -0500
Subject: [PATCH] [mlir][python] fix memref._is_constant_int_like

---
 mlir/python/mlir/dialects/memref.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mlir/python/mlir/dialects/memref.py b/mlir/python/mlir/dialects/memref.py
index a3d783415855e1..bc9a3a52728ad4 100644
--- a/mlir/python/mlir/dialects/memref.py
+++ b/mlir/python/mlir/dialects/memref.py
@@ -8,12 +8,13 @@
 from ._memref_ops_gen import *
 from ._ods_common import _dispatch_mixed_values, MixedValues
 from .arith import ConstantOp, _is_integer_like_type
-from ..ir import Value, MemRefType, StridedLayoutAttr, ShapedType
+from ..ir import Value, MemRefType, StridedLayoutAttr, ShapedType, Operation
 
 
 def _is_constant_int_like(i):
     return (
         isinstance(i, Value)
+        and isinstance(i.owner, Operation)
         and isinstance(i.owner.opview, ConstantOp)
         and _is_integer_like_type(i.type)
     )