mirror of
https://github.com/huggingface/candle.git
synced 2025-06-15 02:16:37 +00:00
d9904a3baf
* Start updating to cudarc 0.14. * Adapt a couple more things. * And a couple more fixes. * More tweaks. * And a couple more fixes. * Bump the major version number. * Proper module system for the cuda kernels. * Proper ptx loading. * Launch the sort kernel. * Custom op. * Start using the builder pattern. * More builder. * More builder. * Get candle-core to compile. * Get the tests to pass. * Get candle-nn to work too. * Support for custom cuda functions. * cudnn fixes. * Get flash attn to run. * Switch the crate versions to be alpha. * Bump the ug dependency.
12 lines
434 B
Rust
12 lines
434 B
Rust
fn main() {
|
|
println!("cargo:rerun-if-changed=build.rs");
|
|
println!("cargo:rerun-if-changed=src/compatibility.cuh");
|
|
println!("cargo:rerun-if-changed=src/cuda_utils.cuh");
|
|
println!("cargo:rerun-if-changed=src/binary_op_macros.cuh");
|
|
|
|
let builder = bindgen_cuda::Builder::default();
|
|
println!("cargo:info={builder:?}");
|
|
let bindings = builder.build_ptx().unwrap();
|
|
bindings.write("src/ptx.rs").unwrap();
|
|
}
|