From 80246b22a62140fd30131bee571d7fd84d2e2522 Mon Sep 17 00:00:00 2001 From: Jeff Niu Date: Thu, 12 Jan 2023 14:17:32 -0800 Subject: [PATCH] [mlir][python] fix python build --- mlir/python/mlir/dialects/_memref_ops_ext.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/mlir/python/mlir/dialects/_memref_ops_ext.py b/mlir/python/mlir/dialects/_memref_ops_ext.py index 9cc22a2..a00a087 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 @@ class LoadOp: 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) -- 2.7.4