diff --git a/python/tvm/tir/expr.py b/python/tvm/tir/expr.py index 4ba8c5471b5d..2bfa0aacb184 100644 --- a/python/tvm/tir/expr.py +++ b/python/tvm/tir/expr.py @@ -523,6 +523,9 @@ def __init__(self, dtype, value, span=None): tvm.ir._ffi_api.FloatImm, dtype, value, span # type: ignore ) + def __float__(self): + return self.value + @tvm._ffi.register_object class IntImm(ConstExpr):