[package]
edition = "2024"
name = "ruPRIM-host"
version = "0.1.40"
authors = ["Dilshod Tadjibaev (@antimora)"]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "ruPRIM CPU tensor primitives, reductions, indexing and SIMD kernels."
readme = "README.md"
license = "MIT OR Apache-2.0"
repository = "https://github.com/shuqi2077/RUDA"
[features]
default = [
"std",
"simd",
"rayon",
]
rayon = [
"dep:rayon",
"ruda-core/tensor-host-parallel",
]
simd = [
"dep:macerator",
"dep:aligned-vec",
]
std = [
"ruda-core/std",
"half/std",
"num-traits/std",
]
[lib]
name = "ruprim_host"
path = "src/lib.rs"
[[example]]
name = "select_bench"
path = "examples/select_bench.rs"
[[example]]
name = "topk_bench"
path = "examples/topk_bench.rs"
[dependencies.aligned-vec]
version = "0.6"
optional = true
default-features = false
[dependencies.bytemuck]
version = "1.25.0"
[dependencies.half]
version = "2.7.1"
features = [
"alloc",
"num-traits",
"serde",
]
default-features = false
[dependencies.libm]
version = "0.2.15"
[dependencies.macerator]
version = "0.3.0"
optional = true
[dependencies.num-traits]
version = "0.2.19"
features = ["libm"]
default-features = false
[dependencies.rayon]
version = "1.10.0"
optional = true
[dependencies.ruda-core]
version = "0.1.5"
features = ["tensor-host-storage"]
default-features = false