baracuda-forge 0.0.1-alpha.68

Build-time CUDA kernel compiler for the baracuda ecosystem: nvcc-driven incremental builds, parallel compilation, GPU auto-detection, and CUTLASS / custom git dependency support.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

//! # baracuda-forge
//!
//! Build-time CUDA kernel compiler for the [baracuda] ecosystem. Drop it into
//! your `[build-dependencies]` and turn `.cu` files into a static library or
//! PTX with `nvcc`, with incremental rebuilds, parallel compilation, GPU
//! compute-capability auto-detection, and integrated CUTLASS support.
//!
//! `baracuda-forge` is the **build-time** companion to baracuda's runtime
//! wrappers (`baracuda-driver`, `baracuda-runtime`, ...). Use forge to turn
//! your `.cu` files into a library; use the runtime crates to launch the
//! kernels from Rust.
//!
//! [baracuda]: https://github.com/ciresnave/baracuda
//!
//! ## Features
//!
//! - **Compute Capability Detection** — auto-detect from `nvidia-smi` or
//!   `CUDA_COMPUTE_CAP`, with per-file overrides for mixed architectures.
//! - **Incremental Builds** — only recompile modified kernels using SHA-256
//!   content hashing.
//! - **CUDA Toolkit Auto-Detection** — find `nvcc` and include paths via the
//!   shared [`baracuda_build`] detector.
//! - **C++ Standard Auto-Select** — defaults to `c++20` on CUDA ≥ 12.0,
//!   `c++17` on older toolkits. Override via [`KernelBuilder::cpp_std`].
//! - **External Dependencies** — built-in CUTLASS support, or fetch any git repo.
//! - **Parallel Compilation** — configurable thread percentage for parallel builds.
//! - **Flexible Source Selection** — directory, glob, files, or exclude patterns.
//!
//! ## Quick start
//!
//! ```no_run
//! use baracuda_forge::KernelBuilder;
//!
//! fn main() {
//!     let out_dir = std::env::var("OUT_DIR").expect("OUT_DIR must be set");
//!
//!     KernelBuilder::new()
//!         .source_dir("src/kernels")
//!         .exclude(&["*_test.cu"])
//!         .arg("-O3")
//!         .thread_percentage(0.5)
//!         .build_lib(format!("{}/libkernels.a", out_dir))
//!         .expect("CUDA compilation failed");
//!
//!     println!("cargo:rustc-link-search={}", out_dir);
//!     println!("cargo:rustc-link-lib=kernels");
//! }
//! ```
//!
//! ## Acknowledgments
//!
//! `baracuda-forge` is a vendored fork of [`cudaforge`] by Guoqing Bao,
//! adapted to baracuda's workspace conventions. See the `NOTICE` file at the
//! crate root for full provenance.
//!
//! [`cudaforge`]: https://github.com/guoqingbao/cudaforge

#![deny(missing_docs)]

mod builder;
mod compute_cap;
mod dependency;
mod error;
mod hash;
mod parallel;
mod source;
mod toolkit;

pub use builder::{KernelBuilder, PtxOutput};
pub use compute_cap::{detect_compute_cap, get_gpu_arch_string, ComputeCapability, GpuArch};
pub use dependency::{resolve_cutlass_from_cargo_checkouts, DependencyManager, ExternalDependency};
pub use error::{Error, Result};
pub use hash::BuildCache;
pub use parallel::ParallelConfig;
pub use source::{collect_headers, SourceSelector};
pub use toolkit::CudaToolkit;