Hugging Face's logo

kernels-community
/
vllm-flash-attn3

kernel
Model card Files
xet
Community
8
vllm-flash-attn3 / flash-attn /instantiations /flash_bwd_hdim128_bf16_softcapall_sm90.cu
danieldk's picture
danieldk HF Staff
Convert FA3 to Kernel Hub format
4d1b54e
raw
history blame
348 Bytes
 // Copyright (c) 2024, Jay Shah, Ganesh Bikshandi, Ying Zhang, Vijay Thakkar, Pradeep Ramani, Tri Dao.
// Splitting the different template instantiations to different files to speed up compilation.
// This file is auto-generated. See "generate_kernels.py"
#include "flash_bwd_hdim128_bf16_sm90.cu"
#include "flash_bwd_hdim128_bf16_softcap_sm90.cu"