[package] name = "rtx-fusion" version = "1.0.0" edition.workspace = true authors.workspace = true license.workspace = true repository.workspace = true description = "Automatic kernel fusion for RustyTorch - stream-based fusion similar to Burn's burn-fusion" keywords = ["gpu", "kernel-fusion", "deep-learning", "optimization"] categories = ["science", "algorithms"] rust-version = "1.92" [dependencies] # Core RTX dependencies rtx-backend = { workspace = true } rtx-runtime = { workspace = true } # CubeCL for kernel generation (optional) cubecl = { workspace = true, optional = true } cubecl-core = { workspace = true, optional = true } cubecl-runtime = { workspace = true, optional = true } # Error handling anyhow.workspace = true thiserror.workspace = true # Logging and tracing tracing.workspace = true # Serialization (for cache persistence) serde = { workspace = true, features = ["derive"] } # Concurrency parking_lot.workspace = true # Collections indexmap = { workspace = true } dashmap.workspace = true # Hashing for kernel signatures sha2.workspace = true [dev-dependencies] criterion.workspace = true tokio-test.workspace = true tempfile.workspace = true [features] default = ["cubecl"] cubecl = ["dep:cubecl", "dep:cubecl-core", "dep:cubecl-runtime"] [[bench]] name = "fusion_bench" harness = false [lints] workspace = true