How to use kernels-community/flash-attn2 with Kernels:
# !pip install kernels from kernels import get_kernel kernel = get_kernel("kernels-community/flash-attn2")
I am aware of the vllm build but wanted to just know if this collection of builds includes the hopper one for FA3.
vllm
hopper
· Sign up or log in to comment