Jackoatmon commited on
Commit
3fa82de
·
verified ·
1 Parent(s): 2a5cb1f

Update Feather H200 training runtime image

Browse files
Files changed (1) hide show
  1. overlay/htm_rust/build.rs +4 -4
overlay/htm_rust/build.rs CHANGED
@@ -44,10 +44,10 @@ fn main() {
44
  ];
45
 
46
  let kernels_dir = PathBuf::from("src/gpu/kernels");
47
- for k in kernels {
48
- let src = kernels_dir.join(format!("{k}.cu"));
49
- println!("cargo:rerun-if-changed={}", src.display());
50
- }
51
 
52
  let out_dir = PathBuf::from(env::var("OUT_DIR").expect("OUT_DIR"));
53
  let arch = env::var("HTM_CUDA_ARCH").unwrap_or_else(|_| "sm_90a".into());
 
44
  ];
45
 
46
  let kernels_dir = PathBuf::from("src/gpu/kernels");
47
+ for k in &kernels {
48
+ let src = kernels_dir.join(format!("{k}.cu"));
49
+ println!("cargo:rerun-if-changed={}", src.display());
50
+ }
51
 
52
  let out_dir = PathBuf::from(env::var("OUT_DIR").expect("OUT_DIR"));
53
  let arch = env::var("HTM_CUDA_ARCH").unwrap_or_else(|_| "sm_90a".into());