fix: align kernel source with latest reference source c743a32
drbh commited on
How to use kernels-community/flash-attn2 with Kernels:
# !pip install kernels
from kernels import get_kernel
kernel = get_kernel("kernels-community/flash-attn2")