gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
Phil Tillet
2023-01-09 20:14:06 -08:00
parent bae4c40379
commit d88353a5a4
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/tutorials/06-fused-attention.py
View File

@@ -288,8 +288,8 @@ class _attention(torch.autograd.Function):
BLOCK_DMODEL=ctx.BLOCK_DMODEL, num_warps=8, BLOCK_DMODEL=ctx.BLOCK_DMODEL, num_warps=8,
num_stages=1, num_stages=1,
) )
print(pgm.asm["ttgir"]) # print(pgm.asm["ttgir"])
exit() # exit()
return dq, dk, dv, None return dq, dk, dv, None