[package] authors = ["nathanielsimard "] categories = ["science"] description = "CUDA runtime for CubeCL" edition.workspace = true keywords = ["gpu", "cuda"] license.workspace = true name = "cubecl-cuda" readme.workspace = true repository = "https://github.com/tracel-ai/cubecl/tree/main/crates/cubecl-cuda" version.workspace = true [features] default = [ "cubecl-runtime/default", "cubecl-common/default", "cubecl-core/default", "cudarc/cuda-12050", ] std = ["cubecl-runtime/std", "cubecl-common/std", "cubecl-core/std"] [dependencies] cubecl-common = { path = "../cubecl-common", version = "0.3.0", default-features = false } cubecl-cpp = { path = "../cubecl-cpp", version = "0.3.0", default-features = false, features = ["cuda"] } cubecl-core = { path = "../cubecl-core", version = "0.3.0", default-features = false } cubecl-runtime = { path = "../cubecl-runtime", version = "0.3.0", default-features = false, features = [ "channel-mutex", ] } bytemuck = { workspace = true } cudarc = { version = "0.12.1", features = [ "std", "driver", "cuda-version-from-build-system", ], default-features = false } derive-new = { workspace = true } half = { workspace = true } log = { workspace = true } [dev-dependencies] cubecl-core = { path = "../cubecl-core", version = "0.3.0", features = [ "export_tests", ] } cubecl-linalg = { path = "../cubecl-linalg", version = "0.3.0", features = [ "export_tests", ] } pretty_assertions = { workspace = true }