huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix flash-attn build. (#2754)

Browse Source
This commit is contained in:
Laurent Mazare
2025-01-28 18:49:46 +01:00
committed by GitHub
parent 8f20f2a722
commit 2a2852d1c1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-flash-attn/build.rs
View File

@ -73,7 +73,7 @@ fn main() -> Result<()> {
}; };
let kernels = KERNEL_FILES.iter().collect(); let kernels = KERNEL_FILES.iter().collect();
let builder = bindgen_cuda::Builder::default() let mut builder = bindgen_cuda::Builder::default()
.kernel_paths(kernels) .kernel_paths(kernels)
.out_dir(build_dir.clone()) .out_dir(build_dir.clone())
.arg("-std=c++17") .arg("-std=c++17")