[PyTorch] Restore FlashAttention 2 head dim support on sm103 - #3360
Open
kalectory wants to merge 1 commit into
Open
[PyTorch] Restore FlashAttention 2 head dim support on sm103#3360kalectory wants to merge 1 commit into
kalectory wants to merge 1 commit into