datafusion-gpu/Cargo.toml at main · gabotechs/datafusion-gpu · GitHub

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
[package]
name = "datafusion-gpu"
version = "0.1.0"
edition = "2021"

[features]
cuda = ["cubecl/cuda", "cudarc"]

[dependencies]
datafusion = "43.0.0"
anyhow = "1.0.93"
tokio = { version = "1.41.1" , features = ["full"]}
delegate = "0.13.1"
cubecl = { git = "https://github.com/tracel-ai/cubecl", rev = "1c4e0036c043422773fd6824c2a888160fca8e5e", features = ["wgpu"]}
cubecl-reduce = { git = "https://github.com/tracel-ai/cubecl", rev = "1c4e0036c043422773fd6824c2a888160fca8e5e"}
rand = "0.8.5"
clap = { version = "4.5.21", features = ["derive"] }
rustyline = "15.0.0"
cudarc = { version = "0.12.1" , optional = true }

[dev-dependencies]
criterion = { version = "0.6", features = ["async_tokio"] }

[[bench]]
name = "cuda_sum"
harness = false
required-features = ["cuda"]