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

Merge pull request #40 from LaurentMazare/fix_kernel_cache

Browse Source 
Fixing kernel cache (a bit brutal for now, but if build triggers, rebuild ALL kernels).
This commit is contained in:
Nicolas Patry
2023-06-29 18:02:06 +02:00
committed by GitHub
parent 889f7e0971 1ea08a19cb
commit 4b148b5414
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
candle-kernels/build.rs
View File

@ -83,6 +83,15 @@ mod cuda {
.map(|p| p.unwrap()) .map(|p| p.unwrap())
.collect(); .collect();
for out_path in glob::glob(&format!("{out_dir}/**/*.ptx")).unwrap() {
std::fs::remove_file(out_path.unwrap()).unwrap();
}
println!("cargo:rerun-if-changed=src/");
for path in &kernel_paths {
println!("cargo:rerun-if-changed={}", path.display());
}
for path in &mut include_directories { for path in &mut include_directories {
println!("cargo:rerun-if-changed={}", path.display()); println!("cargo:rerun-if-changed={}", path.display());
let destination = let destination =