Kernels
flash-attention-1-triton / flash_attention_kernel

Commit History

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

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

    8cf888e