add assert to builder

Author: hughperkins

taichi/ir/ir_builder.cpp

@@ -129,6 +129,12 @@ ContinueStmt *IRBuilder::create_continue() {
   return insert(Stmt::make_typed<ContinueStmt>());
 }
 
+void IRBuilder::create_assert(Stmt *cond, const std::string &msg) {
+  std::vector<Stmt *> empty_args;
+  auto assert_stmt = Stmt::make_typed<AssertStmt>(cond, msg, empty_args);
+  insert(std::move(assert_stmt));
+}
+
 FuncCallStmt *IRBuilder::create_func_call(Function *func,
                                           const std::vector<Stmt *> &args) {
   return insert(Stmt::make_typed<FuncCallStmt>(func, args));

taichi/ir/ir_builder.h

@@ -121,6 +121,7 @@ class IRBuilder {
   IfStmt *create_if(Stmt *cond);
   WhileControlStmt *create_break();
   ContinueStmt *create_continue();
+  void create_assert(Stmt *cond, const std::string &msg);
 
   // Function.
   FuncCallStmt *create_func_call(Function *func,

tests/cpp/ir/ir_builder_test.cpp

@@ -29,6 +29,22 @@ TEST(IRBuilder, Bool) {
 )");
 }
 
+TEST(IRBuilder, Assert) {
+  IRBuilder builder;
+  auto *bool_true = builder.get_bool(true);
+  builder.create_assert(bool_true, "assertion failed");
+  auto block = builder.extract_ir();
+
+  auto print = irpass::make_pass_printer(true, true, "", bool_true);
+  std::string ir_string;
+  irpass::print(block->get_ir_root(), &ir_string);
+  EXPECT_STREQ(ir_string.c_str(),
+               "kernel {\n"
+               "  <u1> $0 = const true\n"
+               "  1 : assert $0, \"assertion failed\"\n"
+               "}\n");
+}
+
 TEST(IRBuilder, Basic) {
   IRBuilder builder;
   auto *lhs = builder.get_int32(40);