bindings for ModuleOp

2022-03-30 13:32:52 +08:00
parent 38e67b4293
commit e95d98a886
4 changed files with 29 additions and 8 deletions
--- a/rewrite-test/test_ir.py
+++ b/rewrite-test/test_ir.py
@@ -6,7 +6,7 @@ ctx.load_triton()
 # TODO
 builder = ir.builder(ctx)

-# module = builder.create_module()
+module = builder.create_module()


 i1_ty = builder.get_int1_ty()
@@ -49,7 +49,7 @@ c_ptrs = builder.create_broadcast(entry.arg(2), [128])
 c_ptrs = builder.create_gep(c_ptrs, offsets)
 builder.create_store(c_ptrs, c)

-func.dump()
+# func.dump()

-# module.push_back(func)
-# module.dump()
+module.push_back(func)
+module.dump()