diff --git a/python/tvm/ir/expr.py b/python/tvm/ir/expr.py index 5a83d8e5d92e..f0f2245e7f62 100644 --- a/python/tvm/ir/expr.py +++ b/python/tvm/ir/expr.py @@ -177,3 +177,9 @@ def from_min_extent(min_value, extent, span=None): The constructed range. """ return _ffi_api.Range_from_min_extent(min_value, extent, span) + + def __eq__(self, other): + return tvm.ir.structural_equal(self, other) + + def __ne__(self, other): + return not self.__eq__(other)