[package]
edition = "2024"
name = "ruPRIM"
version = "0.1.3"
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Parallel primitives, reductions, scans, and indexing for Ruda."
readme = "README.md"
license-file = "LICENSE"
repository = "https://github.com/shuqi2077/RUDA"
[features]
cpu-reference = [
"kernel-ir",
"dep:ruda-test-utils",
]
cuda-tests = ["device-primitives"]
default = [
"kernel-ir",
"std",
]
device-primitives = [
"kernel-ir",
"ruda-kernel/device-tensor",
"dep:ruda-core",
]
elementwise = [
"kernel-ir",
"ruda-kernel/device-tensor",
"dep:ruda-core",
]
extended = ["heavy"]
full = [
"extended",
"device-primitives",
]
heavy = []
indexing = ["elementwise"]
kernel-ir = [
"dep:ruda-kernel",
"ruda/runtime",
"dep:num-traits",
"dep:serde",
"dep:thiserror",
]
std = [
"kernel-ir",
"ruda-kernel/frontend-std",
"thiserror/std",
]
tensor-reduce = [
"kernel-ir",
"ruda-kernel/device-tensor",
"dep:ruda-core",
]
tensor-reduce-autotune = ["tensor-reduce"]
tensor-scan = [
"kernel-ir",
"ruda-kernel/device-tensor",
"dep:ruda-core",
]
[lib]
name = "ruprim"
path = "src/lib.rs"
[[test]]
name = "device_primitives"
path = "tests/device_primitives.rs"
[[test]]
name = "record_frontend"
path = "tests/record_frontend.rs"
[[test]]
name = "reduce"
path = "tests/reduce/lib.rs"
required-features = [
"kernel-ir",
"cpu-reference",
]
[dependencies.half]
version = "=2.7.1"
features = [
"alloc",
"num-traits",
"serde",
]
[dependencies.num-traits]
version = "0.2.19"
optional = true
[dependencies.pretty_assertions]
version = "1.4"
optional = true
[dependencies.ruda]
version = "0.1.4"
default-features = false
[dependencies.ruda-core]
version = "0.1.3"
features = ["tensor"]
optional = true
default-features = false
[dependencies.ruda-kernel]
version = "0.1.3"
features = ["kernel-ir"]
optional = true
default-features = false
[dependencies.ruda-test-utils]
version = "0.1.2"
optional = true
default-features = false
[dependencies.serde]
version = "1.0.204"
features = [
"derive",
"alloc",
]
optional = true
default-features = false
[dependencies.thiserror]
version = "2"
optional = true
default-features = false
[dev-dependencies]
[lints.rust]
unsafe_code = "deny"