Spaces:
Runtime error
Runtime error
Update Feather H200 training runtime image
Browse files
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());
|