Kernels
flash-attn2 / torch-ext /flash_attn

Commit History

  1. fix: refactor bindings and update py library for correctness

    drbh
    876ac68

  1. feat improve readme and library code

    drbh
    d6cc1b0

  1. feat: pass vars into fwd and include build

    drbh
    39b4aba
  2. feat: include source and enable build

    drbh
    a7165c8