kernels-community/flash-attn2 ships variants for torch 2.7–2.12. With torch 2.13 installed, get_kernel fails:
Cannot find a build variant for this system in kernels-community/flash-attn2
torch212-cxx11-cu126-x86_64-linux: Torch version (2.12) does not match environment Torch version (2.13)
Could torch 2.13 builds be added?
Hit in TRL CI (huggingface/trl#6741), where attn_implementation="kernels-community/flash-attn2" tests are now xfailed.
kernels-community/flash-attn2ships variants for torch 2.7–2.12. With torch 2.13 installed,get_kernelfails:Could torch 2.13 builds be added?
Hit in TRL CI (huggingface/trl#6741), where
attn_implementation="kernels-community/flash-attn2"tests are now xfailed.