huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 02:38:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
cfcbec9fc70aca2b0e08f382dec8634f88b61bce
candle/candle-flash-attn
History
Zsombor cfcbec9fc7 Add small customization to the build (#768) 
* Add ability to override the compiler used by NVCC from an environment variable

* Allow relative paths in CANDLE_FLASH_ATTN_BUILD_DIR

* Add the compilation failure to the readme, with a possible solution

* Adjust the error message, and remove the special handling of the relative paths
2023-09-08 08:15:14 +01:00
..
cutlass @ c4f6b8c6bc
Add flash attention (#241)
2023-07-26 07:48:10 +01:00
kernels
Add back the bf16 flash-attn kernels. (#730)
2023-09-04 07:50:52 +01:00
src
Properly set the is_bf16 flag. (#738)
2023-09-04 16:45:26 +01:00
tests
Flash attention without padding (varlen). (#281)
2023-07-31 09:45:39 +01:00
build.rs
Add small customization to the build (#768)
2023-09-08 08:15:14 +01:00
Cargo.toml
Add some documentation. (#673)
2023-08-30 11:54:00 +01:00
README.md
Add some missing readme files. (#304)
2023-08-02 10:57:12 +01:00

README.md

candle-flash-attn