[Mlir-commits] [mlir] 80246b2 - [mlir][python] fix python build

[Jeff Niu]

Author: Jeff Niu
Date: 2023-01-12T14:17:59-08:00
New Revision: 80246b22a62140fd30131bee571d7fd84d2e2522

URL: https://github.com/llvm/llvm-project/commit/80246b22a62140fd30131bee571d7fd84d2e2522
DIFF: https://github.com/llvm/llvm-project/commit/80246b22a62140fd30131bee571d7fd84d2e2522.diff

LOG: [mlir][python] fix python build

Added: 
    

Modified: 
    mlir/python/mlir/dialects/_memref_ops_ext.py

Removed: 
    


################################################################################
diff  --git a/mlir/python/mlir/dialects/_memref_ops_ext.py b/mlir/python/mlir/dialects/_memref_ops_ext.py
index 9cc22a21c6283..a00a087be79b2 100644
--- a/mlir/python/mlir/dialects/_memref_ops_ext.py
+++ b/mlir/python/mlir/dialects/_memref_ops_ext.py
@@ -4,7 +4,8 @@
 
 try:
   from ..ir import *
-  from ._ods_common import get_op_result_or_value as _get_op_result_or_value, get_op_results_or_values as _get_op_results_or_values
+  from ._ods_common import get_op_result_or_value as _get_op_result_or_value
+  from ._ods_common import get_op_results_or_values as _get_op_results_or_values
 except ImportError as e:
   raise RuntimeError("Error loading imports from extension module") from e
 
@@ -30,8 +31,6 @@ def __init__(self,
       loc: user-visible location of the operation.
       ip: insertion point.
     """
-    memref_resolved = _get_op_result_or_value(memref)
     indices_resolved = [] if indices is None else _get_op_results_or_values(
         indices)
-    return_type = MemRefType(memref_resolved.type).element_type
-    super().__init__(return_type, memref, indices_resolved, loc=loc, ip=ip)
+    super().__init__(memref, indices_resolved, loc=loc, ip=ip)