sigmoid-neuron
/

flash-attention-1-triton

Model card Files Files and versions

flash-attention-1-triton / flash_attention_kernel

Commit History

feat: add support for ROCm backend and update device validation for HIP and XPU compatibility

a391959

sigmoid-neuron commited on Apr 9

feat: implement Flash Attention 1 using Triton with PyTorch C++ bindings and test suite

8cf888e

sigmoid-neuron commited on Apr 9