@bernhardmgruber
10.0 blackwell b100/b200
10.1 blackwell thor
10.1a blackwell digits
12.0 blackwell rtx50
??
I have RTX5090


more references:
pytorch/pytorch#145270 I added to pytorch
Dao-AILab/flash-attention#1436
Originally posted by @johnnynunez in #3166 (comment)