adding boolean to builder

Author: hughperkins

taichi/ir/ir_builder.cpp

@@ -138,6 +138,12 @@ LoopIndexStmt *IRBuilder::get_loop_index(Stmt *loop, int index) {
   return insert(Stmt::make_typed<LoopIndexStmt>(loop, index));
 }
 
+ConstStmt *IRBuilder::get_bool(bool value) {
+  return insert(Stmt::make_typed<ConstStmt>(TypedConstant(
+      TypeFactory::get_instance().get_primitive_type(PrimitiveTypeID::u1),
+      value)));
+}
+
 ConstStmt *IRBuilder::get_int32(int32 value) {
   return insert(Stmt::make_typed<ConstStmt>(TypedConstant(
       TypeFactory::get_instance().get_primitive_type(PrimitiveTypeID::i32),

taichi/ir/ir_builder.h

@@ -130,6 +130,7 @@ class IRBuilder {
   LoopIndexStmt *get_loop_index(Stmt *loop, int index = 0);
 
   // Constants. TODO: add more types
+  ConstStmt *get_bool(bool value);
   ConstStmt *get_int32(int32 value);
   ConstStmt *get_int64(int64 value);
   ConstStmt *get_uint32(uint32 value);

tests/cpp/ir/ir_builder_test.cpp

@@ -4,12 +4,30 @@
 #include "taichi/ir/statements.h"
 #include "tests/cpp/program/test_program.h"
 #include "tests/cpp/ir/ndarray_kernel.h"
+#include "taichi/ir/transforms.h"
 #ifdef TI_WITH_VULKAN
 #include "taichi/rhi/vulkan/vulkan_loader.h"
 #endif
 
 namespace taichi::lang {
 
+TEST(IRBuilder, Bool) {
+  IRBuilder builder;
+  auto *bool_true = builder.get_bool(true);
+  auto *bool_false = builder.get_bool(false);
+  auto ir = builder.extract_ir();
+  // auto *result = builder.create_print(bool_true, "", "");
+  // auto *print = result->cast<PrintStmt>();
+  auto print = irpass::make_pass_printer(true, true, "", bool_true);
+  // irpass::re_id(ir);
+  std::cout << std::flush;
+  irpass::print(bool_true);
+  std::cout << std::flush;
+  std::cout << std::flush;
+  irpass::print(bool_false);
+  std::cout << std::flush;
+}
+
 TEST(IRBuilder, Basic) {
   IRBuilder builder;
   auto *lhs = builder.get_int32(40);