ruPRIM 0.1.9

Parallel primitives, reductions, scans, and indexing for Ruda.
Documentation 
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

[package]
name = "ruPRIM"
version.workspace = true
edition.workspace = true
description = "Parallel primitives, reductions, scans, and indexing for Ruda."
license-file = "LICENSE"
repository = "https://github.com/shuqi2077/RUDA"

[lib]
name = "ruprim"

[features]
default = ["kernel-ir", "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"]
device-primitives = ["kernel-ir", "ruda-kernel/device-tensor", "dep:ruda-core"]
cuda-tests = ["device-primitives", "ruda-test-runtime/cuda"]
indexing = ["elementwise"]
elementwise = ["kernel-ir", "ruda-kernel/device-tensor", "dep:ruda-core"]
kernel-ir = ["dep:ruda-kernel", "ruda/runtime", "dep:num-traits", "dep:serde", "dep:thiserror"]
std = ["kernel-ir", "ruda-kernel/frontend-std", "ruda-test-runtime/std", "thiserror/std"]
cpu-reference = ["kernel-ir", "dep:ruda-test-utils"]
heavy = []
extended = ["heavy"]
full = ["extended", "device-primitives"]

[dependencies]
ruda = { version = "0.1.5", path = "../ruda", default-features = false }
half = { version = "=2.7.1", features = ["alloc", "num-traits", "serde"] }

[dev-dependencies]
ruda-kernel = { path = "../ruda-kernel" }

[lints.rust]
unsafe_code = "deny"

[dependencies.ruda-kernel]
path = "../ruda-kernel"
version = "0.1.4"
default-features = false
features = ["kernel-ir"]
optional = true

[dependencies.num-traits]
version = "0.2.19"
optional = true

[dependencies.pretty_assertions]
version = "1.4"
optional = true

[dependencies.serde]
version = "1.0.204"
features = ["derive", "alloc"]
default-features = false
optional = true

[dependencies.thiserror]
version = "2"
default-features = false
optional = true

[dependencies.ruda-test-utils]
path = "../ruda-test-utils"
version = "0.1.9"
default-features = false
optional = true

[dev-dependencies.ruda-test-runtime]
path = "../ruda-test-runtime"
features = ["test-runtime"]
default-features = false

[dev-dependencies.ruda-test-utils]
path = "../ruda-test-utils"

[dev-dependencies.ruda-core]
path = "../ruda-core"
default-features = false

[[test]]
name = "reduce"
path = "tests/reduce/lib.rs"
required-features = ["kernel-ir", "cpu-reference"]

[dependencies.ruda-core]
path = "../ruda-core"
version = "0.1.3"
default-features = false
features = ["tensor"]
optional = true