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

Moving to a proper build crate bindgen_cuda. (#1531)

Browse Source 
* Moving to a proper build crate `bindgen_cuda`.

* Fmt.
This commit is contained in:
Nicolas Patry
2024-01-07 12:29:24 +01:00
committed by GitHub
parent e72d52b1a2
commit 30313c3081
4 changed files with 41 additions and 483 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -15,9 +15,9 @@ candle = { path = "../candle-core", features = ["cuda"], package = "candle-core"
half = { version = "2.3.1", features = ["num-traits"] }
[build-dependencies]
bindgen_cuda = "0.1.1"
anyhow = { version = "1", features = ["backtrace"] }
num_cpus = "1.15.0"
rayon = "1.7.0"
[dev-dependencies]
anyhow = { version = "1", features = ["backtrace"] }