[TIR] Add conversion from FloatImm to float in Python (apache#9009)

This method matches the IntImm method for converting from IntImm to int.
ylc · Jan 13, 2022 · 7907f45 · 7907f45
1 parent 04a0330
commit 7907f45
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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):