diff options
Diffstat (limited to 'candle-kernels/build.rs')
| -rw-r--r-- | candle-kernels/build.rs | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/candle-kernels/build.rs b/candle-kernels/build.rs index 37cf28ae..1c16e17c 100644 --- a/candle-kernels/build.rs +++ b/candle-kernels/build.rs @@ -83,6 +83,14 @@ mod cuda { .map(|p| p.unwrap()) .collect(); + for out_path in glob::glob(&format!("{out_dir}/**/*.ptx")).unwrap() { + std::fs::remove_file(out_path.unwrap()).unwrap(); + } + + for path in &kernel_paths { + println!("cargo:rerun-if-changed={}", path.display()); + } + for path in &mut include_directories { println!("cargo:rerun-if-changed={}", path.display()); let destination = |