[TVMScript][Fix] Fix bool printing for roundtrip

src/script/printer/tir/ir.cc

@@ -32,7 +32,8 @@ TVM_STATIC_IR_FUNCTOR(IRDocsifier, vtable)
       if (dtype == d->cfg->int_dtype) {
         return LiteralDoc::Int(imm->value, imm_p->Attr("value"));
       } else if (dtype == DataType::Bool()) {
-        return LiteralDoc::Boolean(imm->value, imm_p->Attr("value"));
+        return TIR(d, DType2Str(dtype))
+            ->Call({LiteralDoc::Boolean(imm->value, imm_p->Attr("value"))});
       } else {
         return TIR(d, DType2Str(dtype))->Call({LiteralDoc::Int(imm->value, imm_p->Attr("value"))});
       }

tests/python/unittest/test_tir_nodes.py

@@ -314,7 +314,7 @@ def test_isnan():
     y = te.var("y", "float16")
     assert str(tvm.tir.isnan(y)) == 'T.isnan(T.Cast("float32", y))'
     z = te.var("z", "int32")
-    assert str(tvm.tir.isnan(z)) == "False"
+    assert str(tvm.tir.isnan(z)) == "T.bool(False)"
     k = te.var("k", "int8x2")
     assert str(tvm.tir.isnan(k).dtype) == "uint1x2"
 

tests/python/unittest/test_tir_schedule_compute_inline.py

@@ -1154,7 +1154,7 @@ def test_reverse_compute_inline_producer_predicate_disallowed():
         sch.reverse_compute_inline(sch.get_block("compute_4"))
 
     assert (
-        "that cannot be implied by the synthesized predicate True of the new inlined block"
+        "that cannot be implied by the synthesized predicate T.bool(True) of the new inlined block"
         in str(e)
     )
 

tests/python/unittest/test_tvmscript_printer_tir.py

@@ -606,7 +606,7 @@ def test_select():
     obj = tir.Select(True, 0, 2)
     _assert_print(
         obj,
-        """T.Select(True, 0, 2)
+        """T.Select(T.bool(True), 0, 2)
 """,
     )
 

tests/python/unittest/test_tvmscript_roundtrip.py

@@ -3444,7 +3444,9 @@ def func() -> None:
 def bool_cast():
     @T.prim_func
     def func() -> None:
+        a = T.bool()
         T.evaluate(T.bool(T.int32(0)))
+        T.evaluate(a == T.bool(False))
 
     return func