More progress on WhileOp

2022-04-05 17:55:43 +08:00
parent d7fbddc7d4
commit 39fad2b18a
4 changed files with 33 additions and 9 deletions
--- a/rewrite-test/jit/vecadd-loop.py
+++ b/rewrite-test/jit/vecadd-loop.py
@@ -47,6 +47,5 @@ z = torch.empty_like(x)
 # add_kernel[(1,)](x, y, z, size, 256)
 # print(add_kernel[(1,)].kernel.compile_to_ttir())
 mod, ctx = add_kernel.compile_to_ttir(x, y, z, size, 128, 8, grid=(1,))
-mod.get_context()
 mod.dump()
 # print(mod)