diff --git a/python/tvm/expr.py b/python/tvm/expr.py index c3b0845aecb6..8ae1d45d8b5e 100644 --- a/python/tvm/expr.py +++ b/python/tvm/expr.py @@ -37,6 +37,24 @@ def __rtruediv__(self, other): def __neg__(self): return self.__mul__(-1) + def __lt__(self, other): + return _make.LT(self, other) + + def __le__(self, other): + return _make.LE(self, other) + + def __eq__(self, other): + return _make.EQ(self, other) + + def __ne__(self, other): + return _make.NE(self, other) + + def __gt__(self, other): + return _make.GT(self, other) + + def __ge__(self, other): + return _make.GE(self, other) + class Expr(NodeBase, ExprOp): pass