huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 18:28:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a cudnn feature to candle-nn/candle-transformers. (#2890)

Browse Source
This commit is contained in:
Laurent Mazare
2025-04-13 17:43:41 +02:00
committed by GitHub
parent 2f9606b187
commit fb660b8d43
4 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
candle-flash-attn/Cargo.toml
View File

@ -18,7 +18,10 @@ half = { version = "2.3.1", features = ["num-traits"] }
bindgen_cuda = "0.1.1"
anyhow = { version = "1", features = ["backtrace"] }
[dev-dependencies]
anyhow = { version = "1", features = ["backtrace"] }
candle-nn = { path = "../candle-nn", features = ["cuda"] }
[features]
default = []
cudnn = ["candle/cudnn"]