More conversion patterns

2022-05-04 12:50:02 +08:00
parent 5f08e2fdae
commit 3ad7bee35e
4 changed files with 69 additions and 20 deletions
--- a/rewrite-test/jit/vecadd.py
+++ b/rewrite-test/jit/vecadd.py
@@ -41,5 +41,7 @@ z = torch.empty_like(x)
 # print(add_kernel[(1,)].kernel.compile_to_ttir())
 # print(add_kernel.annotations)
 mod, ctx = add_kernel.compile_to_ttir(x, y, z, size, BLOCK_SIZE=256, grid=(1,))
+assert mod.verify()
+mod.dump()
 mod = add_kernel.compile_ttir_to_llir(mod, ctx)
 mod.dump()