Hugging Face's logo

kernels-community
/
flash-attn2

Kernels
Model card Files
xet
 Community
5
flash-attn2 / torch-ext /torch_binding.cpp

Commit History

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

    drbh
    876ac68

  1. fix: simplify and align signatures further

    drbh
    3099d65

  1. fix: revise bindings and wrapper typing

    drbh
    ab4cc6a

  1. feat: add remaining torch binding def and impls

    drbh
    eda872e

  1. fix: expand build combinations and include all files

    drbh
    b0d3c12

  1. feat: include source and enable build

    drbh
    a7165c8