Fix flash-attn build. (#2754)

This commit is contained in:
Laurent Mazare
2025-01-28 18:49:46 +01:00
committed by GitHub
parent 8f20f2a722
commit 2a2852d1c1

View File

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