Історія комітів

Автор SHA1 Опис Дата
  Tri Dao 65f723bb9a Split bwd into more .cu files to speed up compilation 4 місяців тому
  Tri Dao 751c762c9c Don't specialize for hdim 224 to speed up compilation 4 місяців тому
  rocking d8f104e97a Support AMD ROCm on FlashAttention 2 (#1010) 4 місяців тому
  Tri Dao 844912dca0 [CI] Switch from CUDA 12.2 to 12.3 5 місяців тому
  Tri Dao 908511b2b6 Split into more .cu files to speed up compilation 5 місяців тому
  Tri Dao beb2bf2a32 Drop support for pytorch 1.12, 1.13, and python 3.7 5 місяців тому
  Nicolas Patry 8f873cc6ac Implement softcapping. (#1025) 5 місяців тому
  Corey James Levinson beb8b8ba9f add exception to Timeout Error (#963) 6 місяців тому
  Wei Ji 9c0e9ee86d Move packaging and ninja from install_requires to setup_requires (#937) 7 місяців тому
  Tri Dao 2aea958f89 [CI] Compile with torch 2.3.0.dev20240207 8 місяців тому
  Arvind Sundararajan 26c9e82743 Support ARM builds (#757) 9 місяців тому
  Chirag Jain 50896ec574 Make nvcc threads configurable via environment variable (#885) 9 місяців тому
  Qubitium f45bbb4c94 Optimize compile to 1: avoid oom 2: minimize swap usage 3: avoid threads starvation when letting ninja decide how many workers to spawn or manual MAX_JOBS "guesses". Logic is to take the min value of MAX_JOBS auto-calculated by two metrics: 1: cpu cores 2: free memory. This should allow flash-attn to compile close to the most efficient manner under any consumer/server env. (#832) 10 місяців тому
  Tri Dao d4a7c8ffbb [CI] Only compile for CUDA 11.8 & 12.2, MAX_JOBS=2,add torch-nightly 1 рік тому
  Tri Dao 5e525a8dc8 [CI] Use official Pytorch 2.1, add CUDA 11.8 for Pytorch 2.1 1 рік тому
  Tri Dao 1879e089c7 Reduce number of templates for headdim > 128 1 рік тому
  Tri Dao bff3147175 Re-enable compilation for Hopper 1 рік тому
  Tri Dao dfe29f5e2b [Gen] Don't use ft_attention, use flash_attn_with_kvcache instead 1 рік тому
  Federico Berto fa3ddcbaaa [Minor] add nvcc note on bare_metal_version `RuntimeError` (#552) 1 рік тому
  Tri Dao 799f56fa90 Don't compile for Pytorch 2.1 on CUDA 12.1 due to nvcc segfaults 1 рік тому
  Tri Dao bb9beb3645 Remove some unused headers 1 рік тому
  Tri Dao 0c04943fa2 Require CUDA 11.6+, clean up setup.py 1 рік тому
  Tri Dao b1fbbd8337 Implement splitKV attention 1 рік тому
  Tri Dao cbb4cf5f46 Don't need to set TORCH_CUDA_ARCH_LIST in setup.py 1 рік тому
  Aman Gupta Karmani aab603af4f fix binary wheel installation when nvcc is not available (#448) 1 рік тому
  Tri Dao 9c531bdc0a Use single thread compilation for cuda12.1, torch2.1 to avoid OOM CI 1 рік тому
  Tri Dao 2ddeaa406c Fix wheel building 1 рік тому
  Tri Dao 3c458cff77 Merge branch 'feature/demo-wheels' of https://github.com/piercefreeman/flash-attention into piercefreeman-feature/demo-wheels 1 рік тому
  Tri Dao 1c41d2b0e5 Fix race condition in bwd (overwriting sK) 1 рік тому
  Tri Dao 4f285b3547 FlashAttention-2 release 1 рік тому