windows
num_warps=num_warps, num_stages=4) https://github.com/kyegomez/FlashAttention20Triton
查看全文