Zhihao Shen 30e1ef0f79 minify torch.torch.int32 to torch.int32 (#1237) hace 2 meses
..
layers 8c20cfef49 [Rotary] Support qkv block layout from GQA hace 3 meses
losses c7f32a8409 [CrossEntropy] Support precomputed LSE hace 3 meses
models 30e1ef0f79 minify torch.torch.int32 to torch.int32 (#1237) hace 2 meses
modules 3f1b4d38e7 Fix: check the type of max_seqlen_k instead of checking max_seqlen twice (#1127) hace 4 meses
ops 8c20cfef49 [Rotary] Support qkv block layout from GQA hace 3 meses
utils 320fb59487 Update citation hace 6 meses
__init__.py 418d677192 Bump to v2.6.3 hace 4 meses
bert_padding.py 30e1ef0f79 minify torch.torch.int32 to torch.int32 (#1237) hace 2 meses
flash_attn_interface.py 83e41b3ca4 Add custom ops for compatibility with PT Compile (#1139) hace 2 meses
flash_attn_triton.py f1a73d0740 Run isort and black on python files hace 1 año
flash_attn_triton_og.py f1a73d0740 Run isort and black on python files hace 1 año
flash_blocksparse_attention.py cdbbe844b1 minor changes to unpad_input test util func hace 2 meses
flash_blocksparse_attn_interface.py f1a73d0740 Run isort and black on python files hace 1 año
fused_softmax.py f1a73d0740 Run isort and black on python files hace 1 año
pyproject.toml 73bd3f3bbb Move pyproject.toml to flash-attn and tests dir to avoid PEP 517 hace 1 año