.

2023-01-09 20:14:06 -08:00
parent bae4c40379
commit d88353a5a4
1 changed files with 2 additions and 2 deletions
--- a/python/tutorials/06-fused-attention.py
+++ b/python/tutorials/06-fused-attention.py
@@ -288,8 +288,8 @@ class _attention(torch.autograd.Function):
            BLOCK_DMODEL=ctx.BLOCK_DMODEL, num_warps=8,
            num_stages=1,
        )
-        print(pgm.asm["ttgir"])
-        exit()
+        # print(pgm.asm["ttgir"])
+        # exit()
        return dq, dk, dv, None