Add flash-attn - #26239

Merged

carterbox merged 23 commits into

conda-forge:mainfrom

weiji14:add-flash-attn

May 7, 2024

Commits on May 4, 2024

Add flash-attn
weiji14
committed
07ec11e
Add cuda compiler
weiji14
committed
a4def75
Add c and cxx compilers
weiji14
committed
06713e3
Add stdlib c
weiji14
committed
51d7d74
Skip build on non-cuda platforms
weiji14
committed
aa17a2c
Add libcublas-dev, libcusolver-dev, libcusparse-dev to host deps
weiji14
committed
4d8b37c
Remove noarch: python
weiji14
committed
04a346a
Remove minimum pin on python version
weiji14
committed
16414ff
BLD: Fixup build environment and variables
carterbox
committed
2d2212d

Commits on May 5, 2024

Sort dependencies alphabetically
weiji14
committed
2cde3c1

Commits on May 6, 2024

Set azure timeout_minutes to 360 in conda-forge.yml
weiji14
committed
501aa9d
Set TORCH_CUDA_ARCH_LIST to 8.0 and above
weiji14
committed
a1b1faa
Drop quotes from `script_env`
weiji14
committed
5235314
DEV: Reduce archs and jobs
carterbox
authored
ef03f90
Add missing host deps and rpaths
carterbox
authored
ebae578
Merge remote-tracking branch 'forge/main' into add-flash-attn
carterbox
committed
460eeb2
BLD: Replace setup script with simpler one
carterbox
committed
317646a
Update recipes/flash-attn/meta.yaml

carterbox
and
weiji14
authored
0b81f6f
Add license for CUTLASS
weiji14
committed
96e817a
Add libcublas-dev, libcusolver-dev, libcusparse-dev to host deps again
weiji14
committed
0733767

Commits on May 7, 2024

ignore_run_exports_from libcublas-dev, libcusolver-dev, libcusparse-dev
weiji14
committed
37e676a
Temporarily set TORCH_CUDA_ARCH_LIST=8.6+PTX and MAX_JOBS=1
weiji14
committed
fc2fc76
BLD: Lower CUDA arch target to 8.0
carterbox
authored
63dcb65