fix i386 fp16 cases

Author: wrongtest-intellif

tests/python/unittest/test_tir_imm_values.py

@@ -209,7 +209,7 @@ def check_tir_const_fold(
     if isinstance(x_range, (int, float)):
         x = x_range
     elif dtype.startswith("int") or dtype.startswith("uint"):
-        x = np.random.randint(x_range[0], x_range[1] + 1)
+        x = np.random.randint(x_range[0], x_range[1] + 1, dtype=dtype)
     else:
         x = np.random.uniform(x_range[0], x_range[1])
 
@@ -218,7 +218,7 @@ def check_tir_const_fold(
     if isinstance(y_range, (int, float)):
         y = y_range
     elif dtype.startswith("int") or dtype.startswith("uint"):
-        y = np.random.randint(y_range[0], y_range[1] + 1)
+        y = np.random.randint(y_range[0], y_range[1] + 1, dtype=dtype)
     else:
         y = np.random.uniform(y_range[0], y_range[1])
 
@@ -249,25 +249,35 @@ def test_tir_floatimm_const_fold():
     """Behavior check: folding fp32 match platform f32 arithmetic"""
 
     @T.prim_func
-    def float_imm_multiply(x: T.float32, y: T.float32) -> T.float32:
-        T.evaluate(T.ret(x * y, dtype="float32"))
+    def float_imm_multiply(x: T.float32, y: T.float32, z: T.Buffer[(), "float32"]):
+        z[()] = x * y
 
     @T.prim_func
-    def float_imm_add(x: T.float32, y: T.float32) -> T.float32:
-        T.evaluate(T.ret(x + y, dtype="float32"))
+    def float_imm_add(x: T.float32, y: T.float32, z: T.Buffer[(), "float32"]):
+        z[()] = x + y
 
     @T.prim_func
-    def float_imm_sub(x: T.float32, y: T.float32) -> T.float32:
-        T.evaluate(T.ret(x - y, dtype="float32"))
+    def float_imm_sub(x: T.float32, y: T.float32, z: T.Buffer[(), "float32"]):
+        z[()] = x - y
 
     @T.prim_func
-    def float_imm_div(x: T.float32, y: T.float32) -> T.float32:
-        T.evaluate(T.ret(x / y, dtype="float32"))
+    def float_imm_div(x: T.float32, y: T.float32, z: T.Buffer[(), "float32"]):
+        z[()] = x / y
 
-    fmul = tvm.build(float_imm_multiply, target="llvm")
-    fadd = tvm.build(float_imm_add, target="llvm")
-    fsub = tvm.build(float_imm_sub, target="llvm")
-    fdiv = tvm.build(float_imm_div, target="llvm")
+    def __wrap_build(f):
+        lib = tvm.build(f, target="llvm")
+        z = tvm.nd.array(np.zeros([]).astype("float32"))
+
+        def _func(x, y):
+            lib(x, y, z)
+            return z.numpy()
+
+        return _func
+
+    fmul = __wrap_build(float_imm_multiply)
+    fadd = __wrap_build(float_imm_add)
+    fsub = __wrap_build(float_imm_sub)
+    fdiv = __wrap_build(float_imm_div)
 
     # overflow
     check_tir_const_fold("float32", lambda x, y: x * y, fmul, 3.0e30, 3.0e30, np.inf)