Update Cuda Blackwell

[johnnynunez]

Jimver/cuda-toolkit#377

[alisson-anjos]

@johnnynunez were you able to compile this locally?

[johnnynunez]

@johnnynunez were you able to compile this locally?

Jetson

[johnnynunez]

@johnnynunez were you able to compile this locally?

Jetson

https://github.com/nektos/act

[wangxianggang1997]

@johnnynunez were you able to compile this locally?

Jetson

May I ask if you can compile flash attention on the Thor board based on the Blackwell architecture? When I encountered a paragraph error during compilation, I don't know how to solve it. Can you provide me with some ideas? Looking forward to your reply very much. Thank you very much

[johnnynunez]

yo have to add codegen here:

flash-attention/setup.py

Line 188 in 7ac28b4

if bare_metal_version >= Version("12.8") and "120" in cuda_archs():